43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from lib.gen_2 import (
|
|
Employee,
|
|
EmployeeContainer,
|
|
Partner,
|
|
PartnerContainer,
|
|
UserContainer,
|
|
)
|
|
|
|
|
|
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()
|