Database access in data migrations defaults to the 'default' database. Force it to the user selected database instead.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2018-09-19 20:53:04 -04:00
parent fdfa4755e9
commit eda8d18146
13 changed files with 31 additions and 26 deletions

View File

@@ -13,8 +13,10 @@ def add_user_options_to_existing_users(apps, schema_editor):
app_label='user_management', model_name='UserOptions'
)
for user in User.objects.all():
UserOptions.objects.create(user=user)
for user in User.objects.using(schema_editor.connection.alias).all():
UserOptions.objects.using(
schema_editor.connection.alias
).create(user=user)
def remove_user_options_from_existing_users(apps, schema_editor):
@@ -23,8 +25,10 @@ def remove_user_options_from_existing_users(apps, schema_editor):
app_label='user_management', model_name='UserOptions'
)
for user in User.objects.all():
UserOptions.objects.filter(user=user).delete()
for user in User.objects.using(schema_editor.connection.alias).all():
UserOptions.objects.using(
schema_editor.connection.alias
).filter(user=user).delete()
class Migration(migrations.Migration):