class LoginMenu:
"""Kirjautumisesta vastaava luokka"""
def __init__(self):
"""Luokan konstruktori. Käyttäliittymä ja tietokantayhteys"""
self._ui = UI()
self._repository = CourseRepository(get_database_connection())
def login(self):
"""Kirjautuminen
Returns:
Käyttäjänimi
"""
username=""
while username=="":
username = self._ui.read("Anna käyttäjätunnus: ")
result = self._repository.find_user(username)
if result == []:
self._ui.write("Ei tunnusta valmiina")
self._singup(username)
self.login()
else:
result=[]
while result==[]:
password = self._ui.read_password("Anna salasana: ")
result = self._repository.find_password(username,password)
return username
def _singup(self,username):
"""Uuden tunnuksen luonti
Args:
username: käyttäjänimi
"""
result = self._ui.read("Haluatko luoda tunnuksen k/e: ")
if result == "k":
password_check=False
while password_check is False:
password=self._ui.read_password("Anna uusi salasana: ")
if len(password)>4:
password_check=True
else:
self._ui.write("Liian lyhyt salasana")
self._repository.new_user(username, password)
class HomeMenu:
"""Päävalikon logiikasta vastaava luokka
Attributes:
username: kirjautuneen käyttäjän nimi
"""
def __init__(self, username):
"""Luokan konstruktori. Luo päävalikon valinnat
Args:
username: kirjautuneen käyttäjän nimi
"""
self._actions = {
"x": "x Lopeta",
"1": "1 Uusi kurssi",
"2": "2 Hae kurssit",
"3": "3 Poista kurssi",
"4": "4 Muuta tietoja",
"5": "5 Syötä tenttipäivämäärä",
"6": "6 Ilmoittaudu kurssille",
"7": "7 Tulevat tenttini"
}
self._headers = ["Kurssi", "Opintopisteet"]
self._ui = UI()
self._repository = CourseRepository(get_database_connection())
self._username = username
def start(self):
"""Päävalikko"""
self._help()
while True:
action = self._ui.read("Anna komento: ")
if not action in self._actions:
self._ui.write("virheellinen komento")
self._help()
continue
if action == "1":
self._new_course()
elif action == "2":
self._get_courses()
elif action == "3":
self._delete_course()
elif action == "4":
self._change_course()
elif action == "5":
self._exam_date()
elif action == "6":
self._register_course()
elif action == "7":
self._my_exam()
elif action == "x":
break
def _input_course(self):
"""Kurssin tarkistaminen tietokannasta
Returns:
Kurssin nimi
jos ei löydy niin False
"""
name = self._ui.read("Anna kurssinimi: ")
if self._repository.one_course(name) == []:
self._ui.write("Kurssia ei löydy")
return False
return name
def _new_course(self):
"""Uuden kurssin luominen"""
name = self._ui.read("Anna kurssinimi: ")
credit = self._ui.read("Anna opintopisteet: ")
self._repository.create(name, credit)
def _get_courses(self):
"""Kurssitietojen haku"""
self._ui.write("{:<15}{:<18}".format(self._headers[0],
self._headers[1]))
for i in self._repository.get():
self._ui.write("{:<15}{:<18}".format(i.course_name, i.credit))
def _delete_course(self):
"""Kurssin poistaminen"""
name = self._input_course()
if name is not False:
self._repository.delete_one(name)
def _change_course(self):
"""Kurssin opintopisteiden vaihtaminen"""
name = self._input_course()
if name is not False:
credit = self._ui.read("Anna uudet opintopisteet: ")
self._repository.change(name, credit)
def _exam_date(self):
"""Kurssin tenttipäivän lisääminen"""
name = self._input_course()
if name is not False:
date = self._ui.read("Anna tenttipäivämäärä dd/mm/yyyy: ")
day, month, year = date.split('/')
try:
datetime.datetime(int(year), int(month), int(day))
except ValueError:
self._ui.write("Väärä päivämäärä")
return
today_date = datetime.datetime.today().strftime('%Y%m%d')
if date < today_date:
self._ui.write("Väärä päivämäärä")
return
self._repository.course_date(name, year + month + day)
def _register_course(self):
"""Kurssille rekisteröinti"""
name = self._input_course()
if name is not False:
self._repository.register_course(name, self._username)
def _my_exam(self):
"""Käyttäjän tulevat kurssit"""
today_date = datetime.datetime.today().strftime('%Y%m%d')
for i in self._repository.my_exam(self._username, today_date):
self._ui.write("{:<15}{:<18}".format(
i[0], i[1][6:8] + "/" + i[1][4:6] + "/" + i[1][2:4]))
def _help(self):
"""Päävalikko"""
for i in self._actions:
self._ui.write(self._actions[i])