145 lines
5.9 KiB
Python
145 lines
5.9 KiB
Python
from flaschengeist.system.controller import Singleton, userController
|
|
from flaschengeist.system.models.user import User
|
|
from datetime import datetime, timedelta
|
|
from flaschengeist.system.database import db
|
|
from flask import current_app
|
|
from werkzeug.local import LocalProxy
|
|
logger = LocalProxy(lambda: current_app.logger)
|
|
|
|
class MainController(#mainJobKindController.Base,
|
|
#mainCreditListController.Base,
|
|
#mainPricelistController.Base,
|
|
#mainWorkerController.Base,
|
|
#mainWorkgroupController.Base,
|
|
#mainJobInviteController.Base,
|
|
#mainJobRequestController.Base,
|
|
#mainRegistrationController.Base,
|
|
#mainPasswordReset.Base,
|
|
#mainFreeDrinkListConfigController.Base,
|
|
metaclass=Singleton):
|
|
|
|
def __init__(self):
|
|
logger.debug("init UserController")
|
|
pass
|
|
|
|
def setLockedDay(self, date, locked, hard=False):
|
|
logger.info(
|
|
"set day locked on {{ {} }} with state {{ {} }}".format(date, locked))
|
|
retVal = db.setLockedDay(date.date(), locked, hard)
|
|
logger.debug("seted day locked is {{ {} }}".format(retVal))
|
|
return retVal
|
|
|
|
def getLockedDays(self, from_date, to_date):
|
|
logger.info("get locked days from {{ {} }} to {{ {} }}".format(
|
|
from_date.date(), to_date.date()))
|
|
oneDay = timedelta(1)
|
|
delta = to_date.date() - from_date.date()
|
|
retVal = []
|
|
startdate = from_date - oneDay
|
|
for _ in range(delta.days + 1):
|
|
startdate += oneDay
|
|
lockday = self.getLockedDay(startdate)
|
|
retVal.append(lockday)
|
|
logger.debug("lock days are {{ {} }}".format(retVal))
|
|
return retVal
|
|
|
|
def getLockedDaysFromList(self, date_list):
|
|
logger.info("get locked days from list {{ {} }}".format(date_list))
|
|
retVal = []
|
|
for on_date in date_list:
|
|
day = datetime(on_date['on_date']['year'], on_date['on_date']['month'], on_date['on_date']['day'], 12)
|
|
retVal.append(self.getLockedDay(day))
|
|
return retVal
|
|
|
|
def getLockedDay(self, date):
|
|
logger.info("get locked day on {{ {} }}".format(date))
|
|
now = datetime.now()
|
|
logger.debug("now is {{ {} }}".format(now))
|
|
oldMonth = False
|
|
logger.debug("check if date old month or current month")
|
|
for i in range(1, 8):
|
|
if datetime(now.year, now.month, i).weekday() == 2:
|
|
if now.day < i:
|
|
oldMonth = True
|
|
break
|
|
logger.debug("oldMonth is {{ {} }}".format(oldMonth))
|
|
lockedYear = now.year
|
|
lockedMonth = now.month if now.month < now.month else now.month - \
|
|
1 if oldMonth else now.month
|
|
endDay = 1
|
|
logger.debug("calculate end day of month")
|
|
lockedYear = lockedYear if lockedMonth != 12 else (lockedYear + 1)
|
|
lockedMonth = (lockedMonth + 1) if lockedMonth != 12 else 1
|
|
for i in range(1, 8):
|
|
nextMonth = datetime(lockedYear, lockedMonth, i)
|
|
if nextMonth.weekday() == 2:
|
|
endDay = i
|
|
break
|
|
|
|
monthLockedEndDate = datetime(
|
|
lockedYear, lockedMonth, endDay) - timedelta(1)
|
|
logger.debug("get lock day from database")
|
|
retVal = db.getLockedDay(date.date())
|
|
if not retVal:
|
|
logger.debug(
|
|
"lock day not exists, retVal is {{ {} }}".format(retVal))
|
|
if date.date() <= monthLockedEndDate.date():
|
|
logger.debug("lock day {{ {} }}".format(date.date()))
|
|
self.setLockedDay(date, True)
|
|
retVal = db.getLockedDay(date.date())
|
|
else:
|
|
retVal = {"daydate": date.date(), "locked": False}
|
|
logger.debug("locked day is {{ {} }}".format(retVal))
|
|
return retVal
|
|
|
|
def __updateDataFromLDAP(self, user):
|
|
logger.info("update data from ldap for user {{ {} }}".format(user))
|
|
groups = ldap.getGroup(user.userid)
|
|
logger.debug("ldap gorups are {{ {} }}".format(groups))
|
|
user_data = ldap.getUserData(user.userid)
|
|
logger.debug("ldap data is {{ {} }}".format(user_data))
|
|
user_data['gruppe'] = groups
|
|
user_data['group'] = groups
|
|
user.updateData(user_data)
|
|
db.updateUser(user)
|
|
|
|
def checkBarUser(self, user):
|
|
logger.info("check if user {{ {} }} is baruser")
|
|
date = datetime.now()
|
|
zero = date.replace(hour=0, minute=0, second=0, microsecond=0)
|
|
end = zero + timedelta(hours=12)
|
|
startdatetime = date.replace(
|
|
hour=12, minute=0, second=0, microsecond=0)
|
|
if date > zero and end > date:
|
|
startdatetime = startdatetime - timedelta(days=1)
|
|
enddatetime = startdatetime + timedelta(days=1)
|
|
logger.debug("startdatetime is {{ {} }} and enddatetime is {{ {} }}".format(
|
|
startdatetime, end))
|
|
result = False
|
|
if date >= startdatetime and date < enddatetime:
|
|
result = db.getWorker(user, startdatetime)
|
|
logger.debug("worker is {{ {} }}".format(result))
|
|
return True if result else False
|
|
|
|
def sendMail(self, username):
|
|
logger.info("send mail to user {{ {} }}".format(username))
|
|
if type(username) == User:
|
|
user = username
|
|
if type(username) == str:
|
|
user = db.getUser(username)
|
|
retVal = emailController.sendMail(user)
|
|
logger.debug("send mail is {{ {} }}".format(retVal))
|
|
return retVal
|
|
|
|
def sendAllMail(self):
|
|
debug.info("send mail to all users")
|
|
retVal = []
|
|
users = db.getAllUser()
|
|
debug.debug("users are {{ {} }}".format(users))
|
|
for user in users:
|
|
retVal.append(self.sendMail(user))
|
|
debug.debug("send mails are {{ {} }}".format(retVal))
|
|
return retVal
|
|
|
|
|