from lib.gen_2 import (
    Employee,
    EmployeeContainer,
    Partner,
    PartnerContainer,
    UserContainer,
)


def main():
    users = UserContainer[Employee](
        [
            Employee(id=1, first_name="Pero", last_name="Perić", department="Dev"),
            Employee(id=2, first_name="Mirko", last_name="Mirković", department="Dev"),
            Employee(id=3, first_name="Ivo", last_name="Ivić", department="Sales"),
        ]
    )
    users[0]

    employees = EmployeeContainer(
        [
            Employee(id=1, first_name="Pero", last_name="Perić", department="Dev"),
            Employee(id=2, first_name="Mirko", last_name="Mirković", department="Dev"),
            Employee(id=3, first_name="Ivo", last_name="Ivić", department="Sales"),
        ]
    )
    employees[0]

    partners = PartnerContainer(
        [
            Partner(id=101, first_name="Ante", last_name="Antić", company="CocaCola"),
            Partner(id=102, first_name="Šime", last_name="Šimić", company="CocaCola"),
            Partner(id=103, first_name="Mate", last_name="Matić", company="Pepsi"),
        ]
    )
    partners[0]

    employee = employees.get_user_by_id(id=2)
    print("Found user:", employee)

    cocacola_partners = partners.get_by_company(company="CocaCola")
    for partner in cocacola_partners:
        print(partner)


if __name__ == "__main__":
    main()