fix import error if using setup.py install

Author: thurask

soccer/jsonhandler.py

@@ -0,0 +1,15 @@
+import os
+import json
+
+def load_json(file):
+    """Load JSON file at app start"""
+    here = os.path.dirname(os.path.abspath(__file__))
+    with open(os.path.join(here, file)) as jfile:
+        data = json.load(jfile)
+    return data
+
+TEAM_DATA = load_json("teams.json")["teams"]
+TEAM_NAMES = {team["code"]: team["id"] for team in TEAM_DATA}
+LEAGUE_DATA = load_json("leagues.json")["leagues"]
+LEAGUE_IDS = {league["code"]: league["id"] for league in LEAGUE_DATA}
+LEAGUE_PROPERTIES = {league["code"]: league["properties"] for league in LEAGUE_DATA if league["properties"] != "null"}

soccer/main.py

@@ -2,31 +2,17 @@
 import os
 import requests
 import sys
-import json
 
 
 from soccer.exceptions import IncorrectParametersException, APIErrorException
 from soccer.writers import get_writer
+from soccer.jsonhandler import TEAM_NAMES, LEAGUE_IDS, LEAGUE_PROPERTIES
 
 
 BASE_URL = 'http://api.football-data.org/alpha/'
 LIVE_URL = 'http://soccer-cli.appspot.com/'
 
 
-def load_json(file):
-    """Load JSON file at app start"""
-    here = os.path.dirname(os.path.abspath(__file__))
-    with open(os.path.join(here, file)) as jfile:
-        data = json.load(jfile)
-    return data
-
-
-TEAM_DATA = load_json("teams.json")["teams"]
-TEAM_NAMES = {team["code"]: team["id"] for team in TEAM_DATA}
-LEAGUE_DATA = load_json("leagues.json")["leagues"]
-LEAGUE_IDS = {league["code"]: league["id"] for league in LEAGUE_DATA}
-LEAGUE_PROPERTIES = {league["code"]: league["properties"] for league in LEAGUE_DATA if league["properties"] != "null"}
-
 
 def get_input_key():
     """Input API key and validate"""

soccer/writers.py

@@ -8,7 +8,7 @@
 from itertools import groupby
 from collections import namedtuple
 
-from soccer.main import LEAGUE_IDS, LEAGUE_PROPERTIES
+from soccer.jsonhandler import LEAGUE_IDS, LEAGUE_PROPERTIES
 
 
 def get_writer(output_format='stdout', output_file=None):