diff --git a/Classes/main_classes.py b/Classes/main_classes.py index 49610a8..498178d 100644 --- a/Classes/main_classes.py +++ b/Classes/main_classes.py @@ -119,6 +119,15 @@ class CL_SET_HOST(BaseModel): hddtotal: str ipadresse: str +class CL_UPDATE_HOST(BaseModel): + hostname: str + dbname: str + ram: int + hddfree: str + hddused: str + hddtotal: str + ipadresse: str + class CL_SET_LASTLOGON_USER(BaseModel): userID: int dbname: str diff --git a/main.py b/main.py index 7f51b5c..048bad1 100644 --- a/main.py +++ b/main.py @@ -113,6 +113,17 @@ async def sethost(get: CL.CL_SET_HOST,api_key: APIKey = Depends(auth.get_api_key else: return result[0][0] +@app.post("/host/update") +async def updatehost(get: CL.CL_UPDATE_HOST,api_key: APIKey = Depends(auth.get_api_key)): + database(f"UPDATE `{get.dbname}`.hardware SET ipadresse = '{get.ipadresse}', ram = '{get.ram}', hddused = '{get.hddused}' ,hddfree = '{get.hddfree}',hddtotal = '{get.hddtotal}' WHERE hostname = '{get.hostname}'","","") + result = database( + f"SELECT id FROM `{get.dbname}`.hardware WHERE hostname = '{get.hostname}')", + "", "") + if not result: + return False + else: + return result[0][0] + @app.post("/user/set") async def set_user(get: CL.CL_SET_HOST_USER,api_key: APIKey = Depends(auth.get_api_key)): database(f"INSERT INTO `{get.dbname}`.user (user) VALUES ('{get.user}')","","")