mirror of https://github.com/jumpserver/jumpserver
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
# -*- 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.decorator import (
|
|
register_as_period_task, after_app_ready_start, after_app_shutdown_clean_periodic
|
|
)
|
|
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_periodic
|
|
def delete_terminal_status_period():
|
|
yesterday = timezone.now() - datetime.timedelta(days=1)
|
|
Status.objects.filter(date_created__lt=yesterday).delete()
|
|
|
|
|
|
@shared_task
|
|
@register_as_period_task(interval=600)
|
|
@after_app_ready_start
|
|
@after_app_shutdown_clean_periodic
|
|
def clean_orphan_session():
|
|
active_sessions = Session.objects.filter(is_finished=False)
|
|
for session in active_sessions:
|
|
if session.is_active():
|
|
continue
|
|
session.is_finished = True
|
|
session.date_end = timezone.now()
|
|
session.save()
|
|
|
|
|
|
@shared_task
|
|
@register_as_period_task(interval=3600*24)
|
|
@after_app_ready_start
|
|
@after_app_shutdown_clean_periodic
|
|
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()
|
|
|