Merge pull request #33 from SoyBeansLab/feature/contest-domain

feat: 存在しないcontest_idでリクエストしたときただのnullが返ってたのをjsonの中に入れるようにした
SoyBeansLab · Aug 19, 2019 · 998b369 · 998b369
2 parents a75fcb5 + 1aae922
commit 998b369
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/domain/Contest/database/contest_repository.py b/src/domain/Contest/database/contest_repository.py
@@ -19,4 +19,6 @@ def find(self, contest_id: str) -> Contest:
         row = self.sqlhandler.query(
             "SELECT * FROM contests WHERE contest_id=%s", (contest_id,)
         ).fetch_one()
+        if len(row) == 0:
+            return None
         return Contest(*row)
diff --git a/src/interface/controllers/contest_controller.py b/src/interface/controllers/contest_controller.py
@@ -16,5 +16,11 @@ async def contests(self, req, resp):
 
     async def contest(self, req, resp, *, contest_id):
         contest = self.interactor.contest(contest_id)
-        resp.media = {"contest": contest.as_json()}
-        resp.status_code = 200
+        if contest is None:
+            res_data = None
+            res_code = 400
+        else:
+            res_data = contest.as_json()
+            res_code = 200
+        resp.media = {"contest": res_data}
+        resp.status_code = res_code