63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
"""App configuration class for the 'users' app"""
|
|
|
|
import logging
|
|
|
|
from django.apps import AppConfig
|
|
from django.db.utils import OperationalError, ProgrammingError
|
|
|
|
from InvenTree.ready import canAppAccessDatabase
|
|
|
|
logger = logging.getLogger('inventree')
|
|
|
|
|
|
class UsersConfig(AppConfig):
|
|
"""Config class for the 'users' app"""
|
|
|
|
name = 'users'
|
|
|
|
def ready(self):
|
|
"""Called when the 'users' app is loaded at runtime"""
|
|
if canAppAccessDatabase(allow_test=True):
|
|
|
|
try:
|
|
self.assign_permissions()
|
|
except (OperationalError, ProgrammingError):
|
|
pass
|
|
|
|
try:
|
|
self.update_owners()
|
|
except (OperationalError, ProgrammingError):
|
|
pass
|
|
|
|
def assign_permissions(self):
|
|
"""Update role permissions for existing groups"""
|
|
from django.contrib.auth.models import Group
|
|
|
|
from users.models import RuleSet, update_group_roles
|
|
|
|
# First, delete any rule_set objects which have become outdated!
|
|
for rule in RuleSet.objects.all():
|
|
if rule.name not in RuleSet.RULESET_NAMES: # pragma: no cover # can not change ORM without the app being loaded
|
|
logger.info(f"Deleting outdated ruleset: {rule.name}")
|
|
rule.delete()
|
|
|
|
# Update group permission assignments for all groups
|
|
for group in Group.objects.all():
|
|
|
|
update_group_roles(group)
|
|
|
|
def update_owners(self):
|
|
"""Create an 'owner' object for each user and group instance"""
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib.auth.models import Group
|
|
|
|
from users.models import Owner
|
|
|
|
# Create group owners
|
|
for group in Group.objects.all():
|
|
Owner.create(group)
|
|
|
|
# Create user owners
|
|
for user in get_user_model().objects.all():
|
|
Owner.create(user)
|