Code layout change.

[wolnelektury.git] / src / waiter / tasks.py

diff --git a/src/waiter/tasks.py b/src/waiter/tasks.py
new file mode 100644 (file)
index 0000000..89db402
--- /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)