diff --git a/apps/common/event/__init__.py b/apps/common/event/__init__.py index 909740b7a..462d8da79 100644 --- a/apps/common/event/__init__.py +++ b/apps/common/event/__init__.py @@ -13,6 +13,6 @@ from .listener_manage import * def run(): listener_manage.ListenerManagement().run() - QuerySet(Document).filter(status=Status.embedding).update(**{'status': Status.error}) + QuerySet(Document).filter(status__in=[Status.embedding, Status.queue_up]).update(**{'status': Status.error}) QuerySet(Model).filter(status=setting.models.Status.DOWNLOAD).update(status=setting.models.Status.ERROR, meta={'message': "下载程序被中断,请重试"}) diff --git a/apps/common/event/common.py b/apps/common/event/common.py index 480748c31..a723592c5 100644 --- a/apps/common/event/common.py +++ b/apps/common/event/common.py @@ -30,8 +30,10 @@ def get_cache_key(poxy_function, args): def get_cache_poxy_function(poxy_function, cache_key): def fun(args, **keywords): - poxy_function(args, **keywords) - memory_cache.delete(cache_key) + try: + poxy_function(args, **keywords) + finally: + memory_cache.delete(cache_key) return fun