Added db querry for user subscribtion

app/assistants.py

@@ -660,6 +660,7 @@ class Assistant:
                             # "growth_guide_session",
                             # "life_score",
                             # "recent_wins",
+                            # "subscription/payments"
                             # ]
                 logger.info(f"Getting user information: {category}", 
                         extra={"user_id": self.cm.user.user_id, "endpoint": "assistant_get_user_info"})

app/requirements.txt

@@ -14,4 +14,5 @@ regex==2024.9.11
 Requests==2.32.3
 cachetools>=5.0.0
 pdfkit==1.0.0
-PyPDF2==3.0.1
+PyPDF2==3.0.1
+sentry-sdk==2.19.2

app/utils.py

@@ -1393,6 +1393,58 @@ def get_users_mementos(user_id, date):
     finally:
         return mementos
 
+@catch_error
+def get_user_subscriptions(user_id):
+    function_name = get_user_subscriptions.__name__
+    logger.info(f"Retrieving subscriptions for user {user_id}", extra={'endpoint': function_name})
+    db_params = {
+        'dbname': 'ourcoach',
+        'user': 'ourcoach',
+        'password': 'hvcTL3kN3pOG5KteT17T',
+        'host': 'staging-ourcoach.cx8se8o0iaiy.ap-southeast-1.rds.amazonaws.com',
+        'port': '5432'
+    }
+    try:
+        with psycopg2.connect(**db_params) as conn:
+            with conn.cursor() as cursor:
+                query = sql.SQL("""
+                    SELECT * FROM {table} 
+                    WHERE user_id = %s 
+                    ORDER BY period_started DESC
+                """).format(table=sql.Identifier('public', 'user_subscription'))
+                cursor.execute(query, (user_id,))
+                rows = cursor.fetchall()
+                colnames = [desc[0] for desc in cursor.description]
+                df = pd.DataFrame(rows, columns=colnames)
+                logger.info(f"Retrieved {len(df)} subscriptions for user {user_id}", extra={'endpoint': function_name})
+                return df
+    except psycopg2.Error as e:
+        logger.error(f"Database error while retrieving user subscriptions: {e}", extra={'endpoint': function_name})
+        raise DBError(user_id=user_id, message="Error retrieving user subscriptions", code="SQLError", e=str(e))
+def get_user_subscriptions():
+    function_name = get_user_subscriptions.__name__
+    logger.info(f"Retrieving all user subscriptions", extra={'endpoint': function_name})
+    db_params = {
+        'dbname': 'ourcoach',
+        'user': 'ourcoach',
+        'password': 'hvcTL3kN3pOG5KteT17T',
+        'host': 'staging-ourcoach.cx8se8o0iaiy.ap-southeast-1.rds.amazonaws.com',
+        'port': '5432'
+    }
+    try:
+        with psycopg2.connect(**db_params) as conn:
+            with conn.cursor() as cursor:
+                query = sql.SQL("SELECT * FROM {table}").format(table=sql.Identifier('public', 'user_subscription'))
+                cursor.execute(query)
+                rows = cursor.fetchall()
+                colnames = [desc[0] for desc in cursor.description]
+                df = pd.DataFrame(rows, columns=colnames)
+                logger.info(f"Retrieved {len(df)} user subscriptions", extra={'endpoint': function_name})
+                return df
+    except psycopg2.Error as e:
+        logger.error(f"Database error while retrieving user subscriptions: {e}", extra={'endpoint': function_name})
+        raise DBError(user_id='no-user', message="Error retrieving user subscriptions", code="SQLError", e=str(e))
+
 def generate_uuid():
     return str(uuid.uuid4())