• 老广's avatar
    Session task (#2196) · 1293d721
    老广 authored
    * [Bugfix] 修复错误
    
    * [Update] 增加会话定期清理
    1293d721
tasks.py 2.06 KB
# -*- coding: utf-8 -*-
#

import datetime

from celery import shared_task
from celery.utils.log import get_task_logger
from django.utils import timezone
from django.conf import settings
from django.core.files.storage import default_storage


from ops.celery.utils import register_as_period_task, after_app_ready_start, \
    after_app_shutdown_clean
from .models import Status, Session, Command


CACHE_REFRESH_INTERVAL = 10
RUNNING = False
logger = get_task_logger(__name__)


@shared_task
@register_as_period_task(interval=3600)
@after_app_ready_start
@after_app_shutdown_clean
def delete_terminal_status_period():
    yesterday = timezone.now() - datetime.timedelta(days=3)
    Status.objects.filter(date_created__lt=yesterday).delete()


@shared_task
@register_as_period_task(interval=3600)
@after_app_ready_start
@after_app_shutdown_clean
def clean_orphan_session():
    active_sessions = Session.objects.filter(is_finished=False)
    for session in active_sessions:
        if not session.terminal or not session.terminal.is_active:
            session.is_finished = True
            session.save()


@shared_task
@register_as_period_task(interval=3600*24)
@after_app_ready_start
@after_app_shutdown_clean
def clean_expired_session_period():
    logger.info("Start clean expired session record, commands and replay")
    days = settings.TERMINAL_SESSION_KEEP_DURATION
    dt = timezone.now() - timezone.timedelta(days=days)
    expired_sessions = Session.objects.filter(date_start__lt=dt)
    for session in expired_sessions:
        logger.info("Clean session: {}".format(session.id))
        Command.objects.filter(session=str(session.id)).delete()
        # 删除录像文件
        session_path = session.get_rel_replay_path()
        local_path = session.get_local_path()
        local_path_v1 = session.get_local_path(version=1)

        # 去default storage中查找
        for _local_path in (local_path, local_path_v1, session_path):
            if default_storage.exists(_local_path):
                default_storage.delete(_local_path)
        # 删除session记录
        session.delete()