diff --git a/openslides/utils/user/__init__.py b/openslides/utils/user/__init__.py index 0dbc44ce4..36c0e9b86 100644 --- a/openslides/utils/user/__init__.py +++ b/openslides/utils/user/__init__.py @@ -1,164 +1,19 @@ #!/usr/bin/env python # -*- coding: utf-8 -*- """ - openslides.utils.user.api - ~~~~~~~~~~~~~~~~~~~~~~~~~ + openslides.utils.user + ~~~~~~~~~~~~~~~~~~~~~ - Useful functions for the user object. + User api for OpenSlides :copyright: 2011, 2012 by OpenSlides team, see AUTHORS. :license: GNU GPL, see LICENSE for more details. """ -from django.db import models -from django import forms - from openslides.utils.user.signals import receiv_users - - -class UserChoices(object): - def __init__(self, field): - self.field = field - - def __iter__(self): - if self.field.empty_label is not None: - yield (u"", self.field.empty_label) - for user in Users(): - yield (user.uid, user) - - -class UserFormField(forms.fields.ChoiceField): - def __init__(self, required=True, initial=None, empty_label=u"---------", *args, **kwargs): - if required and (initial is not None): - self.empty_label = None - else: - self.empty_label = empty_label - forms.fields.Field.__init__(self, required=required, initial=initial, *args, **kwargs) - self.widget.choices = self.choices - - def __deepcopy__(self, memo): - result = super(forms.fields.ChoiceField, self).__deepcopy__(memo) - return result - - - def _get_choices(self): - # If self._choices is set, then somebody must have manually set - # the property self.choices. In this case, just return self._choices. - if hasattr(self, '_choices'): - return self._choices - return UserChoices(self) - - choices = property(_get_choices, forms.fields.ChoiceField._set_choices) - - def to_python(self, value): - return get_user(value) - - def valid_value(self, value): - return super(UserFormField, self).valid_value(value.uid) - - -class MultipleUserFormField(UserFormField): - widget = forms.widgets.SelectMultiple - - def __init__(self, *args, **kwargs): - super(MultipleUserFormField, self).__init__(empty_label=None, *args, **kwargs) - - def to_python(self, value): - if hasattr(value, '__iter__'): - return [super(MultipleUserFormField, self).to_python(v) for v in value] - return super(MultipleUserFormField, self).to_python(value) - - def valid_value(self, value): - if hasattr(value, '__iter__'): - return [super(MultipleUserFormField, self).valid_value(v) for v in value] - return super(MultipleUserFormField, self).valid_value(value) - - -class UserField(models.fields.Field): - __metaclass__ = models.SubfieldBase - - def __init__(self, *args, **kwargs): - super(UserField, self).__init__(max_length=255, *args, **kwargs) - # TODO: Validate the uid - - def get_internal_type(self): - return "CharField" - - def to_python(self, value): - """ - Convert string value to a User Object. - """ - if hasattr(value, 'uid'): - user = value - else: - user = get_user(value) - - user.prepare_database_save = lambda unused: UserField().get_prep_value(user) - return user - - def get_prep_value(self, value): - return value.uid - - def value_to_string(self, obj): - value = self._get_val_from_obj(obj) - return self.get_prep_value(value) - - def formfield(self, **kwargs): - defaults = {'form_class': UserFormField} - defaults.update(kwargs) - return super(UserField, self).formfield(**defaults) - - -def split_uid(uid): - data = uid.split(':', 1) - if len(data) == 2 and data[0] and data[1]: - return data - raise TypeError("Invalid uid: '%s'" % uid) - - -def get_user(uid): - try: - user_prefix, id = split_uid(uid) - except TypeError: - return EmtyUser() - - return Users(user_prefix=user_prefix, id=id)[0] - - -class Users(object): - """ - A Storage for a multiplicity of different User-Objects. - """ - def __init__(self, user_prefix=None, id=None): - self.user_prefix = user_prefix - self.id = id - - def __iter__(self): - for receiver, users in receiv_users.send(sender='users', user_prefix=self.user_prefix, id=self.id): - for user in users: - yield user - - def __getitem__(self, key): - user_list = list(self) - return user_list[key] - - -def generate_uid(prefix, id): - if ':' in prefix: - raise ValueError("':' is not allowed in a the 'user_prefix'") - return "%s:%d" % (prefix, id) - - -class UserMixin(object): - @property - def uid(self): - try: - return generate_uid(self.user_prefix, self.pk) - except AttributeError: - raise AttributeError("%s has to have a attribute 'user_prefix'" % self) - - def __repr__(self): - return 'User: %s' % self.uid +from openslides.utils.user.api import generate_uid, split_uid, get_user, Users +from openslides.utils.user.forms import UserFormField, MultipleUserFormField +from openslides.utils.user.model import UserField, UserMixin class EmtyUser(UserMixin): diff --git a/openslides/utils/user/api.py b/openslides/utils/user/api.py new file mode 100644 index 000000000..13976aad8 --- /dev/null +++ b/openslides/utils/user/api.py @@ -0,0 +1,53 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +""" + openslides.utils.user.api + ~~~~~~~~~~~~~~~~~~~~~~~~~ + + Usefull functions for the OpenSlides user api. + + :copyright: 2011, 2012 by OpenSlides team, see AUTHORS. + :license: GNU GPL, see LICENSE for more details. +""" + +from openslides.utils.user.signals import receiv_users + +class Users(object): + """ + A Storage for a multiplicity of different User-Objects. + """ + def __init__(self, user_prefix=None, id=None): + self.user_prefix = user_prefix + self.id = id + + def __iter__(self): + for receiver, users in receiv_users.send( + sender='users', user_prefix=self.user_prefix, id=self.id): + for user in users: + yield user + + def __getitem__(self, key): + user_list = list(self) + return user_list[key] + + +def generate_uid(prefix, id): + if ':' in prefix: + raise ValueError("':' is not allowed in a the 'user_prefix'") + return "%s:%d" % (prefix, id) + + +def split_uid(uid): + data = uid.split(':', 1) + if len(data) == 2 and data[0] and data[1]: + return data + raise TypeError("Invalid uid: '%s'" % uid) + + +def get_user(uid): + try: + user_prefix, id = split_uid(uid) + except TypeError: + return EmtyUser() + + return Users(user_prefix=user_prefix, id=id)[0] diff --git a/openslides/utils/user/forms.py b/openslides/utils/user/forms.py new file mode 100644 index 000000000..436517a82 --- /dev/null +++ b/openslides/utils/user/forms.py @@ -0,0 +1,77 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +""" + openslides.utils.user.forms + ~~~~~~~~~~~~~~~~~~~~~~~~~~~ + + Forms and FormFields for the OpenSlides user api. + + :copyright: 2011, 2012 by OpenSlides team, see AUTHORS. + :license: GNU GPL, see LICENSE for more details. +""" + +from django import forms + +from openslides.utils.user.api import Users, get_user + +class UserChoices(object): + def __init__(self, field): + self.field = field + + def __iter__(self): + if self.field.empty_label is not None: + yield (u"", self.field.empty_label) + for user in Users(): + yield (user.uid, user) + + +class UserFormField(forms.fields.ChoiceField): + def __init__(self, required=True, initial=None, empty_label=u"---------", + *args, **kwargs): + if required and (initial is not None): + self.empty_label = None + else: + self.empty_label = empty_label + forms.fields.Field.__init__(self, required=required, initial=initial, + *args, **kwargs) + self.widget.choices = self.choices + + def __deepcopy__(self, memo): + result = super(forms.fields.ChoiceField, self).__deepcopy__(memo) + return result + + + def _get_choices(self): + # If self._choices is set, then somebody must have manually set + # the property self.choices. In this case, just return self._choices. + if hasattr(self, '_choices'): + return self._choices + return UserChoices(self) + + choices = property(_get_choices, forms.fields.ChoiceField._set_choices) + + def to_python(self, value): + return get_user(value) + + def valid_value(self, value): + return super(UserFormField, self).valid_value(value.uid) + + +class MultipleUserFormField(UserFormField): + widget = forms.widgets.SelectMultiple + + def __init__(self, *args, **kwargs): + super(MultipleUserFormField, self).__init__(empty_label=None, + *args, **kwargs) + + def to_python(self, value): + if hasattr(value, '__iter__'): + return [super(MultipleUserFormField, self).to_python(v) + for v in value] + return super(MultipleUserFormField, self).to_python(value) + + def valid_value(self, value): + if hasattr(value, '__iter__'): + return [super(MultipleUserFormField, self).valid_value(v) + for v in value] + return super(MultipleUserFormField, self).valid_value(value) diff --git a/openslides/utils/user/model.py b/openslides/utils/user/model.py new file mode 100644 index 000000000..5c461da88 --- /dev/null +++ b/openslides/utils/user/model.py @@ -0,0 +1,65 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +""" + openslides.utils.user.models + ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + + Models and ModelFields for the OpenSlides user api. + + :copyright: 2011, 2012 by OpenSlides team, see AUTHORS. + :license: GNU GPL, see LICENSE for more details. +""" + +from django.db import models + +from openslides.utils.user.forms import UserFormField +from openslides.utils.user.api import get_user, generate_uid + + +class UserField(models.fields.Field): + __metaclass__ = models.SubfieldBase + + def __init__(self, *args, **kwargs): + super(UserField, self).__init__(max_length=255, *args, **kwargs) + # TODO: Validate the uid + + def get_internal_type(self): + return "CharField" + + def to_python(self, value): + """ + Convert string value to a User Object. + """ + if hasattr(value, 'uid'): + user = value + else: + user = get_user(value) + + user.prepare_database_save = lambda unused: UserField() \ + .get_prep_value(user) + return user + + def get_prep_value(self, value): + return value.uid + + def value_to_string(self, obj): + value = self._get_val_from_obj(obj) + return self.get_prep_value(value) + + def formfield(self, **kwargs): + defaults = {'form_class': UserFormField} + defaults.update(kwargs) + return super(UserField, self).formfield(**defaults) + + +class UserMixin(object): + @property + def uid(self): + try: + return generate_uid(self.user_prefix, self.pk) + except AttributeError: + raise AttributeError("%s has to have a attribute 'user_prefix'" + % self) + + def __repr__(self): + return 'User: %s' % self.uid diff --git a/openslides/utils/user/models.py b/openslides/utils/user/models.py new file mode 100644 index 000000000..31a8a57c7 --- /dev/null +++ b/openslides/utils/user/models.py @@ -0,0 +1,73 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +""" + openslides.utils.user.forms + ~~~~~~~~~~~~~~~~~~~~~~~~~~~ + + Forms and FormFields for the OpenSlides user api + + :copyright: 2011, 2012 by OpenSlides team, see AUTHORS. + :license: GNU GPL, see LICENSE for more details. +""" + +class UserChoices(object): + def __init__(self, field): + self.field = field + + def __iter__(self): + if self.field.empty_label is not None: + yield (u"", self.field.empty_label) + for user in Users(): + yield (user.uid, user) + + +class UserFormField(forms.fields.ChoiceField): + def __init__(self, required=True, initial=None, empty_label=u"---------", + *args, **kwargs): + if required and (initial is not None): + self.empty_label = None + else: + self.empty_label = empty_label + forms.fields.Field.__init__(self, required=required, initial=initial, + *args, **kwargs) + self.widget.choices = self.choices + + def __deepcopy__(self, memo): + result = super(forms.fields.ChoiceField, self).__deepcopy__(memo) + return result + + + def _get_choices(self): + # If self._choices is set, then somebody must have manually set + # the property self.choices. In this case, just return self._choices. + if hasattr(self, '_choices'): + return self._choices + return UserChoices(self) + + choices = property(_get_choices, forms.fields.ChoiceField._set_choices) + + def to_python(self, value): + return get_user(value) + + def valid_value(self, value): + return super(UserFormField, self).valid_value(value.uid) + + +class MultipleUserFormField(UserFormField): + widget = forms.widgets.SelectMultiple + + def __init__(self, *args, **kwargs): + super(MultipleUserFormField, self).__init__(empty_label=None, + *args, **kwargs) + + def to_python(self, value): + if hasattr(value, '__iter__'): + return [super(MultipleUserFormField, self).to_python(v) + for v in value] + return super(MultipleUserFormField, self).to_python(value) + + def valid_value(self, value): + if hasattr(value, '__iter__'): + return [super(MultipleUserFormField, self).valid_value(v) + for v in value] + return super(MultipleUserFormField, self).valid_value(value)