management commands for cli pragma key

Author: benjaoming

sqlcipher/management/__init__.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+from __future__ import print_function
+from django.conf import settings
+from getpass import getpass
+
+
+class PromptForPragmaKeyMixin(object):
+    """""
+    This is a universal command that you can have other management commands
+    inherit from in case they need database access.
+    """
+
+    def handle(self, *args, **options):
+        if not hasattr(settings, 'PRAGMA_KEY') or not settings.PRAGMA_KEY:
+            print("There is no SQL Cipher key defined, it's unsafe to store in your settings. Please input your key")
+            key = getpass("Key: ")
+            settings.PRAGMA_KEY = key
+        super(PromptForPragmaKeyMixin, self).handle(*args, **options)

sqlcipher/management/commands/__init__.py

@@ -0,0 +1,12 @@
+
+from django.core.management.commands.migrate import Command as BaseCommand
+
+from ._mixins import PromptForPragmaKeyMixin
+
+
+class Command(PromptForPragmaKeyMixin, BaseCommand):
+    """
+    Before migrating, we need to know the pragma key to access the database. If
+    it does not exist, retrieve it from command line input.
+    """
+    pass

sqlcipher/management/commands/_mixins.py

@@ -0,0 +1,6 @@
+from django.core.management.commands.runserver import Command as RunserverCommand
+
+from ._mixins import PromptForPragmaKeyMixin
+
+class Command(PromptForPragmaKeyMixin, RunserverCommand):
+    pass