Feature/mx 1524 create db purge script

CHANGELOG.md

@@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
+- db purge script
+
 ### Changes
 
 ### Deprecated

pyproject.toml

@@ -29,6 +29,7 @@ optional-dependencies.dev = [
 
 [project.scripts]
 backend = "mex.backend.main:main"
+purge-db = "scripts.purge:purge_db"
 
 [tool.cruft]
 template = "https://github.com/robert-koch-institut/mex-template"

scripts/purge.py

@@ -0,0 +1,25 @@
+import click
+from neo4j import GraphDatabase
+
+from mex.backend.settings import BackendSettings
+
+
+def purge_db() -> None:
+    """Purge graph db."""
+    settings = BackendSettings.get()
+    if click.confirm(f'Purging graph db "{settings.graph_db}", Continue?:'):
+        click.echo("Purging...")
+        with GraphDatabase.driver(
+            settings.graph_url,
+            auth=(
+                settings.graph_user.get_secret_value(),
+                settings.graph_password.get_secret_value(),
+            ),
+            database=settings.graph_db,
+        ) as driver:
+            driver.execute_query("MATCH (n) DETACH DELETE n;")
+            for row in driver.execute_query("SHOW ALL CONSTRAINTS;").records:
+                driver.execute_query(f"DROP CONSTRAINT {row['name']};")
+            for row in driver.execute_query("SHOW ALL INDEXES;").records:
+                driver.execute_query(f"DROP INDEX {row['name']};")
+        click.echo("Done")