add support the database alias used for synchronization

django_pgviews/apps.py

@@ -14,7 +14,7 @@ class ViewConfig(apps.AppConfig):
     name = 'django_pgviews'
     verbose_name = 'Django Postgres Views'
 
-    def sync_pgviews(self, sender, app_config, **kwargs):
+    def sync_pgviews(self, sender, app_config, using, **kwargs):
         """Forcibly sync the views.
         """
         self.counter = self.counter + 1
@@ -26,7 +26,7 @@ def sync_pgviews(self, sender, app_config, **kwargs):
             # (models in app init are not allowed)
             from .models import ViewSyncer
             vs = ViewSyncer()
-            vs.run(force=True, update=True) 
+            vs.run(force=True, update=True, using=using)
 
     def ready(self):
         """Find and setup the apps to set the post_migrate hooks for.

django_pgviews/management/commands/clear_pgviews.py

@@ -2,7 +2,7 @@
 
 from django.core.management.base import BaseCommand
 from django.apps import apps
-from django.db import connection
+from django.db import DEFAULT_DB_ALIAS, connections
 
 from django_pgviews.view import clear_view, View, MaterializedView
 
@@ -13,9 +13,19 @@
 class Command(BaseCommand):
     help = """Clear Postgres views. Use this before running a migration"""
 
+    def add_arguments(self, parser):
+        parser.add_argument(
+            '--database',
+            default=DEFAULT_DB_ALIAS,
+            help='Nominates a database to synchronize. Defaults to the "default" database.',
+        )
+
     def handle(self, **options):
         """
         """
+        # Get the database we're operating from
+        db = options['database']
+        connection = connections[db]
         for view_cls in apps.get_models():
             if not (isinstance(view_cls, type) and
                     issubclass(view_cls, View) and

django_pgviews/management/commands/sync_pgviews.py

@@ -2,8 +2,7 @@
 import logging
 
 from django.core.management.base import BaseCommand
-from django.db import connection
-from django.apps import apps
+from django.db import DEFAULT_DB_ALIAS
 
 from django_pgviews.models import ViewSyncer
 
@@ -26,7 +25,12 @@ def add_arguments(self, parser):
             default=False,
             help="""Force replacement of pre-existing views where
             breaking changes have been made to the schema.""")
+        parser.add_argument(
+            '--database',
+            default=DEFAULT_DB_ALIAS,
+            help='Nominates a database to synchronize. Defaults to the "default" database.',
+        )
 
-    def handle(self, force, update, **options):
+    def handle(self, force, update, database, **options):
         vs = ViewSyncer()
-        vs.run(force, update)
+        vs.run(force, update, using=database)

django_pgviews/models.py

@@ -1,7 +1,7 @@
 import logging
 
 from django.apps import apps
-from django.db import connection
+from django.db import connections, DEFAULT_DB_ALIAS
 
 from django_pgviews.view import create_view, View, MaterializedView
 from django_pgviews.signals import view_synced, all_views_synced
@@ -10,7 +10,7 @@
 
 
 class ViewSyncer(object):
-    def run(self, force, update, **options):
+    def run(self, force, update, using, **options):
         self.synced = []
         backlog = []
         for view_cls in apps.get_models():
@@ -22,21 +22,22 @@ def run(self, force, update, **options):
         loop = 0
         while len(backlog) > 0 and loop < 10:
             loop += 1
-            backlog = self.run_backlog(backlog, force, update)
+            backlog = self.run_backlog(backlog, force, update, using)
 
         if loop >= 10:
             log.warn('pgviews dependencies hit limit. Check if your model dependencies are correct')
         else:
-            all_views_synced.send(sender=None)
+            all_views_synced.send(sender=None, using=using)
 
-    def run_backlog(self, models, force, update):
+    def run_backlog(self, models, force, update, using=DEFAULT_DB_ALIAS):
         '''Installs the list of models given from the previous backlog
 
         If the correct dependent views have not been installed, the view
         will be added to the backlog.
 
         Eventually we get to a point where all dependencies are sorted.
         '''
+        connection = connections[using]
         backlog = []
         for view_cls in models:
             skip = False
@@ -56,7 +57,8 @@ def run_backlog(self, models, force, update):
                         index=view_cls._concurrent_index)
                 view_synced.send(
                     sender=view_cls, update=update, force=force, status=status,
-                    has_changed=status not in ('EXISTS', 'FORCE_REQUIRED'))
+                    has_changed=status not in ('EXISTS', 'FORCE_REQUIRED'),
+                    using=using)
                 self.synced.append(name)
             except Exception as exc:
                 exc.view_cls = view_cls

django_pgviews/signals.py

@@ -2,5 +2,5 @@
 
 
 view_synced = Signal(
-    providing_args=['update', 'force', 'status', 'has_changed'])
+    providing_args=['update', 'force', 'status', 'has_changed', 'using'])
 all_views_synced = Signal()