diff --git a/Classes/main_classes.py b/Classes/main_classes.py index f90c4f8..e6fb27d 100644 --- a/Classes/main_classes.py +++ b/Classes/main_classes.py @@ -64,3 +64,6 @@ class CL_GET_DATABASE(BaseModel): class CL_GET_VOLUMES(BaseModel): name: str database: str + +class CL_GET_CLIENT(BaseModel): + name: str diff --git a/main.py b/main.py index 548d047..e1c09d8 100644 --- a/main.py +++ b/main.py @@ -72,9 +72,17 @@ async def database_get(get: CL.CL_Database_get,api_key: APIKey = Depends(auth.ge return False @app.post("/info/getvolume") -async def getvolume(get: CL.CL_GET_VOLUMES,api_key: APIKey = Depends(auth.get_api_key)): +async def getvolume(get: CL.CL_GET_CLIENT,api_key: APIKey = Depends(auth.get_api_key)): return database(f"SELECT * FROM `Clients` where `Name` = '{get.name}'",f"{get.database}","") +@app.post("info/getclient") +async def getclient(get: CL.CL_GET_VOLUMES,api_key: APIKey = Depends(auth.get_api_key)): + result = database(f"SELECT Firma FROM `Kunden-Clients` where `Name` = '{get.name}'","Stines-GmbH","") + if not result: + return False + else: + return result + @app.post("/info/getdatabase") async def getdatabase(get: CL.CL_GET_DATABASE,api_key: APIKey = Depends(auth.get_api_key)): return database(f"SELECT name FROM `Kunden-Server` where `IP-Adresse` = '{get.ip}'","Stines-GmbH","")[0][0]