temp push

MisileLab · Oct 31, 2023 · 442b487 · 442b487
1 parent 4015426
commit 442b487
Show file tree

Hide file tree

Showing 4 changed files with 173 additions and 3 deletions.
diff --git a/projects/xobusy/sunrint-decon/.gitignore b/projects/xobusy/sunrint-decon/.gitignore
@@ -1,2 +1,2 @@
-MONGO_URI
+config.toml
 
diff --git a/projects/xobusy/sunrint-decon/main.py b/projects/xobusy/sunrint-decon/main.py
@@ -3,9 +3,12 @@
 
 from fastapi import FastAPI, Header, status, HTTPException
 from pymongo import MongoClient
+from tomli import loads
+from requests import get
 
+config = loads(Path("config.toml").read_text())
 app = FastAPI()
-db = MongoClient(Path("MONGO_URI").read_text().strip())["local"]["schoolfinder"]
+db = MongoClient(config["MONGO_URI"])["schoolfinder"]["login"]
 
 async def is_valid(name: str, password: str) -> bool:
     return db.find_one({"name":name,"password":password}) != None
@@ -23,3 +26,9 @@ async def register(
 
     db.insert_one({"name":name,"password":password})
 
+@app.get("/school/search")
+async def school_search(name: str):
+    a = get("https://open.neis.go.kr/hub/schoolinfo",params={"KEY":config["NEIS_KEY"],"TYPE":"json","SCHUL_NM":name})
+    a.raise_for_status()
+    return a.json()["schoolInfo"][2]["row"]
+
diff --git a/projects/xobusy/sunrint-decon/poetry.lock b/projects/xobusy/sunrint-decon/poetry.lock
diff --git a/projects/xobusy/sunrint-decon/pyproject.toml b/projects/xobusy/sunrint-decon/pyproject.toml
@@ -11,6 +11,8 @@ python = "3.12.0"
 pymongo = "4.5.0"
 fastapi = "0.104.1"
 uvicorn = "0.23.2"
+tomli = "2.0.1"
+requests = "2.31.0"
 
 [build-system]
 requires = ["poetry-core"]