diff options
Diffstat (limited to 'src/database/dbmission.py')
| -rw-r--r-- | src/database/dbmission.py | 70 |
1 files changed, 36 insertions, 34 deletions
diff --git a/src/database/dbmission.py b/src/database/dbmission.py index 9d23288..0bb73e0 100644 --- a/src/database/dbmission.py +++ b/src/database/dbmission.py @@ -10,18 +10,18 @@ # https:#www.freecodecamp.org/news/sql-joins-tutorial/ # Import de dbmain -from src.database import dbmain +from src.database.dbmain import DbMain from datetime import datetime # Extension de cette classe avec dbmain -class DbMission(dbmain.DbMain): +class DbMission(DbMain): # **************************************************************************** # ENREGISTREMENT ET VALIDATION DES MISSIONS # **************************************************************************** # Enregistrement d'une mission - def register_mission(self, subject, clientId, proId, jobId): + def register_mission(self, subject: str, clientId: int, proId: int, jobId: int) -> int: # Préparation de la requète pour enregistrement de la mission reqAddMission = """ INSERT INTO """ + self.TABLEMISSION + """ @@ -29,18 +29,17 @@ class DbMission(dbmain.DbMain): VALUES (%s, %s, %s, %s, %s)""" # Exécution de la requète - self.exec_cmd(reqAddMission, - [datetime.now().strftime("%Y-%m-%d %H:%M:%S"), - subject, - clientId, - proId, - jobId - ] - ) + self.exec_cmd( + reqAddMission, + [datetime.now().strftime("%Y-%m-%d %H:%M:%S"), + subject, clientId, proId, jobId], + True + ) return 1 + # Enregistrement d'une mission - def accept_mission_by_missionid(self, missionId): + def accept_mission_by_missionid(self, missionId: int) -> int: # Préparation de la requète pour enregistrement de la mission reqAddMission = """ UPDATE @@ -50,10 +49,15 @@ class DbMission(dbmain.DbMain): WHERE missionId = %s""" # Exécution de la requète - self.exec_cmd(reqAddMission, [datetime.now().strftime("%Y-%m-%d %H:%M:%S"), missionId], True) + self.exec_cmd( + reqAddMission, + [datetime.now().strftime("%Y-%m-%d %H:%M:%S"), missionId], + True + ) return 1 - def validate_mission_pro_by_missionid(self, missionId): + + def validate_mission_pro_by_missionid(self, missionId: int) -> int: # Préparation de la requète pour enregistrement de la mission reqAddMission = """ UPDATE @@ -63,10 +67,15 @@ class DbMission(dbmain.DbMain): WHERE missionId = %s""" # Exécution de la requète - self.exec_cmd(reqAddMission, [datetime.now().strftime("%Y-%m-%d %H:%M:%S"), missionId], True) + self.exec_cmd( + reqAddMission, + [datetime.now().strftime("%Y-%m-%d %H:%M:%S"), missionId], + True + ) return 1 - def validate_mission_client_by_missionid(self, missionId): + + def validate_mission_client_by_missionid(self, missionId: int) -> int: # Préparation de la requète pour enregistrement de la mission reqAddMission = """ UPDATE @@ -76,9 +85,14 @@ class DbMission(dbmain.DbMain): WHERE missionId = %s""" # Exécution de la requète - self.exec_cmd(reqAddMission, [datetime.now().strftime("%Y-%m-%d %H:%M:%S"), missionId], True) + self.exec_cmd( + reqAddMission, + [datetime.now().strftime("%Y-%m-%d %H:%M:%S"), missionId], + True + ) return 1 + def review_mission(self, missionId: int, review: str, note: int) -> None: # Préparation de la requète pour enregistrement de la mission reqAddMission = """ @@ -91,11 +105,12 @@ class DbMission(dbmain.DbMain): # Exécution de la requète self.exec_cmd(reqAddMission, [review, note, missionId], True) + # **************************************************************************** # RECUPERATION DES INFORMATIONS RELATIVES AUX MISSIONS # **************************************************************************** # Récupérations des missions d'un pro par son id - def get_mission_by_proid(self, proId): + def get_mission_by_proid(self, proId: int) -> list: reqGetMissionsProId = """ SELECT missionId, date, subject, note, review, @@ -113,8 +128,9 @@ class DbMission(dbmain.DbMain): result = self.cur.fetchall() return result + # Récupérations des missions d'un client par son id - def get_mission_by_clientid(self, clientId): + def get_mission_by_clientid(self, clientId: int) -> list: reqGetMissionsClientId = """ SELECT missionId, date, subject, note, review, @@ -132,6 +148,7 @@ class DbMission(dbmain.DbMain): result = self.cur.fetchall() return result + # Récupérations des missions d'un client par son id def get_all_missions(self): reqGetAllMission = """ @@ -142,18 +159,3 @@ class DbMission(dbmain.DbMain): self.exec_cmd(reqGetAllMission) result = self.cur.fetchall() return result - - # Compte le nombre de mission d'un consultant par son nom - def count_pro_missions_by_proid(self, proId): - reqCountMission = """ - SELECT COUNT(*) FROM - """ + self.TABLEMISSION + """ - INNER JOIN - """ + self.TABLEUSERINFO + """ - ON - """ + self.TABLEMISSION + """.proId = """ + self.TABLEUSERINFO + """.userId - WHERE - proId = %s""" - self.exec_cmd(reqCountMission, [proId]) - result = self.cur.fetchall() - return result |