From 590b5a20d979ce0561e1a1895bb691e9f906168c Mon Sep 17 00:00:00 2001 From: Bruno Clermont Date: Fri, 15 Apr 2011 17:38:12 -0400 Subject: [PATCH] add command to create UserSettings for user who do not have one yet --- .../commands/create_usersettings.py | 27 +++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 helpdesk/management/commands/create_usersettings.py diff --git a/helpdesk/management/commands/create_usersettings.py b/helpdesk/management/commands/create_usersettings.py new file mode 100644 index 00000000..2d691faf --- /dev/null +++ b/helpdesk/management/commands/create_usersettings.py @@ -0,0 +1,27 @@ +# -*- coding: utf-8 -*- + +"Management command to add create UserSettings" + +from django.utils.translation import ugettext as _ +from django.core.management.base import BaseCommand +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 if missing') + + 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() + +# vim: tabstop=4 expandtab shiftwidth=4 softtabstop=4 \ No newline at end of file