init

stockpredictor/analysis/Analyzer.py

@@ -4,6 +4,7 @@ import time
 import shutil
 from stockpredictor.analysis.Common import Common
 from stockpredictor.analysis.Stochastic import Stochastic
+from stockpredictor.analysis.BolingerBand import BolingerBand
 import matplotlib.pyplot as plt
 import datetime
 import sqlite3
@@ -45,6 +46,7 @@ class Analyzer:
         self.common = Common()
 
         self.stochastic = Stochastic()
+        self.bolingerBand = BolingerBand()
 
         self.readFnguide()
         return
@@ -459,7 +461,7 @@ class Analyzer:
         cursor = conn.cursor()
 
         # 기존 분석 데이터를 모두 지움
-        cursor.execute('update ' + self.tableName + ' set STOCHASTIC = ""')
+        cursor.execute('update ' + self.tableName + ' set STOCHASTIC = "", BOLINGERBAND = ""')
 
         rowid = 1
         cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
@@ -470,6 +472,9 @@ class Analyzer:
             try:
                 results_STOCHASTIC = self.stochastic.analyze(stock)
                 text_STOCHASTIC = json.dumps(results_STOCHASTIC, ensure_ascii=False)
+
+                results_BolingerBand = self.bolingerBand.analyze(stock)
+                text_BOLINGERBAND = json.dumps(results_BolingerBand, ensure_ascii=False)
             except:
                 print("#", rowid, stock['NAME'])
                 rowid += 1
@@ -477,7 +482,7 @@ class Analyzer:
                 result = cursor.fetchone()
                 continue
 
-            cursor.execute("UPDATE " + self.tableName + " SET STOCHASTIC=? WHERE CODE=?", (text_STOCHASTIC, stock["CODE"]))
+            cursor.execute("UPDATE " + self.tableName + " SET STOCHASTIC=?, BOLINGERBAND=? WHERE CODE=?", (text_STOCHASTIC, text_BOLINGERBAND, stock["CODE"]))
             print("#", rowid, stock['NAME'])
             rowid += 1
             cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
@@ -503,7 +508,7 @@ if __name__ == "__main__":
     analyzer.analyzeStochastic()
     """
 
-    ###analyzer.analyze()
+    analyzer.analyze()
 
     day = datetime.today().strftime("%Y%m%d")