Browse Source

Muda User para get_user_model()

Signed-off-by: Luciano Almeida <mdiebr@gmail.com>
pull/827/head
Luciano Almeida 8 years ago
parent
commit
a83a90ca39
  1. 13
      sapl/legacy/migration.py

13
sapl/legacy/migration.py

@ -4,7 +4,7 @@ import pkg_resources
import yaml import yaml
from django.apps import apps from django.apps import apps
from django.apps.config import AppConfig from django.apps.config import AppConfig
from django.contrib.auth.models import User from django.contrib.auth import get_user_model
from django.core.exceptions import ObjectDoesNotExist from django.core.exceptions import ObjectDoesNotExist
from django.db import connections, models from django.db import connections, models
from django.db.models import CharField, TextField, ProtectedError from django.db.models import CharField, TextField, ProtectedError
@ -332,7 +332,7 @@ class DataMigrator:
# warning: model/app migration order is of utmost importance # warning: model/app migration order is of utmost importance
self.to_delete = [] self.to_delete = []
ProblemaMigracao.objects.all().delete() ProblemaMigracao.objects.all().delete()
User.objects.all().delete() get_user_model().objects.exclude(is_superuser=True).delete()
info('Começando migração: %s...' % obj) info('Começando migração: %s...' % obj)
self._do_migrate(obj) self._do_migrate(obj)
@ -540,12 +540,15 @@ def adjust_autor(new, old):
new.autor_related = Comissao.objects.get(pk=old.cod_comissao) new.autor_related = Comissao.objects.get(pk=old.cod_comissao)
if old.col_username: if old.col_username:
if not User.objects.filter(username=old.col_username).exists(): if not get_user_model().objects.filter(
user = User(username=old.col_username, password=12345) username=old.col_username).exists():
user = get_user_model()(
username=old.col_username, password=12345)
user.save() user.save()
new.user = user new.user = user
else: else:
new.user = User.objects.filter(username=old.col_username)[0] new.user = get_user_model().objects.filter(
username=old.col_username)[0]
AJUSTE_ANTES_SALVAR = { AJUSTE_ANTES_SALVAR = {

Loading…
Cancel
Save