From 307b06d59ac15b842b9526969db25f8d77bb6593 Mon Sep 17 00:00:00 2001 From: Sebastian Serfling Date: Thu, 12 Oct 2023 12:57:51 +0200 Subject: [PATCH] Remove API key From Console --- Classes/main_classes.py | 10 ++++++++++ main.py | 8 ++++++++ 2 files changed, 18 insertions(+) diff --git a/Classes/main_classes.py b/Classes/main_classes.py index 104778e..bbe399f 100644 --- a/Classes/main_classes.py +++ b/Classes/main_classes.py @@ -101,6 +101,16 @@ class CL_GET_HOSTNAME(BaseModel): hostname: str dbname: str +class CL_SET_HOSTNAME(BaseModel): + hostname: str + dbname: str + ram: int + user: str + hddfree: int + hddused: int + hddtotal: int + ipadresse: str + class CL_ADD_CLIENT(BaseModel): query: str database: str diff --git a/main.py b/main.py index 9b2085c..7a00e0b 100644 --- a/main.py +++ b/main.py @@ -94,6 +94,14 @@ async def gethostname(get: CL.CL_GET_HOSTNAME,api_key: APIKey = Depends(auth.get else: return result[0][0] +@app.post("/info/sethostname") +async def sethostname(get: CL.CL_SET_HOSTNAME,api_key: APIKey = Depends(auth.get_api_key)): + result = database(f"INSERT INTO `{get.dbname}`.hardware (hostname,ipadresse,user,ram,hddused,hddfree,hddtotal) VALUES ({get.hostname},{get.ipadresse},{get.user},{get.ram}),{get.hddused},{get.hddfree},{get.hddtotal})","","") + if not result: + return False + else: + return result[0][0] + @app.post("/info/getclient") async def getclient(get: CL.CL_GET_CLIENT,api_key: APIKey = Depends(auth.get_api_key)): result = database(f"SELECT Firma FROM `Kunden-Clients` where `Client-Name` = '{get.name}'","Stines-GmbH","")