mirror of https://github.com/interlegis/sapl.git
LeandroRoberto
8 years ago
12 changed files with 322 additions and 124 deletions
@ -1,8 +1,137 @@ |
|||
from django import apps |
|||
from builtins import LookupError |
|||
|
|||
from django.apps import apps |
|||
from django.contrib.auth.management import _get_all_permissions |
|||
from django.core import exceptions |
|||
from django.db import router |
|||
from django.db.models.signals import pre_migrate, post_migrate |
|||
from django.db.utils import DEFAULT_DB_ALIAS |
|||
from django.utils.translation import string_concat |
|||
from django.utils.translation import ugettext_lazy as _ |
|||
import django |
|||
|
|||
|
|||
def create_proxy_permissions( |
|||
app_config, verbosity=2, interactive=True, |
|||
using=DEFAULT_DB_ALIAS, **kwargs): |
|||
if not app_config.models_module: |
|||
return |
|||
|
|||
# print(app_config) |
|||
|
|||
try: |
|||
Permission = apps.get_model('auth', 'Permission') |
|||
except LookupError: |
|||
return |
|||
|
|||
if not router.allow_migrate_model(using, Permission): |
|||
return |
|||
|
|||
from django.contrib.contenttypes.models import ContentType |
|||
|
|||
permission_name_max_length = Permission._meta.get_field('name').max_length |
|||
|
|||
# This will hold the permissions we're looking for as |
|||
# (content_type, (codename, name)) |
|||
searched_perms = list() |
|||
# The codenames and ctypes that should exist. |
|||
ctypes = set() |
|||
for klass in list(app_config.get_models()): |
|||
opts = klass._meta |
|||
permissions = ( |
|||
("list_" + opts.model_name, |
|||
string_concat( |
|||
_('Visualizaçao da lista de'), ' ', |
|||
opts.verbose_name_plural)), |
|||
("detail_" + opts.model_name, |
|||
string_concat( |
|||
_('Visualização dos detalhes de'), ' ', |
|||
opts.verbose_name_plural)), |
|||
) |
|||
opts.permissions = tuple( |
|||
set(list(permissions) + list(opts.permissions))) |
|||
|
|||
if opts.proxy: |
|||
# Force looking up the content types in the current database |
|||
# before creating foreign keys to them. |
|||
app_label, model = opts.app_label, opts.model_name |
|||
|
|||
try: |
|||
ctype = ContentType.objects.db_manager( |
|||
using).get_by_natural_key(app_label, model) |
|||
except: |
|||
ctype = ContentType.objects.db_manager( |
|||
using).create(app_label=app_label, model=model) |
|||
else: |
|||
ctype = ContentType.objects.db_manager(using).get_for_model(klass) |
|||
|
|||
ctypes.add(ctype) |
|||
for perm in _get_all_permissions(klass._meta, ctype): |
|||
searched_perms.append((ctype, perm)) |
|||
|
|||
class AppConfig(apps.AppConfig): |
|||
# Find all the Permissions that have a content_type for a model we're |
|||
# looking for. We don't need to check for codenames since we already have |
|||
# a list of the ones we're going to create. |
|||
all_perms = set(Permission.objects.using(using).filter( |
|||
content_type__in=ctypes, |
|||
).values_list( |
|||
"content_type", "codename" |
|||
)) |
|||
|
|||
perms = [ |
|||
Permission(codename=codename, name=name, content_type=ct) |
|||
for ct, (codename, name) in searched_perms |
|||
if (ct.pk, codename) not in all_perms |
|||
] |
|||
# Validate the permissions before bulk_creation to avoid cryptic database |
|||
# error when the name is longer than 255 characters |
|||
for perm in perms: |
|||
if len(perm.name) > permission_name_max_length: |
|||
raise exceptions.ValidationError( |
|||
'The permission name %s of %s.%s ' |
|||
'is longer than %s characters' % ( |
|||
perm.name, |
|||
perm.content_type.app_label, |
|||
perm.content_type.model, |
|||
permission_name_max_length, |
|||
) |
|||
) |
|||
Permission.objects.using(using).bulk_create(perms) |
|||
if verbosity >= 2: |
|||
for perm in perms: |
|||
print("Adding permission '%s'" % perm) |
|||
|
|||
|
|||
def run_sql_organizers( |
|||
app_config, verbosity=2, interactive=True, |
|||
using=DEFAULT_DB_ALIAS, **kwargs): |
|||
"""with connection.cursor() as cursor: |
|||
for line in lines: |
|||
line = line.strip() |
|||
if not line or line.startswith('#'): |
|||
continue |
|||
|
|||
try: |
|||
cursor.execute(line)""" |
|||
|
|||
print('aqui run_sql_organizer', app_config) |
|||
|
|||
""" update protocoloadm_protocolo set autor_id = null; |
|||
delete from materia_autoria; |
|||
delete from materia_proposicao; |
|||
|
|||
delete from materia_tipoproposicao; |
|||
""" |
|||
|
|||
|
|||
class AppConfig(django.apps.AppConfig): |
|||
name = 'sapl.base' |
|||
label = 'base' |
|||
verbose_name = _('Dados Básicos') |
|||
|
|||
def ready(self): |
|||
pre_migrate.connect(run_sql_organizers, self) |
|||
|
|||
post_migrate.connect( |
|||
receiver=create_proxy_permissions, |
|||
dispatch_uid="django.contrib.auth.management.create_permissions") |
|||
|
|||
@ -0,0 +1,20 @@ |
|||
# -*- coding: utf-8 -*- |
|||
# Generated by Django 1.9.7 on 2016-10-11 19:45 |
|||
from __future__ import unicode_literals |
|||
|
|||
from django.db import migrations |
|||
|
|||
|
|||
def clear_model_tipo_proposicao(apps, schema_editor): |
|||
TipoProposicao = apps.get_model("materia", "TipoProposicao") |
|||
TipoProposicao.objects.all().delete() |
|||
|
|||
|
|||
class Migration(migrations.Migration): |
|||
|
|||
dependencies = [ |
|||
('materia', '0056_merge'), |
|||
] |
|||
|
|||
operations = [ |
|||
migrations.RunPython(clear_model_tipo_proposicao), ] |
|||
@ -0,0 +1,21 @@ |
|||
# -*- coding: utf-8 -*- |
|||
# Generated by Django 1.9.7 on 2016-10-23 14:44 |
|||
from __future__ import unicode_literals |
|||
|
|||
from django.db import migrations, models |
|||
import django.db.models.deletion |
|||
|
|||
|
|||
class Migration(migrations.Migration): |
|||
|
|||
dependencies = [ |
|||
('protocoloadm', '0003_auto_20161009_1222'), |
|||
] |
|||
|
|||
operations = [ |
|||
migrations.AlterField( |
|||
model_name='tramitacaoadministrativo', |
|||
name='unidade_tramitacao_destino', |
|||
field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='adm_tramitacoes_destino', to='materia.UnidadeTramitacao', verbose_name='Unidade Destino'), |
|||
), |
|||
] |
|||
Loading…
Reference in new issue