From 1d9d1d27ad9caf69fd0197902499df42a40f538a Mon Sep 17 00:00:00 2001
From: Xevion <xevioni@yandex.com>
Date: Mon, 14 Sep 2020 12:15:08 -0500
Subject: [PATCH] character name separation and character.json saving, refactor
 character ID generation to helpers

---
 server/cli.py     | 14 +++++++++++---
 server/helpers.py |  7 +++++++
 server/process.py |  9 +--------
 3 files changed, 19 insertions(+), 11 deletions(-)

diff --git a/server/cli.py b/server/cli.py
index 5fcc962..31aaf6a 100644
--- a/server/cli.py
+++ b/server/cli.py
@@ -9,6 +9,7 @@ import re
 import sys
 import time
 from collections import defaultdict
+from pprint import pprint
 from typing import List, Optional, Tuple, Union
 
 import click
@@ -16,9 +17,8 @@ import enlighten
 import requests
 from bs4 import BeautifulSoup
 
-from server.helpers import algolia_transform
-
 sys.path[0] += '\\..'
+from server.helpers import algolia_transform, character_id
 from server.process import DATA_DIR, get_appearances, get_episodes, get_filepath, load_file, \
     save_file, sleep_from, \
     verify_episode
@@ -389,7 +389,15 @@ def character():
     data = load_file(os.path.join(DATA_DIR, 'algolia.json'), True)
     key_list = [('speaker',), ('text',), ('season',), ('episode_rel', 'episode'), ('section_rel', 'scene'),
                 ('quote_rel', 'quote')]
-    master = map(lambda item: algolia_transform(item, key_list), filter(lambda: True, data))
+    master = map(lambda item: algolia_transform(item, key_list), filter(lambda item: True, data))
+
+    # Separate the quotes based on speaker
+    char_data = defaultdict(list)
+    for quote in master:
+        char_data[character_id(quote['speaker'])].append(quote)
+
+    # Save to characters.json
+    save_file(os.path.join(DATA_DIR, 'characters.json'), char_data, True)
 
 
 @build.command('final')
diff --git a/server/helpers.py b/server/helpers.py
index cda86f7..0b6109f 100644
--- a/server/helpers.py
+++ b/server/helpers.py
@@ -50,3 +50,10 @@ def algolia_transform(old_dictionary: dict, key_list: List[Tuple[str, Optional[s
             new_dictionary[keyItem[0]] = old_dictionary[keyItem[0]]
 
     return new_dictionary
+
+
+def is_main_character(name: str) -> bool:
+    return None
+
+def character_id(name: str) -> str:
+    return '-'.join(name.split(' ')).lower()
diff --git a/server/process.py b/server/process.py
index 96e4e9f..d417c1f 100644
--- a/server/process.py
+++ b/server/process.py
@@ -117,13 +117,6 @@ def get_appearances(season, episode) -> Optional[List[Dict[str, Union[int, str]]
     for scene in scenes:
         for quote in scene.get('quotes', []):
             characters[quote.get('speaker')] += 1
-    characters = [{'name': character, 'appearances': appearances, 'id': '-'.join(character.split(' ')).lower()}
+    characters = [{'name': character, 'appearances': appearances, 'id': character_id(character)}
                   for character, appearances in characters.items()]
     return list(sorted(characters, key=lambda item: item['appearances'], reverse=True))
-
-def get_character(character):
-    """
-    Retrieves a character's appearances from every season and episode available.
-    :param character: The character's name
-    :return: A list of dictionary quotes including reference IDs
-    """