nofusscomputing/centurion_erp
1
0
Fork 0
Code Issues Wiki Activity

fix(access): When creating permission QuerySet prevent app crash if db not setup

Browse Source 
ref: #947 #152
This commit is contained in:
Jon Lockwood
2025-08-10 16:42:33 +09:30
parent 6ff4779cfb
commit 3131258491
1 changed files with 56 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

92
app/access/functions/permissions.py
View File

@ -1,9 +1,10 @@
from django.apps import apps from django.apps import apps
from django.conf import settings
from django.contrib.auth.models import ( from django.contrib.auth.models import (
ContentType, ContentType,
Permission Permission
) )
from django.conf import settings from django.db.models import QuerySet
def permission_queryset(): def permission_queryset():
@ -61,47 +62,66 @@ def permission_queryset():
if not settings.RUNNING_TESTS: if not settings.RUNNING_TESTS:
models = apps.get_models() try:
# This blocks purpose is to cater for fresh install
# so that the app does not crash before the DB is setup.
for model in models: models = apps.get_models()
if( for model in models:
not str(model._meta.object_name).endswith('AuditHistory')
and not str(model._meta.model_name).lower().endswith('history')
):
# check `endswith('history')` can be removed when the old history models are removed
continue
content_type = ContentType.objects.get(
app_label = model._meta.app_label,
model = model._meta.model_name
)
permissions = Permission.objects.filter(
content_type = content_type,
)
for permission in permissions:
if( if(
not permission.codename == 'view_' + str(model._meta.model_name)
and str(model._meta.object_name).endswith('AuditHistory')
):
exclude_permissions += [ permission.codename ]
elif(
not str(model._meta.object_name).endswith('AuditHistory') not str(model._meta.object_name).endswith('AuditHistory')
and str(model._meta.model_name).lower().endswith('history') and not str(model._meta.model_name).lower().endswith('history')
): ):
# This `elif` can be removed when the old history models are removed # check `endswith('history')` can be removed when the old history models are removed
continue
exclude_permissions += [ permission.codename ] content_type = ContentType.objects.get(
app_label = model._meta.app_label,
model = model._meta.model_name
)
permissions = Permission.objects.filter(
content_type = content_type,
)
for permission in permissions:
if(
not permission.codename == 'view_' + str(model._meta.model_name)
and str(model._meta.object_name).endswith('AuditHistory')
):
exclude_permissions += [ permission.codename ]
elif(
not str(model._meta.object_name).endswith('AuditHistory')
and str(model._meta.model_name).lower().endswith('history')
):
# This `elif` can be removed when the old history models are removed
exclude_permissions += [ permission.codename ]
return Permission.objects.select_related('content_type').filter( return Permission.objects.select_related('content_type').filter(
content_type__app_label__in = centurion_apps, content_type__app_label__in = centurion_apps,
).exclude( ).exclude(
content_type__model__in = exclude_models content_type__model__in = exclude_models
).exclude( ).exclude(
codename__in = exclude_permissions codename__in = exclude_permissions
) )
except:
pass
return QuerySet()
else:
return Permission.objects.select_related('content_type').filter(
content_type__app_label__in = centurion_apps,
).exclude(
content_type__model__in = exclude_models
).exclude(
codename__in = exclude_permissions
)