54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
|
from .kimai import (
|
||
|
|
KimaiAPI,
|
||
|
|
Customer as KimaiAPICustomer,
|
||
|
|
Project as KimaiAPIProject,
|
||
|
|
Activity as KimaiAPIActivity,
|
||
|
|
)
|
||
|
|
from .db import (
|
||
|
|
db,
|
||
|
|
KimaiProject,
|
||
|
|
KimaiCustomer,
|
||
|
|
KimaiActivity,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def sync() -> None:
|
||
|
|
api = KimaiAPI()
|
||
|
|
|
||
|
|
KimaiCustomer.delete().execute()
|
||
|
|
KimaiProject.delete().execute()
|
||
|
|
KimaiActivity.delete().execute()
|
||
|
|
|
||
|
|
customers = KimaiAPICustomer.list(api)
|
||
|
|
with db.atomic():
|
||
|
|
KimaiCustomer.insert_many(
|
||
|
|
[{"id": customer.id, "name": customer.name} for customer in customers]
|
||
|
|
).execute()
|
||
|
|
|
||
|
|
projects = KimaiAPIProject.list(api)
|
||
|
|
with db.atomic():
|
||
|
|
KimaiProject.insert_many(
|
||
|
|
[
|
||
|
|
{
|
||
|
|
"id": project.id,
|
||
|
|
"name": project.name,
|
||
|
|
"customer_id": project.customer.id,
|
||
|
|
"allow_global_activities": project.allow_global_activities,
|
||
|
|
}
|
||
|
|
for project in projects
|
||
|
|
]
|
||
|
|
).execute()
|
||
|
|
|
||
|
|
activities = KimaiAPIActivity.list(api)
|
||
|
|
with db.atomic():
|
||
|
|
KimaiActivity.insert_many(
|
||
|
|
[
|
||
|
|
{
|
||
|
|
"id": activity.id,
|
||
|
|
"name": activity.name,
|
||
|
|
"project_id": (activity.project and activity.project.id or None),
|
||
|
|
}
|
||
|
|
for activity in activities
|
||
|
|
]
|
||
|
|
).execute()
|