fix(#1591): Implements fix

Author: igor-petrik-invitae

AUTHORS

@@ -69,6 +69,7 @@ Hamid Hashemi
 Hasan Ramezani
 Hiroki Kiyohara
 Hossein Shakiba
+Igor Petrik
 Islam Kamel
 Ivan Lukyanets
 Jaap Roes

oauth2_provider/migrations/0006_alter_application_client_secret.py

@@ -6,18 +6,18 @@
 import oauth2_provider.models
 from oauth2_provider import settings
 
-
 logger = logging.getLogger()
 
 
 def forwards_func(apps, schema_editor):
     """
     Forward migration touches every application.client_secret which will cause it to be hashed if not already the case.
     """
+    db_alias = schema_editor.connection.alias
     Application = apps.get_model(settings.APPLICATION_MODEL)
-    applications = Application._default_manager.all()
+    applications = Application._default_manager.using(db_alias).all()
     for application in applications:
-        application.save(update_fields=['client_secret'])
+        application.save(using=db_alias, update_fields=['client_secret'])
 
 
 def reverse_func(apps, schema_editor):

oauth2_provider/migrations/0012_add_token_checksum.py

@@ -8,10 +8,10 @@ def forwards_func(apps, schema_editor):
     """
     Forward migration touches every "old" accesstoken.token which will cause the checksum to be computed.
     """
+    db_alias = schema_editor.connection.alias
     AccessToken = apps.get_model(oauth2_settings.ACCESS_TOKEN_MODEL)
-    accesstokens = AccessToken._default_manager.iterator()
-    for accesstoken in accesstokens:
-        accesstoken.save(update_fields=['token_checksum'])
+    for accesstoken in AccessToken._default_manager.using(db_alias).iterator():
+        accesstoken.save(using=db_alias, update_fields=['token_checksum'])
 
 
 class Migration(migrations.Migration):