from Schemas import AddressStreet, Users, Duty
from Services.PostgresService.controllers.pagination_controllers import Pagination, PaginationConfig


if __name__ == "__main__":
    db = AddressStreet.new_session()
    AddressStreet.pre_query = AddressStreet.filter_all(
        AddressStreet.street_name.ilike(
            "A%",
        ),
        db=db,
    ).core_query
    account_list = AddressStreet.filter_all(
        AddressStreet.expiry_starts >= "2000-01-01",
        AddressStreet.expiry_ends >= "2050-12-31",
        AddressStreet.street_name.ilike("%B%"),
        db=db,
    )
    pagination_config = PaginationConfig(
        page=200,
        size=50,
        order_field=["uu_id"],
        order_type=["asc"],
    )
    pagination = Pagination(data=account_list)
    pagination.change(config=pagination_config)

    user_find_one = Users.filter_one(
        Users.uu_id == "45554ebb-422d-4da7-b89a-1fcfb7e41414",
        db=db,
    )
    user_wrong_find_one = Users.filter_one(
        Users.uu_id == "95554ebb-422d-4da7-b89a-1fcfb7e41414",
        db=db,
    )

    created_duty = Duty.find_or_create(
        duty_name="Test",
        duty_code="T50",
        duty_description=1,
        db=db,
    )
    print('created_duty wrong', created_duty)
    created_duty.save(db=db)
    print('created_duty', created_duty)