some admin extension (#196)

* Adds timed exec table & new fields

* Removed tracking flag since all will be tracked atm

* Collect data for admin dashboard

* Remove oidc identifier

Author: JWittmeyer

business_objects/user.py

@@ -137,7 +137,7 @@ def delete(user_id: str, with_commit: bool = False) -> None:
 
 def get_missing_users(user_ids: List[str]):
     query = f"""
-    SELECT jsonb_object_agg(u.id, u.last_interaction)
+    SELECT jsonb_object_agg(u.id, jsonb_build_object('last_interaction', u.last_interaction,'messages_created_this_month', u.messages_created_this_month))
     FROM public.user u
     WHERE id IN ({','.join([f"'{user_id}'" for user_id in user_ids])})
     """

enums.py

@@ -177,6 +177,7 @@ class Tablenames(Enum):
     SUMS_TABLE = "sums_table"
     ADMIN_QUERY_MESSAGE_SUMMARY = "admin_query_message_summary"
     RELEASE_NOTIFICATION = "release_notification"
+    TIMED_EXECUTIONS = "timed_executions"
 
     def snake_case_to_pascal_case(self):
         # the type name (written in PascalCase) of a table is needed to create backrefs
@@ -1012,3 +1013,7 @@ class MessageInitiationType(Enum):
     UI = "UI"
     API = "API"
     MACRO = "MACRO"
+
+
+class TimedExecutionKey(Enum):
+    LAST_RESET_USER_MESSAGE_COUNT = "LAST_RESET_USER_MESSAGE_COUNT"

global_objects/timed_executions.py

@@ -0,0 +1,63 @@
+from typing import List
+from datetime import datetime
+
+from ..business_objects import general
+from ..session import session
+from ..models import TimedExecutions
+from ..enums import TimedExecutionKey
+from ..models import User
+
+
+def get(key: TimedExecutionKey) -> TimedExecutions:
+    return (
+        session.query(TimedExecutions)
+        .filter(
+            TimedExecutions.time_key == key.value,
+        )
+        .first()
+    )
+
+
+def get_all() -> List[TimedExecutions]:
+    return session.query(TimedExecutions).all()
+
+
+def execute_time_key_update(
+    with_commit: bool = True,
+) -> None:
+    # potential optimization by only getting "needed" keys so the time check is already done in the query
+    all_keys = {key for key in TimedExecutionKey}
+    all_existing = {obj.time_key: obj for obj in get_all()}
+
+    for key in all_keys:
+        obj = all_existing.get(key.value)
+        if obj:
+            last_executed_at = obj.last_executed_at
+        else:
+            last_executed_at = datetime.min
+        if __execute_timed_execution_by_key(key, last_executed_at):
+            if obj:
+                obj.last_executed_at = datetime.now()
+            else:
+                obj = TimedExecutions(
+                    time_key=key.value,
+                    last_executed_at=datetime.now(),
+                )
+                general.add(obj, False)
+    general.flush_or_commit(with_commit)
+
+
+def __execute_timed_execution_by_key(
+    key: TimedExecutionKey, last_executed_at: datetime
+) -> bool:
+    if key == TimedExecutionKey.LAST_RESET_USER_MESSAGE_COUNT:
+        # check if month has changed since last execution
+        now = datetime.now()
+        if last_executed_at.year == now.year and last_executed_at.month == now.month:
+            return False  # already executed this month
+
+        session.query(User).update({User.messages_created_this_month: 0})
+        general.flush_or_commit(False)
+        return True
+
+    raise NotImplementedError(f"Timed execution for key {key} is not implemented yet")

models.py

@@ -226,9 +226,9 @@ class User(Base):
     created_at = Column(DateTime, default=sql.func.now())
     metadata_public = Column(JSON)
     sso_provider = Column(String)
-    oidc_identifier = Column(String)
     use_new_cognition_ui = Column(Boolean, default=True)
     auto_logout_minutes = Column(Integer)
+    messages_created_this_month = Column(BigInteger, default=0)
 
 
 class Team(Base):
@@ -2505,3 +2505,10 @@ class ReleaseNotification(Base):
     )
     link = Column(String, nullable=False)
     config = Column(JSON)  # e.g. {"en": {"headline":"", "description":""}, "de": {...}}
+
+
+class TimedExecutions(Base):
+    __tablename__ = Tablenames.TIMED_EXECUTIONS.value
+    __table_args__ = {"schema": "global"}
+    time_key = Column(String, unique=True, primary_key=True)  # enums.TimedExecutionKey
+    last_executed_at = Column(DateTime)