From 2e2144cfa15d0c1cd864f729c59bb65bc592808f Mon Sep 17 00:00:00 2001
From: Sebastian Serfling <sebastian.serfling@stines.de>
Date: Wed, 25 Oct 2023 10:39:36 +0200
Subject: [PATCH] Remove API key From Console

---
 Classes/main_classes.py | 5 +++++
 main.py                 | 6 +++---
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/Classes/main_classes.py b/Classes/main_classes.py
index 851ad7f..8c2fd02 100644
--- a/Classes/main_classes.py
+++ b/Classes/main_classes.py
@@ -112,6 +112,11 @@ class CL_GET_HOST_USER(BaseModel):
 
 class CL_SET_HOST(BaseModel):
     hostname: str
+    user: str
+    ipadresse: str
+    winver: str
+    cpuname: str
+    cores: str
     dbname: str
     ram: int
     hddfree: str
diff --git a/main.py b/main.py
index 695adfb..4df1bd1 100644
--- a/main.py
+++ b/main.py
@@ -88,7 +88,7 @@ async def getuser(get: CL.CL_GET_USER,api_key: APIKey = Depends(auth.get_api_key
 
 @app.post("/host/get")
 async def gethost(get: CL.CL_GET_HOST,api_key: APIKey = Depends(auth.get_api_key)):
-    result = database(f"SELECT id FROM `clients` where `hostname` = '{get.hostname}'",f"{get.dbname}","")
+    result = database(f"SELECT id FROM `Clients` where `hostname` = '{get.hostname}'",f"{get.dbname}","")
     if not result:
         return False
     else:
@@ -115,9 +115,9 @@ async def getdatabase(get: CL.CL_GET_DATABASE,api_key: APIKey = Depends(auth.get
 
 @app.post("/host/set")
 async def sethost(get: CL.CL_SET_HOST,api_key: APIKey = Depends(auth.get_api_key)):
-    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}')","","")
+    database(f"INSERT INTO `{get.dbname}`.Clients (Name,Username,IP-Adresse,Windows-Version,CPU-Name,Prozessor-Anzahl,RAM,Speichergesamt,Speicherfrei,Speicherbelegt) VALUES ('{get.hostname}','{get.user}','{get.ipadresse}','{get.winver}','{get.cpuname}','{get.cores}','{get.ram}','{get.hddtotal}','{get.hddfree}','{get.hddused}')","","")
     result = database(
-        f"SELECT id FROM `{get.dbname}`.hardware WHERE hostname = '{get.hostname}')",
+        f"SELECT id FROM `{get.dbname}`.Clients WHERE hostname = '{get.hostname}')",
         "", "")
     if not result:
         return False