Add encryption router

Author: aclark4life

django_mongodb_backend/routers.py

@@ -16,3 +16,29 @@ def allow_migrate(self, db, app_label, model_name=None, **hints):
         except LookupError:
             return None
         return False if issubclass(model, EmbeddedModel) else None
+
+
+class EncryptionRouter:
+    """
+    Routes database operations for 'encrypted' models to the 'encryption' DB.
+    """
+
+    def db_for_read(self, model, **hints):
+        if getattr(model, "encrypted_fields_map", False):
+            return "encryption"
+        return None
+
+    def db_for_write(self, model, **hints):
+        if getattr(model, "encrypted_fields_map", False):
+            return "encryption"
+        return None
+
+    def allow_migrate(self, db, app_label, model_name=None, **hints):
+        """
+        Ensure that the 'encrypted' models only appear in the 'encryption' DB,
+        and not in the default DB.
+        """
+        model = hints.get("model")
+        if model and getattr(model, "encrypted_fields_map", False):
+            return db == "encryption"
+        return None