Rename server folder to data, server\data to data\old-data

data/api.py

@@ -0,0 +1,128 @@
+"""
+api.py
+
+Provides a accessible protected backend API. JSON I/O only, CSRF protected.
+"""
+import copy
+import json
+import os
+from copy import deepcopy
+
+# from flask_caching import cache
+import flask_wtf
+from flask import current_app, jsonify, request, send_from_directory
+
+from server.helpers import default, get_neighbors
+
+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)
+
+with open(os.path.join(BASE_DIR, 'data', 'characters.json'), 'r', encoding='utf-8') as file:
+    character_data = json.load(file)
+
+# Cached preload character data
+character_list = dict()
+
+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)
+
+
+@current_app.route('/api/quote_surround')
+def api_quote_neighbors():
+    season, episode = int(request.args.get('season')), int(request.args.get('episode'))
+    scene, quote = int(request.args.get('scene')), int(request.args.get('quote'))
+
+    quotes = data[season - 1]['episodes'][episode - 1]['scenes'][scene - 1]['quotes']
+    top, below = get_neighbors(quotes, quote - 1, int(request.args.get('distance', 2)))
+    return jsonify({'above': top, 'below': below})
+
+
+@current_app.route('/api/characters/')
+def api_character_list():
+    _data = copy.deepcopy(character_data)
+    for key in _data.keys():
+        del _data[key]['quotes']
+    return jsonify(_data)
+
+
+@current_app.route('/api/character/<character>/')
+def api_character_all(character: str):
+    _data = copy.deepcopy(character_data[character])
+    _data['quotes'] = _data['quotes'][:10]
+    return jsonify(_data)
+
+
+@current_app.route('/api/character/<character>/quotes/')
+def api_character_quotes(character: str):
+    quotes = character_data[character]['quotes']
+
+    # Compute pagination if argument is available. Static 10 results per page, one-indexed.
+    if 'page' in request.args.keys():
+        index: int = (int(request.args['page']) - 1) * 10
+        return jsonify(quotes[index: index + 10])
+    else:
+        return jsonify(quotes)
+
+
+@current_app.route('/static/img/<path:filename>')
+def custom_static(filename):
+    return send_from_directory('./data/img/', filename)