diff --git a/Classes/main_classes.py b/Classes/main_classes.py index 1ae2267..b46b849 100644 --- a/Classes/main_classes.py +++ b/Classes/main_classes.py @@ -74,3 +74,7 @@ class CL_GET_CLIENT(BaseModel): class CL_ADD_CLIENT(BaseModel): query: str database: str + +class CL_GET_USER(BaseModel): + query: str + database: str diff --git a/main.py b/main.py index d520927..9249a98 100644 --- a/main.py +++ b/main.py @@ -78,15 +78,12 @@ async def getvolume(get: CL.CL_GET_VOLUMES,api_key: APIKey = Depends(auth.get_ap return volume @app.post("/info/getuser") -async def getvolume(get: CL.CL_GET_VOLUMES,api_key: APIKey = Depends(auth.get_api_key)): - result = database(f"SELECT disabled FROM `Active-Directory-RDS-User` where `SamAccountName` = '{get.name}'",f"{get.database}","") - volume = [] +async def getuser(get: CL.CL_GET_USER,api_key: APIKey = Depends(auth.get_api_key)): + result = database(f"SELECT `disabled` FROM `Active-Directory-RDS-User` where `SamAccountName` = '{get.name}'",f"{get.database}","") if not result: return False else: - for i in result: - volume.append(i) - return volume + return result[0][0] @app.post("/info/getclient")