chore(scripts): add script for deleting old users that never logged a…

…ny data

Makefile

@@ -2,6 +2,10 @@
 analytics:
 	@cd scripts && poetry run python3 ./analytics.py
 
+# Deletes old users that have never logged any data
+cleanup_zombie_users:
+	@cd scripts && poetry run python3 ./cleanup_zombie_users.py
+
 # Makes notifications_send.zip
 scripts/cloudformation/lambdas/notifications_send.zip: scripts/cloudformation/lambdas/notifications_send/*
 	@cd scripts/cloudformation/lambdas/notifications_send &&\
@@ -77,4 +81,4 @@ test: cloudformation/test
 test/ci:
 	@cd client && npm run test-ci && echo "🍄 All tests pass! 🍄"
 
-.PHONY: analytics cloudformation/test deploy deploy/dry-run deploy/notifications_send help init init/ci stack-policy start test test/ci
+.PHONY: analytics cleanup_zombie_users cloudformation/test deploy deploy/dry-run deploy/notifications_send help init init/ci stack-policy start test test/ci

scripts/cleanup_zombie_users.py

@@ -0,0 +1,34 @@
+import boto3
+from boto3.dynamodb.conditions import Key
+from datetime import datetime, timedelta, timezone
+
+USER_POOL_ID = "us-east-1_rdB8iu5X4"
+
+events_table = boto3.resource("dynamodb").Table("moodtracker_events")
+cognito_idp_client = boto3.client("cognito-idp")
+
+users_to_delete = [
+    u
+    for u in cognito_idp_client.get_paginator("list_users")
+    .paginate(
+        UserPoolId=USER_POOL_ID,
+        AttributesToGet=[],
+    )
+    .build_full_result()["Users"]
+    if u["UserLastModifiedDate"] < datetime.now(timezone.utc) - timedelta(365)
+    and not events_table.query(
+        KeyConditionExpression=Key("userId").eq(u["Username"]),
+        Limit=1,
+    )["Count"]
+]
+print("Users to delete:", users_to_delete)
+
+if input('Type "yes" to proceed or anything else to abort: ') == "yes":
+    print("Deleting users...")
+    for user in users_to_delete:
+        cognito_idp_client.admin_delete_user(
+            UserPoolId=USER_POOL_ID, Username=user["Username"]
+        )
+        print("Deleted user ID: ", user["Username"])
+else:
+    print("Aborted")

scripts/pyproject.toml

@@ -1,8 +1,9 @@
 [tool.poetry]
-name = "moodtracker-analytics"
+name = "moodtracker-scripts"
 version = "0.1.0"
 description = ""
 authors = ["Your Name <[email protected]>"]
+readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "^3.8"