mirror of https://github.com/jumpserver/jumpserver
20 lines
583 B
Python
20 lines
583 B
Python
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)
|
|
|