workbook 자동화

Author: seyeon22222

workbook/actions.py

@@ -25,7 +25,6 @@ def get_problem_info(workbook_url):
   ret = []
   while True:
     x = txt.find(pattern)
-    print(x)
     if x == -1: break
     txt = txt[x+9:]
     prob_id, prob_name = '', ''
@@ -64,8 +63,9 @@ def gen_ind_workbook(attrs, category):
         category_idx = category[chapter_idx].index(prob_id)
       file_path = solution_path + prob_id
       if not os.path.exists(file_path + '.java'):
-        with open(file_path + '.java', 'w', encoding="UTF-8") as f:
-          f.write(txt)
+          os.makedirs(os.path.dirname(file_path), exist_ok=True)  # 디렉토리 없으면 생성
+          with open(file_path + '.java', 'w', encoding="UTF-8") as f:
+              f.write(txt)
       try:
         codes = open(file_path + '.java', 'r', encoding="UTF-8").read()
       except: # EUC-KR -> UTF-8