add elsa prof functions for seperate profs
This commit is contained in:
@@ -1489,6 +1489,35 @@ class Database:
|
|||||||
conn.commit()
|
conn.commit()
|
||||||
conn.close()
|
conn.close()
|
||||||
return self.getProfId(profdata)
|
return self.getProfId(profdata)
|
||||||
|
|
||||||
|
def getElsaProfId(self, profname):
|
||||||
|
query = f"SELECT id FROM elsa_prof WHERE fullname = '{profname}'"
|
||||||
|
data = self.query_db(query)
|
||||||
|
if data:
|
||||||
|
return data[0][0]
|
||||||
|
else: return None
|
||||||
|
|
||||||
|
def createElsaProf(self, profname)-> int:
|
||||||
|
pId = self.getElsaProfId(profname)
|
||||||
|
if pId:
|
||||||
|
return pId
|
||||||
|
query = f"INSERT INTO elsa_prof (fullname) VALUES ('{profname}')"
|
||||||
|
self.query_db(query)
|
||||||
|
return self.getElsaProfId(profname)
|
||||||
|
|
||||||
|
def getElsaProf(self, prof_id)->str:
|
||||||
|
query = f"SELECT fullname FROM elsa_prof WHERE id = '{prof_id}'"
|
||||||
|
data = self.query_db(query)
|
||||||
|
if data:
|
||||||
|
return data[0][0]
|
||||||
|
else: return None
|
||||||
|
|
||||||
|
def getElsaProfs(self)->list[str]:
|
||||||
|
query = f"SELECT fullname FROM elsa_prof"
|
||||||
|
data = self.query_db(query)
|
||||||
|
if data:
|
||||||
|
return [i[0] for i in data]
|
||||||
|
else: return []
|
||||||
|
|
||||||
def getProfId(self, profdata: dict|Prof):
|
def getProfId(self, profdata: dict|Prof):
|
||||||
conn = self.connect()
|
conn = self.connect()
|
||||||
@@ -1496,9 +1525,13 @@ class Database:
|
|||||||
if isinstance(profdata, Prof):
|
if isinstance(profdata, Prof):
|
||||||
fullname = profdata.fullname
|
fullname = profdata.fullname
|
||||||
else:
|
else:
|
||||||
fname = profdata["profname"].split(", ")[1].strip()
|
name = profdata["profname"]
|
||||||
lname = profdata["profname"].split(", ")[0].strip()
|
if ","in name:
|
||||||
fullname = f"{lname} {fname}"
|
fname = name.split(", ")[1].strip()
|
||||||
|
lname = name.split(", ")[0].strip()
|
||||||
|
fullname = f"{lname} {fname}"
|
||||||
|
else:
|
||||||
|
fullname = profdata["profname"]
|
||||||
query = f"SELECT id FROM prof WHERE fullname = '{fullname}'"
|
query = f"SELECT id FROM prof WHERE fullname = '{fullname}'"
|
||||||
self.logger.log_info(query)
|
self.logger.log_info(query)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user