Files
csv-loader-generics-present…/generics_1.py
2023-02-16 13:00:04 +01:00

42 lines
1.1 KiB
Python

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