def add_vacation(self, date_start, date_end, comment_user=None):
try:
self.vacation = Vacation(user=self.user,
date_start=date_start,
date_end=date_end,
comment_user=comment_user)
self.vacation.save()
except (ValidationError, ValueError, TypeError) as e:
if isinstance(e, ValidationError):
msg = e.message_dict
else:
msg = e.args[0]
raise ServiceException(msg)
return self.vacation
def add_vacation(self, date_start, date_end, comment_user=None):
try:
self.vacation = Vacation(
user=self.user,
date_start=date_start,
date_end=date_end,
comment_user=comment_user
)
self.vacation.save()
except (ValidationError, ValueError, TypeError) as e:
if isinstance(e, ValidationError):
msg = e.message_dict
else:
msg = e.args[0]
raise ServiceException(msg)
return self.vacation
class VacationService(object):
def __init__(self, **kwargs):
self.user = kwargs.pop('user', None)
self.vacation = kwargs.pop('vacation', None)
def add_vacation(self, date_start, date_end, comment_user=None):
try:
self.vacation = Vacation(user=self.user,
date_start=date_start,
date_end=date_end,
comment_user=comment_user)
self.vacation.save()
except (ValidationError, ValueError, TypeError) as e:
if isinstance(e, ValidationError):
msg = e.message_dict
else:
msg = e.args[0]
raise ServiceException(msg)
return self.vacation
def update_vacation(self, vacation, state, comment_admin=None):
if not state:
raise ServiceException(_('State is invalid.'))
if self.user.group_code == Employee.GUSER:
raise ServiceException(
_('You have not permissions '
'for update vacation.'))
self.vacation = vacation
if comment_admin and \
self.user.group_code == Employee.GADMIN:
self.vacation.comment_admin = comment_admin
if (state == Vacation.VACATION_APPROVED_BY_ADMIN and
self.user.group_code == Employee.GADMIN) or \
(state == Vacation.VACATION_APPROVED_BY_MANAGER and
self.user.group_code == Employee.GMGER):
return self.approve_vacation(commit=True)
elif (state == Vacation.VACATION_REJECTED_BY_ADMIN and
self.user.group_code == Employee.GADMIN) or \
(state == Vacation.VACATION_REJECTED_BY_MANAGER and
self.user.group_code == Employee.GMGER):
return self.reject_vacation(commit=True)
else:
raise ServiceException(_('State was wrong.'))
def approve_vacation(self, commit=False):
if self.user.group_code == Employee.GADMIN:
self._check_update_for_admin()
self.vacation.state = Vacation.VACATION_APPROVED_BY_ADMIN
if self.user.group_code == Employee.GMGER:
self._check_update_for_manager()
self.vacation.state = Vacation.VACATION_APPROVED_BY_MANAGER
if commit:
self.vacation.save()
return self.vacation
def reject_vacation(self, commit=False):
if self.user.group_code == Employee.GADMIN:
self._check_update_for_admin()
self.vacation.state = Vacation.VACATION_REJECTED_BY_ADMIN
if self.user.group_code == Employee.GMGER:
self._check_update_for_manager()
self.vacation.state = Vacation.VACATION_REJECTED_BY_MANAGER
if commit:
self.vacation.save()
return self.vacation
def _check_update_for_manager(self):
if self.vacation.state != Vacation.VACATION_NEW:
raise ServiceException(_('Vacation can not be changed.'))
def _check_update_for_admin(self):
if self.vacation.state in [
Vacation.VACATION_APPROVED_BY_ADMIN,
Vacation.VACATION_REJECTED_BY_ADMIN
]:
raise ServiceException(_('Vacation can not be changed.'))
class VacationService(object):
def __init__(self, **kwargs):
self.user = kwargs.pop('user', None)
self.vacation = kwargs.pop('vacation', None)
def add_vacation(self, date_start, date_end, comment_user=None):
try:
self.vacation = Vacation(
user=self.user,
date_start=date_start,
date_end=date_end,
comment_user=comment_user
)
self.vacation.save()
except (ValidationError, ValueError, TypeError) as e:
if isinstance(e, ValidationError):
msg = e.message_dict
else:
msg = e.args[0]
raise ServiceException(msg)
return self.vacation
def update_vacation(self, vacation, state, comment_admin=None):
if not state:
raise ServiceException(_('State is invalid.'))
if self.user.group_code == Employee.GUSER:
raise ServiceException(_('You have not permissions '
'for update vacation.'))
self.vacation = vacation
if comment_admin and \
self.user.group_code == Employee.GADMIN:
self.vacation.comment_admin = comment_admin
if (state == Vacation.VACATION_APPROVED_BY_ADMIN and
self.user.group_code == Employee.GADMIN) or \
(state == Vacation.VACATION_APPROVED_BY_MANAGER and
self.user.group_code == Employee.GMGER):
return self.approve_vacation(commit=True)
elif (state == Vacation.VACATION_REJECTED_BY_ADMIN and
self.user.group_code == Employee.GADMIN) or \
(state == Vacation.VACATION_REJECTED_BY_MANAGER and
self.user.group_code == Employee.GMGER):
return self.reject_vacation(commit=True)
else:
raise ServiceException(_('State was wrong.'))
def approve_vacation(self, commit=False):
if self.user.group_code == Employee.GADMIN:
self._check_update_for_admin()
self.vacation.state = Vacation.VACATION_APPROVED_BY_ADMIN
if self.user.group_code == Employee.GMGER:
self._check_update_for_manager()
self.vacation.state = Vacation.VACATION_APPROVED_BY_MANAGER
if commit:
self.vacation.save()
return self.vacation
def reject_vacation(self, commit=False):
if self.user.group_code == Employee.GADMIN:
self._check_update_for_admin()
self.vacation.state = Vacation.VACATION_REJECTED_BY_ADMIN
if self.user.group_code == Employee.GMGER:
self._check_update_for_manager()
self.vacation.state = Vacation.VACATION_REJECTED_BY_MANAGER
if commit:
self.vacation.save()
return self.vacation
def _check_update_for_manager(self):
if self.vacation.state != Vacation.VACATION_NEW:
raise ServiceException(_('Vacation can not be changed.'))
def _check_update_for_admin(self):
if self.vacation.state in [
Vacation.VACATION_APPROVED_BY_ADMIN,
Vacation.VACATION_REJECTED_BY_ADMIN
]:
raise ServiceException(_('Vacation can not be changed.'))
