2017-12-04 12:15:47 +00:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
#
|
2016-11-10 15:54:21 +00:00
|
|
|
|
2019-11-11 10:50:05 +00:00
|
|
|
import sys
|
2016-11-10 15:54:21 +00:00
|
|
|
from celery import shared_task
|
2019-09-27 10:30:13 +00:00
|
|
|
from django.conf import settings
|
2018-11-22 10:02:12 +00:00
|
|
|
|
2019-11-11 10:50:05 +00:00
|
|
|
from ops.celery.utils import (
|
|
|
|
create_or_update_celery_periodic_tasks, disable_celery_periodic_task
|
|
|
|
)
|
2019-07-31 08:57:21 +00:00
|
|
|
from ops.celery.decorator import after_app_ready_start
|
2018-11-22 10:02:12 +00:00
|
|
|
from common.utils import get_logger
|
2019-03-04 03:36:10 +00:00
|
|
|
from .models import User
|
2019-07-31 08:57:21 +00:00
|
|
|
from .utils import (
|
|
|
|
send_password_expiration_reminder_mail, send_user_expiration_reminder_mail
|
|
|
|
)
|
2019-11-11 08:41:32 +00:00
|
|
|
from settings.utils import LDAPServerUtil, LDAPImportUtil
|
2018-11-22 10:02:12 +00:00
|
|
|
|
|
|
|
|
|
|
|
logger = get_logger(__file__)
|
2016-11-10 15:54:21 +00:00
|
|
|
|
|
|
|
|
2018-11-22 10:02:12 +00:00
|
|
|
@shared_task
|
|
|
|
def check_password_expired():
|
2020-07-28 09:53:01 +00:00
|
|
|
users = User.objects.filter(source=User.SOURCE_LOCAL).exclude(role=User.ROLE.APP)
|
2018-11-22 10:02:12 +00:00
|
|
|
for user in users:
|
2019-07-30 04:52:45 +00:00
|
|
|
if not user.is_valid:
|
|
|
|
continue
|
2018-11-22 10:02:12 +00:00
|
|
|
if not user.password_will_expired:
|
|
|
|
continue
|
2019-07-30 04:52:45 +00:00
|
|
|
msg = "The user {} password expires in {} days"
|
|
|
|
logger.info(msg.format(user, user.password_expired_remain_days))
|
2020-03-13 04:33:09 +00:00
|
|
|
send_password_expiration_reminder_mail(user)
|
2018-11-22 10:02:12 +00:00
|
|
|
|
|
|
|
|
|
|
|
@shared_task
|
|
|
|
@after_app_ready_start
|
|
|
|
def check_password_expired_periodic():
|
|
|
|
tasks = {
|
|
|
|
'check_password_expired_periodic': {
|
|
|
|
'task': check_password_expired.name,
|
|
|
|
'interval': None,
|
|
|
|
'crontab': '0 10 * * *',
|
|
|
|
'enabled': True,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
create_or_update_celery_periodic_tasks(tasks)
|
2019-03-04 02:39:44 +00:00
|
|
|
|
|
|
|
|
2019-07-31 08:57:21 +00:00
|
|
|
@shared_task
|
|
|
|
def check_user_expired():
|
2020-07-28 09:53:01 +00:00
|
|
|
users = User.objects.exclude(role=User.ROLE.APP)
|
2019-07-31 08:57:21 +00:00
|
|
|
for user in users:
|
|
|
|
if not user.is_valid:
|
|
|
|
continue
|
|
|
|
if not user.will_expired:
|
|
|
|
continue
|
2020-03-13 04:33:09 +00:00
|
|
|
msg = "The user {} will expires in {} days"
|
|
|
|
logger.info(msg.format(user, user.expired_remain_days))
|
2019-07-31 08:57:21 +00:00
|
|
|
send_user_expiration_reminder_mail(user)
|
|
|
|
|
|
|
|
|
|
|
|
@shared_task
|
|
|
|
@after_app_ready_start
|
|
|
|
def check_user_expired_periodic():
|
|
|
|
tasks = {
|
|
|
|
'check_user_expired_periodic': {
|
|
|
|
'task': check_user_expired.name,
|
|
|
|
'interval': None,
|
|
|
|
'crontab': '0 14 * * *',
|
|
|
|
'enabled': True,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
create_or_update_celery_periodic_tasks(tasks)
|
2019-03-04 03:36:10 +00:00
|
|
|
|
2019-09-27 06:04:10 +00:00
|
|
|
|
|
|
|
@shared_task
|
2019-11-11 08:41:32 +00:00
|
|
|
def import_ldap_user():
|
|
|
|
logger.info("Start import ldap user task")
|
|
|
|
util_server = LDAPServerUtil()
|
|
|
|
util_import = LDAPImportUtil()
|
|
|
|
users = util_server.search()
|
|
|
|
errors = util_import.perform_import(users)
|
|
|
|
if errors:
|
|
|
|
logger.error("Imported LDAP users errors: {}".format(errors))
|
|
|
|
else:
|
|
|
|
logger.info('Imported {} users successfully'.format(len(users)))
|
2019-09-27 06:04:10 +00:00
|
|
|
|
|
|
|
|
|
|
|
@shared_task
|
|
|
|
@after_app_ready_start
|
2019-11-11 08:41:32 +00:00
|
|
|
def import_ldap_user_periodic():
|
2019-09-27 06:04:10 +00:00
|
|
|
if not settings.AUTH_LDAP:
|
|
|
|
return
|
2019-09-27 10:19:19 +00:00
|
|
|
if not settings.AUTH_LDAP_SYNC_IS_PERIODIC:
|
2019-11-11 10:50:05 +00:00
|
|
|
task_name = sys._getframe().f_code.co_name
|
|
|
|
disable_celery_periodic_task(task_name)
|
2019-09-27 10:19:19 +00:00
|
|
|
return
|
|
|
|
|
|
|
|
interval = settings.AUTH_LDAP_SYNC_INTERVAL
|
|
|
|
if isinstance(interval, int):
|
|
|
|
interval = interval * 3600
|
|
|
|
else:
|
|
|
|
interval = None
|
|
|
|
crontab = settings.AUTH_LDAP_SYNC_CRONTAB
|
2019-09-27 06:04:10 +00:00
|
|
|
tasks = {
|
2019-11-11 08:41:32 +00:00
|
|
|
'import_ldap_user_periodic': {
|
|
|
|
'task': import_ldap_user.name,
|
2019-09-27 10:19:19 +00:00
|
|
|
'interval': interval,
|
|
|
|
'crontab': crontab,
|
2019-09-27 06:04:10 +00:00
|
|
|
'enabled': True,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
create_or_update_celery_periodic_tasks(tasks)
|