django-helpdeskmig/helpdesk/management/commands/create_usersettings.py

38 lines
1.2 KiB
Python

#!/usr/bin/python
"""
django-helpdesk - A Django powered ticket tracker for small enterprise.
See LICENSE for details.
create_usersettings.py - Easy way to create helpdesk-specific settings for
users who don't yet have them.
"""
from django.utils.translation import ugettext as _
from django.core.management.base import BaseCommand
try:
from django.contrib.auth import get_user_model
User = get_user_model()
except ImportError:
from django.contrib.auth.models import User
from helpdesk.models import UserSettings
from helpdesk.settings import DEFAULT_USER_SETTINGS
class Command(BaseCommand):
"create_usersettings command"
help = _('Check for user without django-helpdesk UserSettings '
'and create settings if required. Uses '
'settings.DEFAULT_USER_SETTINGS which can be overridden to '
'suit your situation.')
def handle(self, *args, **options):
"handle command line"
for u in User.objects.all():
try:
s = UserSettings.objects.get(user=u)
except UserSettings.DoesNotExist:
s = UserSettings(user=u, settings=DEFAULT_USER_SETTINGS)
s.save()