272 lines
13 KiB
Python
272 lines
13 KiB
Python
import pymysql
|
|
from . import Singleton
|
|
from geruecht import db
|
|
from geruecht.model.user import User
|
|
from geruecht.model.creditList import CreditList
|
|
from datetime import datetime, timedelta
|
|
from geruecht.exceptions import UsernameExistDB, DatabaseExecption
|
|
import traceback
|
|
from MySQLdb._exceptions import IntegrityError
|
|
|
|
class DatabaseController(metaclass=Singleton):
|
|
'''
|
|
DatabaesController
|
|
|
|
Connect to the Database and execute sql-executions
|
|
'''
|
|
|
|
def __init__(self):
|
|
self.db = db
|
|
|
|
def getAllUser(self):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute("select * from user")
|
|
data = cursor.fetchall()
|
|
|
|
if data:
|
|
retVal = []
|
|
for value in data:
|
|
user = User(value)
|
|
creditLists = self.getCreditListFromUser(user)
|
|
user.initGeruechte(creditLists)
|
|
retVal.append(user)
|
|
return retVal
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
def getUser(self, username):
|
|
try:
|
|
retVal = None
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute("select * from user where uid='{}'".format(username))
|
|
data = cursor.fetchone()
|
|
if data:
|
|
retVal = User(data)
|
|
creditLists = self.getCreditListFromUser(retVal)
|
|
retVal.initGeruechte(creditLists)
|
|
return retVal
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
def getUserById(self, id):
|
|
try:
|
|
retVal = None
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute("select * from user where id={}".format(id))
|
|
data = cursor.fetchone()
|
|
if data:
|
|
retVal = User(data)
|
|
creditLists = self.getCreditListFromUser(retVal)
|
|
retVal.initGeruechte(creditLists)
|
|
return retVal
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
def _convertGroupToString(self, groups):
|
|
retVal = ''
|
|
print('groups: {}'.format(groups))
|
|
if groups:
|
|
for group in groups:
|
|
if len(retVal) != 0:
|
|
retVal += ','
|
|
retVal += group
|
|
return retVal
|
|
|
|
|
|
def insertUser(self, user):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
groups = self._convertGroupToString(user.group)
|
|
cursor.execute("insert into user (uid, dn, firstname, lastname, gruppe, lockLimit, locked, autoLock, mail) VALUES ('{}','{}','{}','{}','{}',{},{},{},'{}')".format(
|
|
user.uid, user.dn, user.firstname, user.lastname, groups, user.limit, user.locked, user.autoLock, user.mail))
|
|
self.db.connection.commit()
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
|
|
def updateUser(self, user):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
print('uid: {}; group: {}'.format(user.uid, user.group))
|
|
groups = self._convertGroupToString(user.group)
|
|
sql = "update user set dn='{}', firstname='{}', lastname='{}', gruppe='{}', lockLimit={}, locked={}, autoLock={}, mail='{}' where uid='{}'".format(
|
|
user.dn, user.firstname, user.lastname, groups, user.limit, user.locked, user.autoLock, user.mail, user.uid)
|
|
print(sql)
|
|
cursor.execute(sql)
|
|
self.db.connection.commit()
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
|
|
def getCreditListFromUser(self, user, **kwargs):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
if 'year' in kwargs:
|
|
sql = "select * from creditList where user_id={} and year_date={}".format(user.id, kwargs['year'])
|
|
else:
|
|
sql = "select * from creditList where user_id={}".format(user.id)
|
|
cursor.execute(sql)
|
|
data = cursor.fetchall()
|
|
if len(data) == 1:
|
|
return [CreditList(data[0])]
|
|
else:
|
|
return [CreditList(value) for value in data]
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
|
|
def createCreditList(self, user_id, year=datetime.now().year):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute("insert into creditList (year_date, user_id) values ({},{})".format(year, user_id))
|
|
self.db.connection.commit()
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
|
|
def updateCreditList(self, creditlist):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute("select * from creditList where user_id={} and year_date={}".format(creditlist.user_id, creditlist.year))
|
|
data = cursor.fetchall()
|
|
if len(data) == 0:
|
|
self.createCreditList(creditlist.user_id, creditlist.year)
|
|
sql = "update creditList set jan_guthaben={}, jan_schulden={},feb_guthaben={}, feb_schulden={}, maer_guthaben={}, maer_schulden={}, apr_guthaben={}, apr_schulden={}, mai_guthaben={}, mai_schulden={}, jun_guthaben={}, jun_schulden={}, jul_guthaben={}, jul_schulden={}, aug_guthaben={}, aug_schulden={},sep_guthaben={}, sep_schulden={},okt_guthaben={}, okt_schulden={}, nov_guthaben={}, nov_schulden={}, dez_guthaben={}, dez_schulden={}, last_schulden={} where year_date={} and user_id={}".format(creditlist.jan_guthaben, creditlist.jan_schulden,
|
|
creditlist.feb_guthaben, creditlist.feb_schulden,
|
|
creditlist.maer_guthaben, creditlist.maer_schulden,
|
|
creditlist.apr_guthaben, creditlist.apr_schulden,
|
|
creditlist.mai_guthaben, creditlist.mai_schulden,
|
|
creditlist.jun_guthaben, creditlist.jun_schulden,
|
|
creditlist.jul_guthaben, creditlist.jul_schulden,
|
|
creditlist.aug_guthaben, creditlist.aug_schulden,
|
|
creditlist.sep_guthaben, creditlist.sep_schulden,
|
|
creditlist.okt_guthaben, creditlist.okt_schulden,
|
|
creditlist.nov_guthaben, creditlist.nov_schulden,
|
|
creditlist.dez_guthaben, creditlist.dez_schulden,
|
|
creditlist.last_schulden, creditlist.year, creditlist.user_id)
|
|
print(sql)
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute(sql)
|
|
self.db.connection.commit()
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
def getWorker(self, user, date):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute("select * from bardienste where user_id={} and startdatetime='{}'".format(user.id, date))
|
|
data = cursor.fetchone()
|
|
return {"user": user.toJSON(), "startdatetime": data['startdatetime'], "enddatetime": data['enddatetime'], "start": { "year": data['startdatetime'].year, "month": data['startdatetime'].month, "day": data['startdatetime'].day}} if data else None
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
def getWorkers(self, date):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute("select * from bardienste where startdatetime='{}'".format(date))
|
|
data = cursor.fetchall()
|
|
return [{"user": self.getUserById(work['user_id']).toJSON(), "startdatetime": work['startdatetime'], "enddatetime": work['enddatetime'], "start": { "year": work['startdatetime'].year, "month": work['startdatetime'].month, "day": work['startdatetime'].day}} for work in data]
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
def setWorker(self, user, date):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute("insert into bardienste (user_id, startdatetime, enddatetime) values ({},'{}','{}')".format(user.id, date, date + timedelta(days=1)))
|
|
self.db.connection.commit()
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
|
|
def deleteWorker(self, user, date):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute("delete from bardienste where user_id={} and startdatetime='{}'".format(user.id, date))
|
|
self.db.connection.commit()
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
def changeUsername(self, user, newUsername):
|
|
try:
|
|
cursor= self.db.connection.cursor()
|
|
cursor.execute("select * from user where uid='{}'".format(newUsername))
|
|
data = cursor.fetchall()
|
|
if data:
|
|
raise UsernameExistDB("Username already exists")
|
|
else:
|
|
cursor.execute("update user set uid='{}' where id={}".format(newUsername, user.id))
|
|
self.db.connection.commit()
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
def getLockedDay(self, date):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
cursor.execute("select * from locked_days where daydate='{}'".format(date))
|
|
data = cursor.fetchone()
|
|
return data
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
def setLockedDay(self, date, locked, hard=False):
|
|
try:
|
|
cursor = self.db.connection.cursor()
|
|
sql = "insert into locked_days (daydate, locked) VALUES ('{}', {})".format(date, locked)
|
|
cursor.execute(sql)
|
|
self.db.connection.commit()
|
|
return self.getLockedDay(date)
|
|
except IntegrityError as err:
|
|
self.db.connection.rollback()
|
|
try:
|
|
exists = self.getLockedDay(date)
|
|
if hard:
|
|
sql = "update locked_days set locked={} where id={}".format(locked, exists['id'])
|
|
else:
|
|
sql = False
|
|
if sql:
|
|
cursor.execute(sql)
|
|
self.db.connection.commit()
|
|
return self.getLockedDay(date)
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseController("Something went wrong with Database: {}".format(err))
|
|
except Exception as err:
|
|
traceback.print_exc()
|
|
self.db.connection.rollback()
|
|
raise DatabaseExecption("Something went worng with Datatabase: {}".format(err))
|
|
|
|
if __name__ == '__main__':
|
|
db = DatabaseController()
|
|
user = db.getUser('jhille')
|
|
db.getCreditListFromUser(user, year=2018)
|