Merge pull request #11 from Project-Ticketaka/feature/operation

Modify: Timezone Seoul 추가

Author: yn15

info_performance/controller.py

@@ -8,6 +8,7 @@
 from prf_dao import *
 from prf_session_data import *
 from datetime import datetime
+from pytz import timezone
 import logging
 
 
@@ -43,8 +44,8 @@
     if getData != None:
         prf_data.append(getData)
 
-hol_list = HolCaller().get_holiday(datetime.now().year)
-hol_list += HolCaller().get_holiday(datetime.now().year+1)
+hol_list = HolCaller().get_holiday(datetime.now(timezone('Asia/Seoul')).year)
+hol_list += HolCaller().get_holiday(datetime.now(timezone('Asia/Seoul')).year+1)
 
 session_list = []
 for prf in prf_data:

info_performance/rank_api_caller.py

@@ -4,13 +4,14 @@
 import requests
 from bs4 import BeautifulSoup
 from datetime import datetime, timedelta
+from pytz import timezone
 
 class RankCaller:
     def __init__(self):
-        self.service_key = os.environ['KOPIS_SERVICE_KEY']
+        self.service_key = '5d8e9c530c12433397b94fc06931bed2' # os.environ['KOPIS_SERVICE_KEY']
 
     def get_rank(self):
-        yesterday = (datetime.now().date() - timedelta(1)).strftime("%Y%m%d")
+        yesterday = (datetime.now(timezone('Asia/Seoul')).date() - timedelta(1)).strftime("%Y%m%d")
         url = "http://kopis.or.kr/openApi/restful/boxoffice"
         params = {
             'service': self.service_key,
@@ -21,7 +22,6 @@ def get_rank(self):
         response = requests.get(url, params=params).text
         xmlobj = BeautifulSoup(response, 'lxml-xml').find_all("boxof")
         list = []
-
         for i in range(0, 10):
             data = {}
             data['performance_id'] = xmlobj[i].find("mt20id").string

info_performance/rank_dao.py

@@ -3,7 +3,7 @@
 from pymysql import *
 from pymysql import cursors
 from datetime import datetime, timedelta
-
+from pytz import timezone
 class RankDAO:
     def __init__(self):
         self.conn = connect(
@@ -33,7 +33,7 @@ def insert_rank_data(self, data):
     def delete_rank_data(self):
         try:
             sql = ("DELETE FROM prf_rank WHERE basedate <= %s;")
-            self.curs.execute(sql, (datetime.now().date() - timedelta(3)).strftime("%Y%m%d"))
+            self.curs.execute(sql, (datetime.now(timezone('Asia/Seoul')).date() - timedelta(3)).strftime("%Y%m%d"))
             self.conn.commit()
         except Exception as e:
             print("Exception occured:{}".format(e))