Mbonea commited on
Commit
c353d4b
·
1 Parent(s): fbeb0b0

added metrics

Browse files
App/Analytics/AnalyticsRoutes.py DELETED
@@ -1,7 +0,0 @@
1
- from fastapi import APIRouter, status
2
- from .Schemas import BaseRequest, editRequest
3
- from .Model import Comments
4
- from App.Users.Model import User
5
- from App.Post.Model import Post
6
-
7
- analytics_router = APIRouter(tags=["Analytics"])
 
 
 
 
 
 
 
 
App/Analytics/Model__.py DELETED
@@ -1,23 +0,0 @@
1
- import asyncio
2
- import orm
3
- import psycopg2
4
- import datetime
5
- import pydantic
6
- from App.modelInit import database, models
7
- from App.Users.Model import User
8
-
9
-
10
- class Analytics(orm.Model):
11
- tablename = "Analytics"
12
- registry = models
13
- fields = {
14
- "id": orm.Integer(primary_key=True),
15
- "user": orm.ForeignKey(
16
- User, on_delete=orm.CASCADE, allow_null=True
17
- ), # Optional for unknown users.
18
- "post": orm.ForeignKey(User, on_delete=orm.CASCADE),
19
- "ip": orm.String(max_length=100),
20
- "device": orm.String(max_length=100),
21
- "country": orm.String(max_length=100),
22
- "createdAt": orm.DateTime(index=True, default=datetime.datetime.now),
23
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
App/Analytics/Schemas.py DELETED
@@ -1,13 +0,0 @@
1
- from typing import List, Optional
2
- from pydantic import EmailStr, BaseModel
3
- from datetime import date, datetime, time, timedelta
4
-
5
-
6
- class BaseRequest(BaseModel):
7
- user: Optional[int]
8
- id: Optional[int]
9
- content: Optional[str]
10
-
11
-
12
- class editRequest(BaseRequest):
13
- updatedAt: datetime = datetime.now()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
App/Metrics/MetricsRoutes.py ADDED
@@ -0,0 +1,143 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import APIRouter
2
+ from typing import List
3
+ from datetime import datetime, timedelta
4
+ from decimal import Decimal
5
+ from tortoise.functions import Sum
6
+
7
+ from App.Payments.Model import Payment
8
+ from App.Users.Model import User
9
+ from .Schema import (
10
+ RevenueMetricsResponse,
11
+ RecurringUserResponse,
12
+ )
13
+
14
+ metrics_router = APIRouter(tags=["Metrics"])
15
+
16
+
17
+ # Helper function to get the start of the week, month, and day
18
+ def get_time_frames():
19
+ today = datetime.now()
20
+ start_of_day = today.replace(hour=0, minute=0, second=0, microsecond=0)
21
+ start_of_week = start_of_day - timedelta(
22
+ days=start_of_day.weekday()
23
+ ) # Start of the current week (Monday)
24
+ start_of_month = start_of_day.replace(
25
+ day=1, hour=0, minute=0, second=0, microsecond=0
26
+ ) # Start of the current month
27
+
28
+ return start_of_day, start_of_week, start_of_month
29
+
30
+
31
+ @metrics_router.get("/metrics/revenue/daily", response_model=RevenueMetricsResponse)
32
+ async def get_daily_revenue():
33
+ start_of_day, _, _ = get_time_frames()
34
+
35
+ # Sum up the payments made today using Tortoise ORM's annotate and values
36
+ revenue_data = (
37
+ await Payment.filter(created_time__gte=start_of_day)
38
+ .annotate(total_revenue=Sum("amount"))
39
+ .values("total_revenue")
40
+ )
41
+
42
+ if revenue_data and revenue_data[0]["total_revenue"] is not None:
43
+ total_revenue = revenue_data[0]["total_revenue"]
44
+ else:
45
+ total_revenue = Decimal("0.00")
46
+
47
+ return RevenueMetricsResponse(
48
+ period="daily",
49
+ revenue=float(total_revenue),
50
+ )
51
+
52
+
53
+ @metrics_router.get("/metrics/revenue/weekly", response_model=RevenueMetricsResponse)
54
+ async def get_weekly_revenue():
55
+ _, start_of_week, _ = get_time_frames()
56
+
57
+ # Sum up the payments made in the current week
58
+ revenue_data = (
59
+ await Payment.filter(created_time__gte=start_of_week)
60
+ .annotate(total_revenue=Sum("amount"))
61
+ .values("total_revenue")
62
+ )
63
+
64
+ if revenue_data and revenue_data[0]["total_revenue"] is not None:
65
+ total_revenue = revenue_data[0]["total_revenue"]
66
+ else:
67
+ total_revenue = Decimal("0.00")
68
+
69
+ return RevenueMetricsResponse(
70
+ period="weekly",
71
+ revenue=float(total_revenue),
72
+ )
73
+
74
+
75
+ @metrics_router.get("/metrics/revenue/monthly", response_model=RevenueMetricsResponse)
76
+ async def get_monthly_revenue():
77
+ _, _, start_of_month = get_time_frames()
78
+
79
+ # Sum up the payments made in the current month
80
+ revenue_data = (
81
+ await Payment.filter(created_time__gte=start_of_month)
82
+ .annotate(total_revenue=Sum("amount"))
83
+ .values("total_revenue")
84
+ )
85
+
86
+ if revenue_data and revenue_data[0]["total_revenue"] is not None:
87
+ total_revenue = revenue_data[0]["total_revenue"]
88
+ else:
89
+ total_revenue = Decimal("0.00")
90
+
91
+ return RevenueMetricsResponse(
92
+ period="monthly",
93
+ revenue=float(total_revenue),
94
+ )
95
+
96
+
97
+ @metrics_router.get(
98
+ "/metrics/recurring-users", response_model=List[RecurringUserResponse]
99
+ )
100
+ async def get_recurring_users():
101
+ # Get the start of the day and 6 days ago (total of 7 days including today)
102
+ start_of_day, _, _ = get_time_frames()
103
+ seven_days_ago = start_of_day - timedelta(days=6)
104
+
105
+ # Find all payments in the last 7 days
106
+ payments = await Payment.filter(created_time__gte=seven_days_ago).values(
107
+ "user_id", "created_time"
108
+ )
109
+
110
+ # Initialize a dictionary to hold payment dates per user
111
+ user_payment_dates = {}
112
+
113
+ # Collect payment dates per user
114
+ for payment in payments:
115
+ user_id = payment["user_id"]
116
+ payment_date = payment["created_time"].date()
117
+
118
+ if user_id not in user_payment_dates:
119
+ user_payment_dates[user_id] = set()
120
+
121
+ user_payment_dates[user_id].add(payment_date)
122
+
123
+ # Calculate the set of dates we expect payments on
124
+ expected_dates = set((start_of_day - timedelta(days=i)).date() for i in range(7))
125
+
126
+ # Filter users who have made payments on each expected date
127
+ recurring_users = [
128
+ user_id
129
+ for user_id, payment_dates in user_payment_dates.items()
130
+ if payment_dates == expected_dates
131
+ ]
132
+
133
+ # Fetch user details for recurring users
134
+ recurring_user_details = await User.filter(id__in=recurring_users).all()
135
+
136
+ return [
137
+ RecurringUserResponse(
138
+ user_id=str(user.id),
139
+ username=user.username, # Adjust this field based on your User model
140
+ email=user.email, # Adjust as necessary
141
+ )
142
+ for user in recurring_user_details
143
+ ]
App/Metrics/Schema.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from pydantic import BaseModel
2
+ from typing import List
3
+
4
+
5
+ class RevenueMetricsResponse(BaseModel):
6
+ period: str # daily, weekly, or monthly
7
+ revenue: float
8
+
9
+
10
+ class RecurringUserResponse(BaseModel):
11
+ user_id: str
12
+ username: str
13
+ email: str
App/app.py CHANGED
@@ -7,6 +7,7 @@ from .modelInit import TORTOISE_ORM
7
  from .Payments.PaymentsRoutes import payment_router
8
  from .Plans.PlanRoutes import plan_router
9
  from .Portals.PortalRoutes import portal_router
 
10
 
11
  app = FastAPI()
12
 
@@ -28,6 +29,7 @@ app.include_router(subscription_router)
28
  app.include_router(payment_router)
29
  app.include_router(plan_router)
30
  app.include_router(portal_router)
 
31
 
32
 
33
  if __name__ == "__main__":
 
7
  from .Payments.PaymentsRoutes import payment_router
8
  from .Plans.PlanRoutes import plan_router
9
  from .Portals.PortalRoutes import portal_router
10
+ from .Metrics.MetricsRoutes import metrics_router
11
 
12
  app = FastAPI()
13
 
 
29
  app.include_router(payment_router)
30
  app.include_router(plan_router)
31
  app.include_router(portal_router)
32
+ app.include_router(metrics_router)
33
 
34
 
35
  if __name__ == "__main__":