from lib.generics_4_with_generics 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"),
        ]
    )

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

    developers = employees.get_by_department("Dev")
    print("Developers:")
    for employee in developers:
        print(employee)

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


if __name__ == "__main__":
    main()