Fix compile error

jiajunsu · Sep 1, 2018 · c75d673 · c75d673
1 parent 038c5cd
commit c75d673
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/calculator_of_Onmyoji/convert_json2xls.py b/calculator_of_Onmyoji/convert_json2xls.py
@@ -4,8 +4,8 @@
 import os
 import sys
 
-import load_data
-import write_data
+from calculator_of_Onmyoji import load_data
+from calculator_of_Onmyoji import write_data
 
 
 def get_json_files():

diff --git a/calculator_of_Onmyoji/load_data.py b/calculator_of_Onmyoji/load_data.py
@@ -126,6 +126,8 @@ def sep_mitama_by_loc(mitama_data):
 
     for d_k, d_v in mitama_data.items():
         loc = d_v[u'位置']
+        if loc not in range(1, 6):
+            print('Mitama location must be 1 to 6, please chech you data.')
         mitama_loc_data[loc].append({d_k: d_v})
 
     return mitama_loc_data