Mohammedallyl commited on
Commit
2647087
·
verified ·
1 Parent(s): 39ebd9c

Update Logic/FUsers/AddUserItem.py

Browse files
Files changed (1) hide show
  1. Logic/FUsers/AddUserItem.py +55 -55
Logic/FUsers/AddUserItem.py CHANGED
@@ -1,56 +1,56 @@
1
- from fastapi import APIRouter,File,Form,UploadFile
2
- import sqlite3
3
-
4
- from os.path import exists
5
- from os import makedirs,chmod
6
-
7
- from PIL import Image
8
- from io import BytesIO
9
-
10
- UserItemAddRouter=APIRouter(prefix="/Users")
11
- @UserItemAddRouter.post("/AddUserItem")
12
- def adduser(Email:str=Form(...),Name:str=Form(...),Password:str=Form(...),Images:UploadFile=File(...)):
13
- try:
14
-
15
- State=False
16
- connect=sqlite3.connect("DataBase/DataBase.bd")
17
- cursor=connect.execute(f'''
18
- SELECT UserId,Password FROM Users where Email='{ Email}'
19
- ''')
20
-
21
- FData=cursor.fetchall()
22
-
23
-
24
-
25
- if len(FData) !=0 :
26
- if FData[0][0]==None:
27
- return {"Status":False,"Message":"Email or Password Is Incorrect "}
28
- HasedPassword=FData[0][1]
29
- UserId=FData[0][0]
30
- State= Password==HasedPassword
31
-
32
- Name= Name
33
- if not State:
34
- return {"Status":False,"Message":"Email or Password is not correct"}
35
-
36
- if not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}"):
37
- connect.execute(f'''
38
- INSERT INTO UserItems (UserId ,UserItemName ) VALUES ({UserId},'{Name}')
39
- ''')
40
- connect.commit()
41
- if not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}"):
42
- makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}")
43
- chmod(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}",777)
44
- if not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Test/{Name}"):
45
- makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Test/{Name}")
46
- chmod(f"./FaceRecognition/ExtactedFaces/{UserId}/Test/{Name}",777)
47
- if not exists(f"./static/{UserId}/{Name}"):
48
- makedirs(f"./static/{UserId}/{Name}")
49
- chmod(f"./static/{UserId}/{Name}",777)
50
- Image.open(BytesIO(Images.file.read())).save(f"./static/{UserId}/{Name}/user.png","PNG")
51
- chmod(f"./static/{UserId}/{Name}/user.png",777)
52
- return {"Status":True,"Message":"User Is Added Correctly"}
53
- else:
54
- return {"Status":False,"Message":"Name Is Already Exist"}
55
- except Exception as e:
56
  return {"Status":False,"Message":e}
 
1
+ from fastapi import APIRouter,File,Form,UploadFile
2
+ import sqlite3
3
+
4
+ from os.path import exists
5
+ from os import makedirs,chmod
6
+
7
+ from PIL import Image
8
+ from io import BytesIO
9
+
10
+ UserItemAddRouter=APIRouter(prefix="/Users")
11
+ @UserItemAddRouter.post("/AddUserItem")
12
+ def adduser(Email:str=Form(...),Name:str=Form(...),Password:str=Form(...),Images:UploadFile=File(...)):
13
+ try:
14
+
15
+ State=False
16
+ connect=sqlite3.connect("DataBase/DataBase.bd")
17
+ cursor=connect.execute(f'''
18
+ SELECT UserId,Password FROM Users where Email='{ Email}'
19
+ ''')
20
+
21
+ FData=cursor.fetchall()
22
+
23
+
24
+
25
+ if len(FData) !=0 :
26
+ if FData[0][0]==None:
27
+ return {"Status":False,"Message":"Email or Password Is Incorrect "}
28
+ HasedPassword=FData[0][1]
29
+ UserId=FData[0][0]
30
+ State= Password==HasedPassword
31
+
32
+ Name= Name
33
+ if not State:
34
+ return {"Status":False,"Message":"Email or Password is not correct"}
35
+
36
+ if not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}"):
37
+ connect.execute(f'''
38
+ INSERT INTO UserItems (UserId ,UserItemName ) VALUES ({UserId},'{Name}')
39
+ ''')
40
+ connect.commit()
41
+ if not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}"):
42
+ makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}")
43
+ chmod(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/{Name}",0o777)
44
+ if not exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Test/{Name}"):
45
+ makedirs(f"./FaceRecognition/ExtactedFaces/{UserId}/Test/{Name}")
46
+ chmod(f"./FaceRecognition/ExtactedFaces/{UserId}/Test/{Name}",0o777)
47
+ if not exists(f"./static/{UserId}/{Name}"):
48
+ makedirs(f"./static/{UserId}/{Name}")
49
+ chmod(f"./static/{UserId}/{Name}",0o777)
50
+ Image.open(BytesIO(Images.file.read())).save(f"./static/{UserId}/{Name}/user.png","PNG")
51
+ chmod(f"./static/{UserId}/{Name}/user.png",0o777)
52
+ return {"Status":True,"Message":"User Is Added Correctly"}
53
+ else:
54
+ return {"Status":False,"Message":"Name Is Already Exist"}
55
+ except Exception as e:
56
  return {"Status":False,"Message":e}