mirror of
https://github.com/Xevion/the-office.git
synced 2025-12-12 12:13:18 -06:00
80 lines
1.9 KiB
Python
80 lines
1.9 KiB
Python
"""
|
|
api.py
|
|
|
|
Provides a accessible protected backend API. JSON I/O only, CSRF protected.
|
|
"""
|
|
import json
|
|
import os
|
|
from copy import deepcopy
|
|
|
|
import flask_wtf
|
|
from flask import current_app, jsonify
|
|
|
|
from server.helpers import default
|
|
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
with open(os.path.join(BASE_DIR, 'data', 'data.json'), 'r', encoding='utf-8') as file:
|
|
data = json.load(file)
|
|
|
|
stats = {
|
|
'totals': {
|
|
'quote': 0,
|
|
'scene': 0,
|
|
'episode': 0,
|
|
'season': 0
|
|
}
|
|
}
|
|
|
|
|
|
stats['totals']['season'] += len(default(data, []))
|
|
for season in data:
|
|
stats['totals']['episode'] += len(default(season.get('episodes'), []))
|
|
for episode in season['episodes']:
|
|
stats['totals']['scene'] += len(default(episode.get('scenes'), []))
|
|
for scene in default(episode.get('scenes'), []):
|
|
stats['totals']['quote'] += len(default(scene.get('quotes'), []))
|
|
|
|
|
|
@current_app.route('/api/csrf/')
|
|
def api_csrf():
|
|
"""
|
|
Page used for refreshing expired CSRF tokens via AJAX.
|
|
|
|
Probably secure: https://medium.com/@iaincollins/csrf-tokens-via-ajax-a885c7305d4a
|
|
"""
|
|
return jsonify(flask_wtf.csrf.generate_csrf())
|
|
|
|
|
|
@current_app.route('/api/episode/<int:season>/<int:episode>/')
|
|
def api_episode(season: int, episode: int):
|
|
return jsonify(data[season - 1]['episodes'][episode - 1])
|
|
|
|
|
|
@current_app.route('/api/stats/')
|
|
def api_stats():
|
|
return jsonify(stats)
|
|
|
|
|
|
@current_app.route('/api/episodes/')
|
|
def api_episodes():
|
|
"""
|
|
Returns a list of episodes with basic information (no quotes).
|
|
Used for the left side season bar.
|
|
"""
|
|
seasons = []
|
|
copy = deepcopy(data)
|
|
for season in copy:
|
|
for episode in season.get('episodes'):
|
|
if 'scenes' in episode.keys():
|
|
del episode['scenes']
|
|
seasons.append(season)
|
|
return jsonify(seasons)
|
|
|
|
|
|
@current_app.route('/api/all/')
|
|
def api_data():
|
|
"""
|
|
Season data route
|
|
"""
|
|
return jsonify(data)
|