diff --git a/Classes/main_classes.py b/Classes/main_classes.py index 6122f35..6b9ddcf 100644 --- a/Classes/main_classes.py +++ b/Classes/main_classes.py @@ -97,11 +97,20 @@ class CL_GET_VOLUMES(BaseModel): class CL_GET_CLIENT(BaseModel): name: str -class CL_GET_HOSTNAME(BaseModel): +class CL_GET_HOST(BaseModel): hostname: str dbname: str -class CL_SET_HOSTNAME(BaseModel): +class CL_GET_HOST_USER(BaseModel): + hostname: str + dbname: str + +class CL_SET_HOST_USER(BaseModel): + username: str + lastlogon: str + dbname: str + +class CL_SET_HOST(BaseModel): hostname: str dbname: str ram: int diff --git a/main.py b/main.py index e4c679d..b2e7e5f 100644 --- a/main.py +++ b/main.py @@ -87,7 +87,7 @@ async def getuser(get: CL.CL_GET_USER,api_key: APIKey = Depends(auth.get_api_key return result[0][0] @app.post("/info/gethost") -async def gethostname(get: CL.CL_GET_HOSTNAME,api_key: APIKey = Depends(auth.get_api_key)): +async def gethostname(get: CL.CL_GET_HOST,api_key: APIKey = Depends(auth.get_api_key)): result = database(f"SELECT hostname FROM `hardware` where `hostname` = '{get.hostname}'",f"{get.dbname}","") if not result: return False @@ -95,13 +95,21 @@ async def gethostname(get: CL.CL_GET_HOSTNAME,api_key: APIKey = Depends(auth.get return result[0][0] @app.post("/info/sethost") -async def sethostname(get: CL.CL_SET_HOSTNAME,api_key: APIKey = Depends(auth.get_api_key)): +async def sethostname(get: CL.CL_SET_HOST,api_key: APIKey = Depends(auth.get_api_key)): result = database(f"INSERT INTO `{get.dbname}`.hardware (hostname,ipadresse,ram,hddused,hddfree,hddtotal) VALUES ('{get.hostname}','{get.ipadresse}','{get.ram}','{get.hddused}','{get.hddfree}','{get.hddtotal}')","","") if not result: return False else: return result[0][0] +@app.post("/info/gethost_user") +async def gethost_user(get: CL.CL_GET_HOST_USER,api_key: APIKey = Depends(auth.get_api_key)): + result = database(f"INSERT INTO `{get.dbname}`.user (lastlogon,user) VALUES ('{get.lastlogon}','{get.user}')","","") + if not result: + return False + else: + return database(f"SELECT id FROM `{get.dbname}`.user WHERE user = '{get.user}')","","") + @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","")