Upload files to "asset"

Signed-off-by: nestict <developer@nestict.com>

asset/scheduler.py

@@ -0,0 +1,102 @@
+"""
+scheduler.py
+
+This module is used to register scheduled tasks
+"""
+
+import sys
+from datetime import date, timedelta
+
+from apscheduler.schedulers.background import BackgroundScheduler
+from django.urls import reverse
+
+from notifications.signals import notify
+
+
+def notify_expiring_assets():
+    """
+    Finds all Expiring Assets and send a notification on the notify_before date.
+    """
+    from django.contrib.auth.models import User
+
+    from asset.models import Asset
+
+    today = date.today()
+    assets = Asset.objects.all()
+
+    # Cache bot & superuser once
+    bot = User.objects.filter(username="Horilla Bot").only("id").first()
+    superuser = User.objects.filter(is_superuser=True).only("id").first()
+
+    # Query only assets that are expiring today
+    assets = Asset.objects.filter(
+        expiry_date__isnull=False,
+        expiry_date__gte=today,
+    )
+
+    for asset in assets:
+        if asset.expiry_date:
+            expiry_date = asset.expiry_date
+            notify_date = expiry_date - timedelta(days=asset.notify_before)
+            recipient = getattr(asset.owner, "employee_user_id", None) or superuser
+            if notify_date == today and recipient:
+                notify.send(
+                    bot,
+                    recipient=recipient,
+                    verb=f"The Asset '{asset.asset_name}' expires in {asset.notify_before} days",
+                    verb_ar=f"تنتهي صلاحية الأصل '{asset.asset_name}' خلال {asset.notify_before} من الأيام",
+                    verb_de=f"Das Asset {asset.asset_name} läuft in {asset.notify_before} Tagen ab.",
+                    verb_es=f"El activo {asset.asset_name} caduca en {asset.notify_before} días.",
+                    verb_fr=f"L'actif {asset.asset_name} expire dans {asset.notify_before} jours.",
+                    redirect=reverse("asset-category-view"),
+                    label="System",
+                    icon="information",
+                )
+
+
+def notify_expiring_documents():
+    """
+    Finds all Expiring Documents and send a notification on the notify_before date.
+    """
+    from django.contrib.auth.models import User
+
+    from horilla_documents.models import Document
+
+    today = date.today()
+    documents = Document.objects.all()
+    bot = User.objects.filter(username="Horilla Bot").first()
+    for document in documents:
+        if document.expiry_date:
+            expiry_date = document.expiry_date
+            notify_date = expiry_date - timedelta(days=document.notify_before)
+
+            if notify_date == today:
+                notify.send(
+                    bot,
+                    recipient=document.employee_id.employee_user_id,
+                    verb=f"The document ' {document.title} ' expires in {document.notify_before}\
+                        days",
+                    verb_ar=f"تنتهي صلاحية المستند '{document.title}' خلال {document.notify_before}\
+                    يوم",
+                    verb_de=f"Das Dokument '{document.title}' läuft in {document.notify_before}\
+                        Tagen ab.",
+                    verb_es=f"El documento '{document.title}' caduca en {document.notify_before}\
+                        días",
+                    verb_fr=f"Le document '{document.title}' expire dans {document.notify_before}\
+                        jours",
+                    redirect=reverse("asset-category-view"),
+                    label="System",
+                    icon="information",
+                )
+            if today >= expiry_date:
+                document.is_active = False
+
+
+if not any(
+    cmd in sys.argv
+    for cmd in ["makemigrations", "migrate", "compilemessages", "flush", "shell"]
+):
+    scheduler = BackgroundScheduler()
+    scheduler.add_job(notify_expiring_assets, "interval", days=1)
+    scheduler.add_job(notify_expiring_documents, "interval", hours=4)
+    scheduler.start()