from lib.gen_1 import (
    Employee,
    EmployeeContainer,
    Partner,
    PartnerContainer,
)


def main():
    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"),
        ]
    )

    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"),
        ]
    )

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

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


if __name__ == "__main__":
    main()