new SearchResult component, using proper table structure, click functionality for expanding to nearby quotes in scene

server/.flaskenv

@@ -1,3 +0,0 @@
-FLASK_APP=server.create_app
-FLASK_ENV=development
-FLASK_DEBUG=1

server/api.py

@@ -8,9 +8,9 @@ import os
 from copy import deepcopy
 
 import flask_wtf
-from flask import current_app, jsonify
+from flask import current_app, jsonify, request
 
-from server.helpers import default
+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:
@@ -25,7 +25,6 @@ stats = {
     }
 }
 
-
 stats['totals']['season'] += len(default(data, []))
 for season in data:
     stats['totals']['episode'] += len(default(season.get('episodes'), []))
@@ -77,3 +76,13 @@ 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})

server/create_app.py

@@ -11,7 +11,7 @@ from flask_wtf.csrf import CSRFProtect
 from server.config import configs
 
 csrf = CSRFProtect()
-cors = CORS(resources={r'/*': {'origins': '*'}})
+cors = CORS(resources={r'/api/*': {'origins': '*'}})
 
 
 def create_app(env=None):

server/helpers.py

@@ -1,9 +1,32 @@
+"""
+helpers.py
+
+
+"""
+
+from typing import List, Tuple
+
 episode_counts = [6, 22, 23, 14, 26, 24, 24, 24, 23]
 
 
 def check_validity(season: int, episode: int):
+    """Shorthand function for checking if a specific episode is valid."""
     return (1 <= season <= 9) and (1 <= episode <= episode_counts[season])
 
 
 def default(value, other):
+    """Value default, similar to dict.get, but better."""
     return value if value is not None else other
+
+
+def get_neighbors(array: List, index: int, distance: int = 2) -> Tuple[List, List]:
+    """Get neighbors above and below a specific index in an array. Returns maximum number of items possible."""
+    top, below = [], []
+    for i in range(1, distance + 1):
+        top_index = index - i
+        below_index = index + i
+        if top_index >= 0:
+            top.append(array[top_index])
+        if below_index < len(array):
+            below.append(array[below_index])
+    return top[::-1], below[::-1]