def create_user(firstname, lastname, login, *args, **kwargs):
try:
if firstname.isalpha() and login.isalnum():
query = DatabaseConnection().fetch_query(
"Select login from instruktorzy where login='******'".format(
login))
if query:
with ignored(IndexError):
if [result for result in query][0][0] == login:
return False
cmd = "Insert into instruktorzy (id, login, password, imie, nazwisko, uprawnienia) values " \
"(instruktorzy_id_seq.nextval, " \
"'{login}', '{password}', '{firstname}', '{lastname}', 'User')".format(firstname=firstname,
lastname=lastname,
login=login,
password=hash(login))
DatabaseConnection().execute_command(cmd)
query = DatabaseConnection().fetch_query(
"Select login from instruktorzy where login='******'".format(
login))
if query:
if [result for result in query][0][0] == login:
return True
except:
return None
def get_unoccupied(date, hour, *args, **kwargs):
unoccupied_instructors = []
try:
all_instructors = busy_instructors = {}
query = DatabaseConnection().fetch_query(
"select imie, nazwisko, id from instruktorzy")
if query:
all_instructors = set([result for result in query])
query = DatabaseConnection().fetch_query(
"select instruktorzy.imie, instruktorzy.nazwisko, "
"instruktorzy.id from instruktorzy join "
"lekcja on instruktorzy.id = lekcja.id_instruktora where data = "
"TO_DATE('{}', 'yyyy/mm/dd') and godzina = {}".format(
date, hour))
if query:
busy_instructors = set([result for result in query])
for instructor in list(all_instructors - busy_instructors):
unoccupied_instructors.append(
User(name=instructor[0],
user_id=instructor[2],
permission="User",
lastname=instructor[1]))
return unoccupied_instructors
except:
print "Error loading unoccupied instructors {}".format(
sys.exc_info()[0])
def remove_user(user, *args, **kwargs):
with ignored(Exception):
DatabaseConnection().execute_command(
'Delete from instruktorzy where id={}'.format(user.id))
query = DatabaseConnection().fetch_query(
'Select id from instruktorzy where id={}'.format(user.id))
if query:
if len([result for result in query]) > 0:
return False
return True
def password_update(old_pw, new_pw, user, *args, **kwargs):
cmd = "Update instruktorzy set password = '******' where id = {} and password = '******'".format(
new_pw, user.id, old_pw)
DatabaseConnection().execute_command(cmd)
query = DatabaseConnection().fetch_query(
"Select password from instruktorzy where id = '{}'".format(
user.id))
if query:
return True if [result for result in query
][0][0] == str(new_pw) else False
def change_permissions(*args, **kwargs):
with ignored(Exception):
user_button, permission_button = args
if permission_button in privileges:
DatabaseConnection().execute_command(
"Update instruktorzy set uprawnienia = '{}' where id = {}".
format(permission_button, user_button.id))
query = DatabaseConnection().fetch_query(
'Select uprawnienia from instruktorzy where id = {}'.
format(user_button.id))
if query:
return True if [result for result in query
][0][0] == permission_button else False
def get_earnings(period, user, *args, **kwargs):
with ignored(Exception):
cmd = EarningsCmdChooser().get_earnings_cmd_from[period[0]](period,
user)
query = DatabaseConnection().fetch_query(cmd)
if query:
return [result for result in query][0][0]
def remove_lesson(*args, **kwargs):
try:
if kwargs['lesson_id'] != "0":
DatabaseConnection().execute_command(
"delete from lekcja where id = {}".format(
kwargs['lesson_id']))
except:
print "Couldnt remove lesson from Database {}".format(
sys.exc_info()[0])
def insert_new_lesson(number_of_people, *args, **kwargs):
try:
cmd = "Select id from lekcja where id_instruktora = {instructor} and " \
"godzina = {godzina} and " \
"data = TO_DATE('{data}', 'yyyy/mm/dd')".format(instructor=kwargs['user'].id,
godzina=kwargs['hour'], data=kwargs['date'])
query = DatabaseConnection().fetch_query(cmd)
if query:
if len([result for result in query]) != 0:
return False
if kwargs.get('name').isalpha() and kwargs.get('age').isdigit():
if kwargs['lesson_id'] == "0":
query = DatabaseConnection().fetch_query(
"select lesson_id_seq.nextval from dual")
if query:
kwargs['lesson_id'] = [result
for result in query][0][0]
else:
DatabaseConnection().execute_command(
"delete from lekcja where id = {}".format(
kwargs['lesson_id']))
DatabaseConnection().execute_command(
"insert into lekcja "
"(imie, godzina, data, ilosc_osob, koszt, id, id_instruktora, wiek) "
"values ('{imie}', {godzina}, TO_DATE('{data}', 'yyyy/mm/dd'), "
"{ilosc_osob}, {koszt}, {lesson_id}, {id_instruktora}, {wiek})"
.format(imie=kwargs['name'],
godzina=kwargs['hour'],
data=kwargs['date'],
ilosc_osob=number_of_people,
koszt=cennik.get(int(number_of_people), 0),
lesson_id=kwargs['lesson_id'],
id_instruktora=kwargs['user'].id,
wiek=kwargs['age']))
return True
except:
print "Couldnt add lesson {}".format(sys.exc_info()[0])
def get_all_users(*args, **kwargs):
users = {}
try:
query = DatabaseConnection().fetch_query(
"select imie, nazwisko, id, uprawnienia from instruktorzy")
if query:
for output in query:
name, lastname, user_id, privileges = output
users[name] = User(user_id=user_id,
permission=privileges,
name=name,
lastname=lastname)
finally:
return users
def fetch_logins(username, password, *args, **kwargs):
""" Template method to execute sql commands,
methods needs to get *args, **kwargs, and
to be decorated with @update_bag_with_result """
try:
query = DatabaseConnection().fetch_query(
"select imie, nazwisko, id, uprawnienia "
"from Instruktorzy "
"where login = '******' and password = '******'".format(
username, password))
if query:
return [result for result in query][0]
except:
return None
def get_daily_graph(day, user, *args, **kwargs):
busy_hours = {}
try:
query = DatabaseConnection().fetch_query(
"select godzina, imie, wiek, ilosc_osob, id "
"from lekcja "
"where id_instruktora = {} "
"and data = TO_DATE('{}', 'yyyy/mm/dd')".format(user.id, day))
if query:
for result in query:
busy_hours[result[0]] = (
"{}.00 | {}, {}lat. #{}os.".format(*result[:-1]),
result[-1])
except:
print "Couldnt read daily graph {}".format(sys.exc_info()[0])
finally:
return busy_hours