Upload 2 files

Logic/FUsers/AddUser.py

@@ -1,67 +1,73 @@
-from fastapi import APIRouter,File,Form,UploadFile
-import sqlite3
-from passlib.context import CryptContext
-from os import makedirs
-from os.path import exists
-from PIL import Image
-from io import BytesIO
-UserAddRouter=APIRouter(prefix="/Users")
-
-
-
-@UserAddRouter.post("/AddUser")
-def adduser(AdminEmail:str=Form(...),AdminPassword:str =Form(...),Email:str=Form(...),Name:str=Form(...),Job:str=Form(...),Phonenumber:str=Form(...),Place:str=Form(...),BDay:str=Form(...),Password:str=Form(...),Images:UploadFile=File(...)):
-    # try:
-    pwd_context=CryptContext(schemes=["bcrypt"],deprecated="auto")
-    AdminPassword="$2b$12$33H14GFBvtomzLrEaM6Tqu7Jr621eHlXL9TVtdPU6YCljxc4eug2a"
-    if  AdminEmail !="Mhammed@Admin.com":
-        if not pwd_context.verify(  AdminPassword,AdminPassword):
-            return {"Status":False,"Message":" Admin Email or Password is not correct "}
-    pwd_context=CryptContext(schemes=["bcrypt"],deprecated="auto")
-    UserId=0
-    connect=sqlite3.connect("DataBase/DataBase.bd")
-    cursor=connect.execute(f'''
-                    SELECT UserId FROM Users where Email='{ Email}'
-                    ''')
-    Data=cursor.fetchall()
-    if len(Data)==1:
-        if (Data[0]!=None):
-            return  {"Status":False,"Message":"Email Is Alread Exists"}
-    cursor=connect.execute('''
-                    SELECT MAX(UserId) FROM Users
-                    ''')
-    Data=cursor.fetchall()
-    
-    if len(Data)!=0:
-        if Data[0][0]==None:
-            UserId=0
-        else:
-            UserId=Data[0][0]+1
-    
-    HashedPassword=pwd_context.hash( Password)
-            
-    connect.execute(f'''
-                    INSERT INTO Users ( UserId , Email , UserName , Job, Phonenumber , Place , BDay , Password ) VALUES ({UserId},'{Email}','{Name}','{Job}', '{Phonenumber}' ,'{Place}','{ BDay}','{HashedPassword}')
-                    ''')
-    connect.commit()
-    connect.close() 
-    if (not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Train")):
-        makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Train")
-    if (not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/Negative")):
-        makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/Negative")
-    if (not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Test")):
-        makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Test")
-    if (not exists(f"./FaceRecognition/FaceModel/{UserId}")):
-        makedirs(f"./FaceRecognition/FaceModel/{UserId}")
-    if (not exists(f"./static/{UserId}")):
-        makedirs(f"./static/{UserId}")
-        Image.open(BytesIO(Images.file.read())).save(f"./static/{UserId}/user.png","PNG")
-    if (not exists(f"./IndoorLocalization/Data/{UserId}")):
-        makedirs(f"./IndoorLocalization/Data/{UserId}")
-    if (not exists(f"./IndoorLocalization/IndoorModels/{UserId}")):
-        makedirs(f"./IndoorLocalization/IndoorModels/{UserId}")
-        
-    return {"Status":True,"Message":"User Is Added Correctly"}
-
-    # except sqlite3.Error as e :
-    #     return {"Status":False,"Message":e}
+from fastapi import APIRouter,File,Form,UploadFile
+import sqlite3
+from passlib.context import CryptContext
+from os import makedirs,chmod
+from os.path import exists
+from PIL import Image
+from io import BytesIO
+UserAddRouter=APIRouter(prefix="/Users")
+
+
+@UserAddRouter.post("/AddUser")
+def adduser(AdminEmail:str=Form(...),AdminPassword:str =Form(...),Email:str=Form(...),Name:str=Form(...),Job:str=Form(...),Phonenumber:str=Form(...),Place:str=Form(...),BDay:str=Form(...),Password:str=Form(...),Images:UploadFile=File(...)):
+    try:
+        pwd_context=CryptContext(schemes=["bcrypt"],deprecated="auto")
+        AdminPassword="$2b$12$33H14GFBvtomzLrEaM6Tqu7Jr621eHlXL9TVtdPU6YCljxc4eug2a"
+        if  AdminEmail !="[email protected]":
+            if not pwd_context.verify(  AdminPassword,AdminPassword):
+                return {"Status":False,"Message":" Admin Email or Password is not correct "}
+        pwd_context=CryptContext(schemes=["bcrypt"],deprecated="auto")
+        UserId=0
+        connect=sqlite3.connect("DataBase/DataBase.bd")
+        cursor=connect.execute(f'''
+                        SELECT UserId FROM Users where Email='{ Email}'
+                        ''')
+        Data=cursor.fetchall()
+        if len(Data)==1:
+            if (Data[0]!=None):
+                return  {"Status":False,"Message":"Email Is Alread Exists"}
+        cursor=connect.execute('''
+                        SELECT MAX(UserId) FROM Users
+                        ''')
+        Data=cursor.fetchall()
+        
+        if len(Data)!=0:
+            if Data[0][0]==None:
+                UserId=0
+            else:
+                UserId=Data[0][0]+1
+        
+        HashedPassword=pwd_context.hash( Password)
+                
+        connect.execute(f'''
+                        INSERT INTO Users (UserId ,Email ,UserName ,Job, Phonenumber ,Place,BDay ,Password ) VALUES ({UserId},'{ Email}','{ Name}','{ Job}', '{ Phonenumber}' ,'{ Place}','{ BDay}','{HashedPassword}')
+                        ''')
+        connect.commit()
+        connect.close() 
+        if (not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Train")):
+            makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Train")
+            chmod("./FaceRecognition/ExtactedFaces/{UserId}/Train",777)
+        if (not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/Negative")):
+            makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/Negative")
+            chmod(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/Negative",777)
+        if (not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Test")):
+            makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Test")
+            chmod(f"./FaceRecognition/ExtactedFaces/{UserId}/Test",777)
+        if (not exists(f"./FaceRecognition/FaceModel/{UserId}")):
+            makedirs(f"./FaceRecognition/FaceModel/{UserId}")
+            chmod(f"./FaceRecognition/FaceModel/{UserId}",777)
+        if (not exists(f"./static/{UserId}")):
+            makedirs(f"./static/{UserId}")
+            chmod(f"./static/{UserId}",777)
+            Image.open(BytesIO(Images.file.read())).save(f"./static/{UserId}/user.png","PNG")
+            chmod(f"./static/{UserId}/user.png",777)
+        if (not exists(f"./IndoorLocalization/Data/{UserId}")):
+            makedirs(f"./IndoorLocalization/Data/{UserId}")
+            chmod(f"./IndoorLocalization/Data/{UserId}",777)
+        if (not exists(f"./IndoorLocalization/IndoorModels/{UserId}")):
+            makedirs(f"./IndoorLocalization/IndoorModels/{UserId}")
+            chmod(f"./IndoorLocalization/IndoorModels/{UserId}",777)
+        return {"Status":True,"Message":"User Is Added Correctly"}
+
+    except Exception as e :
+        return {"Status":False,"Message":e}

Logic/FUsers/AddUserItem.py

@@ -2,7 +2,7 @@ from fastapi import APIRouter,File,Form,UploadFile
 import sqlite3
 
 from os.path import exists
-from os import makedirs
+from os import makedirs,chmod
 
 from PIL import Image
 from io import BytesIO
@@ -40,11 +40,15 @@ def adduser(Email:str=Form(...),Name:str=Form(...),Password:str=Form(...),Images
             connect.commit()
             if not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}"):
                 makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}")
+                chmod(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}",777)
             if not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Test/{Name}"):
                 makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Test/{Name}")
+                chmod(f"./FaceRecognition/ExtactedFaces/{UserId}/Test/{Name}",777)
             if not exists(f"./static/{UserId}/{Name}"):
                 makedirs(f"./static/{UserId}/{Name}")
+                chmod(f"./static/{UserId}/{Name}",777)
                 Image.open(BytesIO(Images.file.read())).save(f"./static/{UserId}/{Name}/user.png","PNG")
+                chmod(f"./static/{UserId}/{Name}/user.png",777)
             return {"Status":True,"Message":"User Is Added Correctly"}
         else:
             return {"Status":False,"Message":"Name Is Already Exist"}