account default_func and post_func refactoring

for registration fields, default_func now returns default_value
and post_func include test for mandatory fields.

darcs-hash:20070322161501-86b55-75b6d57fef3ba0df6c6c80a487c9afcdab2d40c3.gz
This commit is contained in:
David Rousselie
2007-03-22 17:15:01 +01:00
parent 6753545187
commit 101e74f6c7
6 changed files with 189 additions and 72 deletions

View File

@@ -426,29 +426,34 @@ class JCLComponent(Component, object):
if accounts_count >= 1:
_account = list(accounts)[0]
else:
if info_query.get_to().node is None:
if info_query.get_to().resource is None:
if len(self.account_classes) > 1:
self.__logger.error("There should be only one account class declared")
self.__logger.error("There might be only one account class declared")
new_account_class = self.account_classes[0]
else:
new_account_class = self._get_account_class(info_query.get_to().node + "Account")
new_account_class = self._get_account_class(info_query.get_to().resource + "Account")
_account = new_account_class(user_jid = unicode(base_from_jid), \
name = name, \
jid = name + u"@" + unicode(self.jid))
field = None
try:
self.__logger.debug("Populating " + str(_account))
for (field, field_type, field_options, field_post_func, \
field_default_func) in _account.get_register_fields():
self.__logger.debug("Testing " + field)
if field is not None:
if field in x_data:
setattr(_account, field, \
field_post_func(x_data[field].value))
value = x_data[field].value
else:
setattr(_account, field, \
field_default_func(field))
value = None
setattr(_account, field, \
field_post_func(value, field_default_func))
except FieldError, exception:
_account.destroySelf()
self.__logger.error(str(exception))
type, value, stack = sys.exc_info()
self.__logger.error("Error while populating account: %s\n%s" % \
(exception, "".join(traceback.format_exception
(type, value, stack, 5))))
iq_error = info_query.make_error_response("not-acceptable")
text = iq_error.get_error().xmlnode.newTextChild(None, \
"text", \
@@ -659,6 +664,7 @@ class JCLComponent(Component, object):
if _account_class.__name__.lower() == account_class_name.lower():
self.__logger.debug(account_class_name + " found")
return _account_class
self.__logger.debug(account_class_name + " not found")
return None
def _list_accounts(self, disco_items, _account_class, base_from_jid, account_type = ""):
@@ -728,6 +734,7 @@ class JCLComponent(Component, object):
% (exception))
self.stream.send(msg)
type, value, stack = sys.exc_info()
# TODO : not checking email here
self.__logger.debug("Error while first checking mail : %s\n%s" \
% (exception, "".join(traceback.format_exception
(type, value, stack, 5))))
@@ -748,7 +755,11 @@ class JCLComponent(Component, object):
name = "name", \
required = True)
for (field_name, field_type, field_options, post_func, default_func) in \
for (field_name, \
field_type, \
field_options, \
post_func, \
default_func) in \
_account_class.get_register_fields():
if field_name is None:
# TODO : Add page when empty tuple given
@@ -760,9 +771,11 @@ class JCLComponent(Component, object):
label = getattr(lang_class, lang_label_attr)
else:
label = field_name
self.__logger.debug("Adding field " + field_name + " to registration form")
field = reg_form.add_field(field_type = field_type, \
label = label, \
name = field_name)
name = field_name, \
value = default_func())
if field_options is not None:
for option_value in field_options:
lang_label_attr = _account_class.__name__.lower() \
@@ -773,7 +786,10 @@ class JCLComponent(Component, object):
label = option_value
field.add_option(label = label, \
values = [option_value])
if default_func == account.mandatory_field:
try:
post_func(None, default_func)
except:
self.__logger.debug("Setting field " + field_name + " required")
field.required = True
## TODO : get default value if any
return reg_form

View File

@@ -37,25 +37,25 @@ OFFLINE = "offline"
ONLINE = "online"
def default_post_func(field_value):
def default_post_func(field_value, default_func):
"""Default post process function: do nothing"""
if field_value is None or str(field_value) == "":
return default_func()
return field_value
def int_post_func(field_value):
def int_post_func(field_value, default_func):
"""Return an integer from integer field value"""
if field_value is None or str(field_value) == "":
return int(default_func())
return int(field_value)
def string_not_null_post_func(field_value):
"""Post process function for not null/empty string"""
if field_value is None or field_value == "":
def mandatory_field(field_name, field_value, default_func):
"""Used as default function for field that must be specified
and cannot have an empty value"""
if field_value is None or str(field_value) == "":
raise FieldError # TODO : add translated message
return field_value
def mandatory_field(field_name):
"""Used as default function for field that must be specified
and cannot have default value"""
raise FieldError # TODO : add translated message
# create a hub to attach a per thread connection
hub = ConnectionHub()
@@ -119,8 +119,7 @@ class Account(InheritableSQLObject):
'list-single', ...
- field_options:
- field_post_func: function called to process received field
- field_default_func: function to return default value (or error if
field is mandatory)
- field_default_func: function to return default value
"""
return [] # "name" field is mandatory
@@ -184,32 +183,55 @@ class PresenceAccount(Account):
def _get_register_fields(cls):
""" See Account._get_register_fields """
def is_action_possible(action):
if int(action) in cls.possibles_actions:
return int(action)
raise FieldError # TODO : add translated message
def get_possibles_actions(presence_action_field):
return cls.get_presence_actions_fields()[presence_action_field][0]
# TODO : check is_action_possible with presence_actions_fields (see partial eval function)
def is_action_possible(presence_action_field, action, default_func):
if int(action) in get_possibles_actions(presence_action_field):
return int(action)
raise default_func()
def get_default_presence_action(presence_action_field):
return cls.get_presence_actions_fields()[presence_action_field][1]
return Account.get_register_fields() + \
[(None, None, None, None, None), \
("chat_action", "list-single", \
[str(action) for action in cls.possibles_actions], \
is_action_possible, mandatory_field), \
[str(action) for action in get_possibles_actions("chat_action")], \
lambda action, default_func: is_action_possible("chat_action", \
action, \
default_func), \
lambda : get_default_presence_action("chat_action")), \
("online_action", "list-single", \
[str(action) for action in cls.possibles_actions], \
is_action_possible, mandatory_field), \
[str(action) for action in get_possibles_actions("online_action")], \
lambda action, default_func: is_action_possible("online_action", \
action, \
default_func), \
lambda : get_default_presence_action("online_action")), \
("away_action", "list-single", \
[str(action) for action in cls.possibles_actions], \
is_action_possible, mandatory_field), \
[str(action) for action in get_possibles_actions("away_action")], \
lambda action, default_func: is_action_possible("away_action", \
action, \
default_func), \
lambda : get_default_presence_action("away_action")), \
("xa_action", "list-single", \
[str(action) for action in cls.possibles_actions], \
is_action_possible, mandatory_field), \
[str(action) for action in get_possibles_actions("xa_action")], \
lambda action, default_func: is_action_possible("xa_action", \
action, \
default_func), \
lambda : get_default_presence_action("xa_action")), \
("dnd_action", "list-single", \
[str(action) for action in cls.possibles_actions], \
is_action_possible, mandatory_field), \
[str(action) for action in get_possibles_actions("dnd_action")], \
lambda action, default_func: is_action_possible("dnd_action", \
action, \
default_func), \
lambda : get_default_presence_action("dnd_action")), \
("offline_action", "list-single", \
[str(action) for action in cls.possibles_actions], \
is_action_possible, mandatory_field)]
[str(action) for action in get_possibles_actions("offline_action")], \
lambda action, default_func: is_action_possible("offline_action", \
action, \
default_func), \
lambda : get_default_presence_action("offline_action"))]
get_register_fields = classmethod(_get_register_fields)