From ee7e3a86d220bea9c7a662355145b5fffc90438e Mon Sep 17 00:00:00 2001
From: Sebastian Serfling <sebastian.serfling@stines.de>
Date: Mon, 23 Oct 2023 14:45:44 +0200
Subject: [PATCH] Remove API key From Console

---
 Classes/main_classes.py | 1 +
 main.py                 | 8 ++++++--
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/Classes/main_classes.py b/Classes/main_classes.py
index 16b30aa..566a2c3 100644
--- a/Classes/main_classes.py
+++ b/Classes/main_classes.py
@@ -159,4 +159,5 @@ class CL_ADD_GROUP_VOLUME(BaseModel):
     label: str
     path: str
     letter: str
+    group: str
     dbname: str
\ No newline at end of file
diff --git a/main.py b/main.py
index 5e62b66..52f22a7 100644
--- a/main.py
+++ b/main.py
@@ -295,10 +295,14 @@ async def add_ad_user(get: CL.CL_ADD_EX_USER,api_key: APIKey = Depends(auth.get_
 
 @app.post("/add/group/volume")
 async def send_group_volume(get: CL.CL_ADD_GROUP_VOLUME,api_key: APIKey = Depends(auth.get_api_key)):
-    volume_name = database(f"SELECT ID FROM Netvolume WHERE Volume_Name = '{get.label}'", f"{get.dbname}", "")
-    if not volume_name:
+    volume_id = database(f"SELECT ID FROM Netvolume WHERE Volume_Name = '{get.label}'", f"{get.dbname}", "")
+    if not volume_id:
         database(f"INSERT INTO `{get.dbname}`.Netvolume (Volume_Key,Volume_Path,Volume_Name) VALUES ('{get.letter}','{get.path}','{get.label}')", "",
                  "")
+    group_id = database(f"SELECT GruppenID FROM GruppenToNetvolumes WHERE VolumeID = '{volume_id}'", f"{get.dbname}", "")
+    if not group_id:
+        group_id = database(f"SELECT ID FROM `{get.dbname}`.Gruppen WHERE Name = {get.group})", "","")
+        database(f"INSERT INTO `{get.dbname}`.GruppenToNetvolumes (GruppenID,VolumeID) VALUES ('{group_id}','{volume_id}')", "","")
     # database(get.query,get.dbname,get.entry)
     return f"Create {get.label} - {get.path}"