53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
import calendar
|
|
import datetime as dt
|
|
from datetime import datetime, timedelta
|
|
|
|
from apscheduler.schedulers.background import BackgroundScheduler
|
|
from dateutil.relativedelta import relativedelta
|
|
|
|
today = datetime.now()
|
|
|
|
|
|
def leave_reset():
|
|
from leave.models import LeaveType
|
|
|
|
today_date = today.date()
|
|
leave_types = LeaveType.objects.filter(reset=True)
|
|
# Looping through filtered leave types with reset is true
|
|
for leave_type in leave_types:
|
|
# Looping through all available leaves
|
|
available_leaves = leave_type.employee_available_leave.all()
|
|
|
|
for available_leave in available_leaves:
|
|
reset_date = available_leave.reset_date
|
|
expired_date = available_leave.expired_date
|
|
if reset_date == today_date:
|
|
available_leave.update_carryforward()
|
|
# new_reset_date = available_leave.set_reset_date(assigned_date=today_date,available_leave = available_leave)
|
|
new_reset_date = available_leave.set_reset_date(
|
|
assigned_date=today_date, available_leave=available_leave
|
|
)
|
|
available_leave.reset_date = new_reset_date
|
|
available_leave.save()
|
|
if expired_date and expired_date <= today_date:
|
|
new_expired_date = available_leave.set_expired_date(
|
|
available_leave=available_leave, assigned_date=today_date
|
|
)
|
|
available_leave.expired_date = new_expired_date
|
|
available_leave.save()
|
|
|
|
if (
|
|
leave_type.carryforward_expire_date
|
|
and leave_type.carryforward_expire_date <= today_date
|
|
):
|
|
leave_type.carryforward_expire_date = leave_type.set_expired_date(
|
|
today_date
|
|
)
|
|
leave_type.save()
|
|
|
|
|
|
scheduler = BackgroundScheduler()
|
|
scheduler.add_job(leave_reset, "interval", seconds=20)
|
|
|
|
scheduler.start()
|