3wordchant/hamster-tools
0
0
Fork 0
Code Issues 8 Pull Requests Releases Wiki Activity

Fix mapping import, show mapping count on list

Browse Source
This commit is contained in:
3wc
2023-10-29 13:37:40 +00:00
parent f8f83ce4d4
commit ccbbc80116
2 changed files with 31 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
hamstertools/app.py
View File

@ -15,6 +15,7 @@ from .db import (
KimaiProject,
KimaiCustomer,
KimaiActivity,
HamsterKimaiMapping,
)
from .kimai import (
KimaiAPI,
@ -78,15 +79,34 @@ class ActivitiesScreen(ListScreen):
def _refresh(self, filter_query=None):
self.table.clear()
facts_count_query = (
HamsterFact
.select(HamsterFact.activity_id, fn.COUNT(HamsterFact.id).alias('facts_count'))
.group_by(HamsterFact.activity_id)
.alias('facts_count_query')
)
mappings_count_query = (
HamsterKimaiMapping
.select(HamsterKimaiMapping.hamster_activity_id,
fn.COUNT(HamsterKimaiMapping.id).alias('mappings_count'))
.group_by(HamsterKimaiMapping.hamster_activity_id)
.alias('mappings_count_query')
)
activities = (
HamsterActivity.select(
HamsterActivity,
HamsterCategory,
fn.Count(HamsterFact.id).alias("facts_count")
HamsterCategory.id,
fn.COALESCE(HamsterCategory.name, 'None').alias("category_name"),
fn.COALESCE(facts_count_query.c.facts_count, 0).alias('facts_count'),
fn.COALESCE(mappings_count_query.c.mappings_count, 0).alias('mappings_count')
)
.join(HamsterCategory, JOIN.LEFT_OUTER)
.switch(HamsterActivity)
.join(HamsterFact, JOIN.LEFT_OUTER)
.join(facts_count_query, JOIN.LEFT_OUTER, on=(HamsterActivity.id == facts_count_query.c.activity_id))
.switch(HamsterActivity)
.join(mappings_count_query, JOIN.LEFT_OUTER, on=(HamsterActivity.id == mappings_count_query.c.hamster_activity_id))
.group_by(HamsterActivity)
)
@ -100,10 +120,11 @@ class ActivitiesScreen(ListScreen):
[
[
activity.category_id,
(activity.category.name if (activity.category_id != -1) else ""),
activity.category_name,
activity.id,
activity.name,
activity.facts_count,
activity.mappings_count,
]
for activity in activities
]
@ -115,7 +136,7 @@ class ActivitiesScreen(ListScreen):
self.table = self.query_one(DataTable)
self.table.cursor_type = "row"
self.columns = self.table.add_columns(
"category id", "category", "activity id", "activity", "entries"
"category id", "category", "activity id", "activity", "entries", "mappings"
)
self.sort = (self.columns[1], self.columns[3])
self._refresh()