d7, d30 retention in convo admin

GooeyAI · Nov 7, 2023 · 4124b4a · 4124b4a
1 parent f5e72c5
commit 4124b4a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/bots/admin.py b/bots/admin.py
@@ -286,7 +286,8 @@ class ConversationAdmin(admin.ModelAdmin):
         "view_messages",
         "view_last_active_delta",
         "d1",
-        "d3",
+        "d7",
+        "d30",
     ]
     readonly_fields = [
         "created_at",

diff --git a/bots/models.py b/bots/models.py
@@ -519,7 +519,8 @@ def to_df(self, tz=pytz.timezone(settings.TIME_ZONE)) -> "pd.DataFrame":
                         convo.last_active_delta().total_seconds() / 3600
                     ),
                     "D1": convo.d1(),
-                    "D3": convo.d3(),
+                    "D7": convo.d7(),
+                    "D30": convo.d30(),
                 }
             except Message.DoesNotExist:
                 pass
@@ -659,11 +660,17 @@ def d1(self):
     d1.short_description = "D1"
     d1.boolean = True
 
-    def d3(self):
-        return self.last_active_delta() > datetime.timedelta(days=3)
+    def d7(self):
+        return self.last_active_delta() > datetime.timedelta(days=7)
 
-    d3.short_description = "D3"
-    d3.boolean = True
+    d7.short_description = "D7"
+    d7.boolean = True
+
+    def d30(self):
+        return self.last_active_delta() > datetime.timedelta(days=30)
+
+    d30.short_description = "D30"
+    d30.boolean = True
 
 
 class MessageQuerySet(models.QuerySet):