initial commit

main.py

@@ -0,0 +1,98 @@
+import time
+from datetime import datetime, timedelta, date
+
+from flask import Flask, jsonify
+from flask import request
+import requests
+import json
+
+app = Flask(__name__)
+app.config['JSON_AS_ASCII'] = False
+
+@app.route("/")
+def user_info():
+    url = "https://traewelling.de/api/v1/auth/user"
+    headers = {'Authorization': request.headers.get('Authorization')}
+    return requests.get(url, headers=headers).json()
+
+@app.route("/self/followers")
+def own_followers():
+    url = "https://traewelling.de/api/v1/user/self/followers"
+    headers = {'Authorization': request.headers.get('Authorization')}
+    req = requests.get(url, headers=headers).json()
+    data = req['data']
+
+    while req['links']['next'] is not None:
+        url = req['links']['next']
+        req = requests.get(url, headers=headers).json()
+        data += req['data']
+    return data
+
+@app.route("/self/followings")
+def own_followings():
+    url = "https://traewelling.de/api/v1/user/self/followings"
+    headers = {'Authorization': request.headers.get('Authorization')}
+    req = requests.get(url, headers=headers).json()
+    data = req['data']
+
+    while req['links']['next'] is not None:
+        url = req['links']['next']
+        req = requests.get(url, headers=headers).json()
+        data += req['data']
+    return data
+
+@app.route("/self/stats")
+def basic_stats():
+    try:
+        detailed = bool(request.args.get('detailed'))
+    except KeyError:
+        detailed = False
+
+    try:
+        poly = bool(request.args.get('polylines'))
+    except KeyError:
+        poly = False
+
+    try:
+        no_days = int(request.args.get('days'))
+        print(no_days)
+    except TypeError:
+        no_days = -1
+
+    if no_days < 0:
+        url = f"https://traewelling.de/api/v1/statistics?from={request.args.get('from')}&until={request.args.get('to')}"
+    else:
+        url = f"https://traewelling.de/api/v1/statistics?from={((date.today() - timedelta(days=no_days)) if no_days>0 else date.today()).strftime('%Y%m%dT%H%M%SZ')}&until={(date.today() + timedelta(days=1)).strftime('%Y%m%dT%H%M%SZ')}"
+
+    headers = {'Authorization': request.headers.get('Authorization')}
+    req = requests.get(url, headers=headers).json()
+    data = req['data']
+
+    if not detailed:
+        return json.dumps(data, ensure_ascii=False)
+    else:
+        detailed_data = []
+        total_distance = 0
+        total_duration = 0
+        total_points = 0
+        for day in data['time']:
+            if day['count'] > 0 or day['date'] == date.today().strftime('%Y-%m-%d'):
+                url = f"https://traewelling.de/api/v1/statistics/daily/{day['date']}?withPolylines={'true' if poly else 'false'}"
+                req = requests.get(url, headers=headers).json()
+                total_distance += req['data']['totalDistance']
+                total_duration += req['data']['totalDuration']
+                total_points += req['data']['totalPoints']
+
+                for status in req['data']['statuses']:
+                    tags = {}
+                    for tag in status['tags']:
+                        tags[str(tag['key']).replace("trwl:", "")] = tag['value']
+                    status['tags'] = tags
+                    try:
+                        status['train']['origin']['departureDelay'] = (datetime.fromisoformat(str(status['train']['origin']['departure'])) - datetime.fromisoformat(str(status['train']['origin']['departurePlanned']))).total_seconds()
+                        status['train']['destination']['arrivalDelay'] = (datetime.fromisoformat(str(status['train']['origin']['arrival'])) - datetime.fromisoformat(str(status['train']['origin']['arrivalPlanned']))).total_seconds()
+                    except KeyError:
+                        continue
+                    detailed_data.append(status)
+        ret_data = {'data': detailed_data, 'totalDistance': total_distance, 'totalDuration': total_duration, 'totalPoints': total_points}
+        return json.dumps(ret_data, ensure_ascii=False)

run.txt

@@ -0,0 +1 @@
+gunicorn -w 4 -b 127.0.0.1:5698 'main:app'