X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/6280673f35e13e75e81c5b7821bd2a44a1831eab..357027375ff8867f42ca34bcbfb5a78b5b185fc3:/src/waiter/tasks.py?ds=sidebyside

diff --git a/src/waiter/tasks.py b/src/waiter/tasks.py
new file mode 100644
index 000000000..89db402a3
--- /dev/null
+++ b/src/waiter/tasks.py
@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from celery.signals import task_postrun
+from waiter.models import WaitedFile
+
+
+def task_delete_after(task_id=None, **kwargs):
+    WaitedFile.objects.filter(task_id=task_id).delete()
+task_postrun.connect(task_delete_after)