Merge pull request #90 from filips123/cleanup-database

Added command for cleaning up the database

Author: PetJer

API/gimvicurnik/__init__.py

@@ -24,6 +24,7 @@
     create_database_command,
     update_eclassroom_command,
     update_menu_command,
+    cleanup_database_command,
     update_timetable_command,
 )
 from .config import Config
@@ -245,6 +246,7 @@ def register_commands(self) -> None:
         self.app.cli.add_command(update_timetable_command)
         self.app.cli.add_command(update_eclassroom_command)
         self.app.cli.add_command(update_menu_command)
+        self.app.cli.add_command(cleanup_database_command)
         self.app.cli.add_command(create_database_command)
 
     def register_routes(self) -> None:

API/gimvicurnik/commands/__init__.py

@@ -4,7 +4,10 @@
 import click
 from flask import current_app
 
-from ..database import Base, SessionFactory
+from datetime import datetime, timedelta
+from sqlalchemy import and_, or_
+
+from ..database import Base, SessionFactory, Document, DocumentType
 from ..updaters import EClassroomUpdater, MenuUpdater, TimetableUpdater
 from ..utils.sentry import with_transaction
 
@@ -51,6 +54,26 @@ def update_menu_command() -> None:
         updater.update()
 
 
+@click.command("cleanup-database", help="Clean up the database.")
+@with_transaction(name="cleanup-database", op="command")
+def cleanup_database_command() -> None:
+    """Remove lunch schedules, snack menus and lunch menus older than 2 weeks from the database."""
+
+    logging.getLogger(__name__).info("Cleaning up the database")
+
+    with SessionFactory.begin() as session:
+        session.query(Document).filter(
+            and_(
+                or_(
+                    Document.type == DocumentType.LUNCH_SCHEDULE,
+                    Document.type == DocumentType.SNACK_MENU,
+                    Document.type == DocumentType.LUNCH_MENU,
+                ),
+                Document.effective < datetime.now().date() - timedelta(weeks=2),
+            )
+        ).delete()
+
+
 @click.command("create-database", help="Create the database.")
 @click.option("--recreate", help="Remove existing tables before creating new ones.", is_flag=True)
 @click.pass_context