diff --git a/OJ-BE/aicontest/views/oj.py b/OJ-BE/aicontest/views/oj.py index 1dd2016..69d6073 100644 --- a/OJ-BE/aicontest/views/oj.py +++ b/OJ-BE/aicontest/views/oj.py @@ -194,8 +194,10 @@ def post(self, request): else: return self.error("Upload failed") logger.info("file_name={}".format(file.name)) + if not file.name.endswith('.csv'): raise APIError("CSV파일이 아닙니다. 파일 형식을 다시 확인해주세요") + logger.info("file={}".format(file)) tmp_file = f"/tmp/{rand_str()}.csv" logger.info("tmp_file={}".format(tmp_file)) @@ -226,7 +228,8 @@ def post(self, request): y_pred = np.array(np.loadtxt(os.path.join(settings.PREDICT_DIR, predict_id, "predict.csv"), delimiter=",", dtype=np.float32, skiprows=1, usecols = (1,))) logger.info("y_true={}".format(str(y_true))) logger.info("y_pred={}".format(str(y_pred))) - if y_true.shape[0]!=y_pred.shape[0]: + + if y_true.shape[0] != y_pred.shape[0]: raise APIError("해당 문제의 csv파일이 아닙니다. 파일 내용을 다시 확인해주세요") eval_type = csv.eval_type diff --git a/OJ-FE/src/pages/oj/components/ModalNotify.vue b/OJ-FE/src/pages/oj/components/ModalNotify.vue new file mode 100644 index 0000000..2b70980 --- /dev/null +++ b/OJ-FE/src/pages/oj/components/ModalNotify.vue @@ -0,0 +1,106 @@ + + + + + \ No newline at end of file diff --git a/OJ-FE/src/pages/oj/views/aicontest/AIContest.vue b/OJ-FE/src/pages/oj/views/aicontest/AIContest.vue index fd40c5f..f7a59b1 100644 --- a/OJ-FE/src/pages/oj/views/aicontest/AIContest.vue +++ b/OJ-FE/src/pages/oj/views/aicontest/AIContest.vue @@ -64,6 +64,7 @@ :data="{id: problem._id}" :show-file-list="true" :on-success="uploadFileSucceeded" + ref="inputFile" accept=".csv" style="margin: 3px">
@@ -71,6 +72,7 @@

한 개의 파일(csv)만 가능합니다.

+