mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
582 B
19 lines
582 B
from django.db.models.signals import pre_save, post_save
|
|
from django.dispatch import receiver
|
|
|
|
from terminal.models import Session
|
|
|
|
|
|
@receiver(pre_save, sender=Session)
|
|
def on_session_pre_save(sender, instance, **kwargs):
|
|
if instance.need_update_cmd_amount:
|
|
instance.cmd_amount = instance.compute_command_amount()
|
|
|
|
|
|
@receiver(post_save, sender=Session)
|
|
def on_session_finished(sender, instance: Session, created, **kwargs):
|
|
if not instance.is_finished:
|
|
return
|
|
# 清理一次可能因 task 未执行的缓存数据
|
|
Session.unlock_session(instance.id)
|