-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathchart_data_api.py
More file actions
52 lines (40 loc) · 1.33 KB
/
chart_data_api.py
File metadata and controls
52 lines (40 loc) · 1.33 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
#!/usr/bin/env python3
"""
Command-line script for API endpoints to call.
Outputs JSON to stdout for API responses.
"""
import sys
import json
import argparse
from database import SplitDataDB
from chart_data import ChartDataGenerator
def main():
parser = argparse.ArgumentParser(description='Get user chart data')
parser.add_argument('--user-id', type=int, required=True, help='User ID')
parser.add_argument('--weeks', type=int, default=12, help='Number of weeks')
parser.add_argument('--months', type=int, default=12, help='Number of months')
args = parser.parse_args()
db = SplitDataDB()
try:
if not db.connect():
error_data = {"error": "Database connection failed"}
print(json.dumps(error_data))
sys.exit(1)
generator = ChartDataGenerator(db)
chart_data = generator.get_user_all_chart_data(
args.user_id,
weeks=args.weeks,
months=args.months
)
# Output JSON to stdout
print(json.dumps(chart_data, default=str))
except Exception as e:
error_data = {"error": str(e)}
print(json.dumps(error_data))
import traceback
traceback.print_exc()
sys.exit(1)
finally:
db.disconnect()
if __name__ == "__main__":
main()