Skip to content

Commit cbe0ec8

Browse files
committed
make logs directory if not exists
1 parent 1372fc5 commit cbe0ec8

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

main.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import logging
22
import os
3+
import os.path
34
import settings
45
import data_manager
56
from policy_learner import PolicyLearner
@@ -11,13 +12,17 @@
1112
# 로그 기록
1213
log_dir = os.path.join(settings.BASE_DIR, 'logs/%s' % stock_code)
1314
timestr = settings.get_time_str()
15+
if not os.path.exists('logs'):
16+
os.makedirs('logs')
17+
if not os.path.exists('logs/%s' % stock_code):
18+
os.makedirs('logs/%s' % stock_code)
1419
file_handler = logging.FileHandler(filename=os.path.join(
1520
log_dir, "%s_%s.log" % (stock_code, timestr)), encoding='utf-8')
1621
stream_handler = logging.StreamHandler()
1722
file_handler.setLevel(logging.DEBUG)
1823
stream_handler.setLevel(logging.INFO)
1924
logging.basicConfig(format="%(message)s",
20-
handlers=[file_handler, stream_handler], level=logging.DEBUG)
25+
handlers=[file_handler, stream_handler], level=logging.DEBUG)
2126

2227
# 주식 데이터 준비
2328
chart_data = data_manager.load_chart_data(

0 commit comments

Comments
 (0)