paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-04 05:53:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

analytics: Factor out UserCount / StreamCount common checks.

Browse Source
This commit is contained in:
Alex Vandiver
2023-11-17 18:03:30 +00:00
committed by Tim Abbott
parent 626768f626
commit efa9bf36eb
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
analytics/lib/counts.py
View File

@@ -111,6 +111,9 @@ class DataCollector:
self.output_table = output_table self.output_table = output_table
self.pull_function = pull_function self.pull_function = pull_function
def depends_on_realm(self) -> bool:
return self.output_table in (UserCount, StreamCount)
## CountStat-level operations ## ## CountStat-level operations ##
@@ -199,7 +202,7 @@ def do_fill_count_stat_at_hour(
def do_delete_counts_at_hour(stat: CountStat, end_time: datetime) -> None: def do_delete_counts_at_hour(stat: CountStat, end_time: datetime) -> None:
if isinstance(stat, LoggingCountStat): if isinstance(stat, LoggingCountStat):
InstallationCount.objects.filter(property=stat.property, end_time=end_time).delete() InstallationCount.objects.filter(property=stat.property, end_time=end_time).delete()
if stat.data_collector.output_table in [UserCount, StreamCount]: if stat.data_collector.depends_on_realm():
RealmCount.objects.filter(property=stat.property, end_time=end_time).delete() RealmCount.objects.filter(property=stat.property, end_time=end_time).delete()
else: else:
UserCount.objects.filter(property=stat.property, end_time=end_time).delete() UserCount.objects.filter(property=stat.property, end_time=end_time).delete()
@@ -220,7 +223,7 @@ def do_aggregate_to_summary_table(
else: else:
realm_clause = SQL("") realm_clause = SQL("")
if output_table in (UserCount, StreamCount): if stat.data_collector.depends_on_realm():
realmcount_query = SQL( realmcount_query = SQL(
""" """
INSERT INTO analytics_realmcount INSERT INTO analytics_realmcount