Show kimai activity counts, clear data before get

This commit is contained in:
3wc
2023-10-28 01:21:23 +01:00
parent cd278b32aa
commit c68e373b18
3 changed files with 45 additions and 8 deletions

View File

@ -240,6 +240,7 @@ class KimaiScreen(ListScreen):
.join(KimaiCustomer, JOIN.LEFT_OUTER)
.switch(KimaiProject)
.join(KimaiActivity, JOIN.LEFT_OUTER)
.where(KimaiActivity.project.is_null(False))
.group_by(KimaiProject)
)
@ -249,7 +250,6 @@ class KimaiScreen(ListScreen):
| KimaiCustomer.name.contains(filter_query)
)
self.table.add_rows(
[
[
@ -268,6 +268,10 @@ class KimaiScreen(ListScreen):
def action_get(self) -> None:
api = KimaiAPI()
KimaiCustomer.delete().execute()
KimaiProject.delete().execute()
KimaiActivity.delete().execute()
customers = KimaiAPICustomer.list(api)
with db.atomic():
KimaiCustomer.insert_many([{
@ -283,6 +287,14 @@ class KimaiScreen(ListScreen):
'customer_id': project.customer.id
} 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()
self._refresh()
def on_mount(self) -> None: