david/jcl
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add bare JID as third argument to register form post functions

Browse Source 
darcs-hash:20070606184618-86b55-d2a169b118adbec03202b7f9508d809d09bebb03.gz
This commit is contained in:
David Rousselie
2007-06-06 20:46:18 +02:00
parent db4d7fd2ac
commit 3a85b455cf
3 changed files with 179 additions and 143 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/jcl/jabber/component.py
View File

@@ -677,8 +677,11 @@ class AccountManager(object):
"""Handle get_register for given account_class"""
info_query = info_query.make_result_response()
query = info_query.new_query("jabber:iq:register")
from_jid = info_query.get_from()
bare_from_jid = unicode(from_jid.bare())
self.get_reg_form(lang_class,
account_class).as_xml(query)
account_class,
bare_from_jid).as_xml(query)
return [info_query]
def account_type_get_register(self, info_query, account_type, lang_class):
@@ -726,6 +729,8 @@ class AccountManager(object):
"""Populate given account"""
field = None
result = []
from_jid = info_query.get_from()
bare_from_jid = unicode(from_jid.bare())
self.db_connect()
try:
for (field, field_type, field_options, field_post_func,
@@ -736,7 +741,8 @@ class AccountManager(object):
else:
value = None
setattr(_account, field,
field_post_func(value, field_default_func))
field_post_func(value, field_default_func,
bare_from_jid))
except FieldError, exception:
_account.destroySelf()
type, value, stack = sys.exc_info()
@@ -1037,7 +1043,7 @@ class AccountManager(object):
"""Delete connection associated to the current thread"""
del account.hub.threadConnection
def get_reg_form(self, lang_class, _account_class):
def get_reg_form(self, lang_class, _account_class, bare_from_jid):
"""Return register form based on language and account class
"""
reg_form = Form(title=lang_class.register_title,
@@ -1078,20 +1084,20 @@ class AccountManager(object):
field.add_option(label=label,
values=[option_value])
try:
post_func(None, default_func)
post_func(None, default_func, bare_from_jid)
except:
self.__logger.debug("Setting field " + field_name + " required")
field.required = True
## TODO : get default value if any
return reg_form
def get_reg_form_init(self, lang_class, _account):
"""Return register form for an existing account (update)
"""
reg_form = self.get_reg_form(lang_class, _account.__class__)
reg_form = self.get_reg_form(lang_class, _account.__class__,
_account.user_jid)
reg_form["name"].value = _account.name
reg_form["name"].type = "hidden"
for field in reg_form.fields: # TODO
for field in reg_form.fields:
if hasattr(_account, field.name):
field.value = getattr(_account, field.name)
return reg_form

84
src/jcl/model/account.py
View File

@@ -1,4 +1,4 @@
# -*- coding: UTF-8 -*-
# -*- coding: utf-8 -*-
##
## account.py
## Login : David Rousselie <dax@happycoders.org>
@@ -37,13 +37,13 @@ OFFLINE = "offline"
ONLINE = "online"
def default_post_func(field_value, default_func):
def default_post_func(field_value, default_func, bare_from_jid):
"""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, default_func):
def int_post_func(field_value, default_func, bare_from_jid):
"""Return an integer from integer field value"""
if field_value is None or str(field_value) == "":
return int(default_func())
@@ -197,42 +197,48 @@ class PresenceAccount(Account):
if real_class is None:
real_class = cls
return Account.get_register_fields(real_class) + \
[(None, None, None, None, None), \
("chat_action", "list-single", \
[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 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 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 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 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 get_possibles_actions("offline_action")], \
lambda action, default_func: is_action_possible("offline_action", \
action, \
default_func), \
[(None, None, None, None, None),
("chat_action", "list-single",
[str(action) for action in get_possibles_actions("chat_action")],
lambda action, default_func, bare_from_jid: \
is_action_possible("chat_action",
action,
default_func),
lambda : get_default_presence_action("chat_action")),
("online_action", "list-single",
[str(action) for action in get_possibles_actions("online_action")],
lambda action, default_func, bare_from_jid: \
is_action_possible("online_action",
action,
default_func),
lambda : get_default_presence_action("online_action")),
("away_action", "list-single",
[str(action) for action in get_possibles_actions("away_action")],
lambda action, default_func, bare_from_jid: \
is_action_possible("away_action",
action,
default_func),
lambda : get_default_presence_action("away_action")),
("xa_action", "list-single",
[str(action) for action in get_possibles_actions("xa_action")],
lambda action, default_func, bare_from_jid: \
is_action_possible("xa_action",
action,
default_func),
lambda : get_default_presence_action("xa_action")),
("dnd_action", "list-single",
[str(action) for action in get_possibles_actions("dnd_action")],
lambda action, default_func, bare_from_jid: \
is_action_possible("dnd_action",
action,
default_func),
lambda : get_default_presence_action("dnd_action")),
("offline_action", "list-single",
[str(action) for action in get_possibles_actions("offline_action")],
lambda action, default_func, bare_from_jid: \
is_action_possible("offline_action",
action,
default_func),
lambda : get_default_presence_action("offline_action"))]
get_register_fields = classmethod(_get_register_fields)

130
src/jcl/model/tests/account.py
View File

@@ -55,18 +55,20 @@ class ExampleAccount(Account):
if real_class is None:
real_class = cls
return Account.get_register_fields(real_class) + \
[("login", "text-single", None, \
lambda field_value, default_func: account.mandatory_field(field_value), \
lambda : ""), \
("password", "text-private", None, \
lambda field_value, default_func: password_post_func(field_value), \
lambda : ""), \
("store_password", "boolean", None, account.default_post_func, \
lambda : True), \
("test_enum", "list-single", ["choice1", "choice2", "choice3"], \
account.default_post_func, \
lambda : "choice2"), \
("test_int", "text-single", None, account.int_post_func, \
[("login", "text-single", None,
lambda field_value, default_func, bare_from_jid: \
account.mandatory_field(field_value),
lambda : ""),
("password", "text-private", None,
lambda field_value, default_func, bare_from_jid: \
password_post_func(field_value),
lambda : ""),
("store_password", "boolean", None, account.default_post_func,
lambda : True),
("test_enum", "list-single", ["choice1", "choice2", "choice3"],
account.default_post_func,
lambda : "choice2"),
("test_int", "text-single", None, account.int_post_func,
lambda : 44)]
get_register_fields = classmethod(_get_register_fields)
@@ -78,30 +80,30 @@ class Example2Account(Account):
if real_class is None:
real_class = cls
return Account.get_register_fields(real_class) + \
[("test_new_int", "text-single", None, account.int_post_func, \
[("test_new_int", "text-single", None, account.int_post_func,
lambda : 43)]
get_register_fields = classmethod(_get_register_fields)
class PresenceAccountExample(PresenceAccount):
DO_SOMETHING_ELSE = 2
possibles_actions = [PresenceAccount.DO_NOTHING, \
PresenceAccount.DO_SOMETHING, \
possibles_actions = [PresenceAccount.DO_NOTHING,
PresenceAccount.DO_SOMETHING,
DO_SOMETHING_ELSE]
def _get_presence_actions_fields(cls):
"""See PresenceAccount._get_presence_actions_fields
"""
return {'chat_action': (cls.possibles_actions, \
PresenceAccountExample.DO_SOMETHING_ELSE), \
'online_action': (cls.possibles_actions, \
PresenceAccountExample.DO_SOMETHING_ELSE), \
'away_action': (cls.possibles_actions, \
PresenceAccountExample.DO_SOMETHING_ELSE), \
'xa_action': (cls.possibles_actions, \
PresenceAccountExample.DO_SOMETHING_ELSE), \
'dnd_action': (cls.possibles_actions, \
PresenceAccountExample.DO_SOMETHING_ELSE), \
'offline_action': (cls.possibles_actions, \
return {'chat_action': (cls.possibles_actions,
PresenceAccountExample.DO_SOMETHING_ELSE),
'online_action': (cls.possibles_actions,
PresenceAccountExample.DO_SOMETHING_ELSE),
'away_action': (cls.possibles_actions,
PresenceAccountExample.DO_SOMETHING_ELSE),
'xa_action': (cls.possibles_actions,
PresenceAccountExample.DO_SOMETHING_ELSE),
'dnd_action': (cls.possibles_actions,
PresenceAccountExample.DO_SOMETHING_ELSE),
'offline_action': (cls.possibles_actions,
PresenceAccountExample.DO_SOMETHING_ELSE)}
get_presence_actions_fields = classmethod(_get_presence_actions_fields)
@@ -112,50 +114,68 @@ class PresenceAccountExample(PresenceAccount):
if real_class is None:
real_class = cls
return PresenceAccount.get_register_fields(real_class) + \
[("test_new_int", "text-single", None, account.int_post_func, \
[("test_new_int", "text-single", None, account.int_post_func,
lambda : 43)]
get_register_fields = classmethod(_get_register_fields)
class AccountModule_TestCase(unittest.TestCase):
def test_default_post_func(self):
result = account.default_post_func("test", None)
result = account.default_post_func("test", None, "user1@jcl.test.com")
self.assertEquals(result, "test")
def test_default_post_func_default_value(self):
result = account.default_post_func("", lambda : "test")
result = account.default_post_func("", lambda : "test", "user1@jcl.test.com")
self.assertEquals(result, "test")
def test_default_post_func_default_value2(self):
result = account.default_post_func(None, lambda : "test")
result = account.default_post_func(None, lambda : "test", "user1@jcl.test.com")
self.assertEquals(result, "test")
def test_int_post_func(self):
result = account.int_post_func("42", None)
result = account.int_post_func("42", None, "user1@jcl.test.com")
self.assertEquals(result, 42)
def test_int_post_func_default_value(self):
result = account.int_post_func("", lambda : 42)
result = account.int_post_func("", lambda : 42, "user1@jcl.test.com")
self.assertEquals(result, 42)
def test_int_post_func_default_value(self):
result = account.int_post_func(None, lambda : 42)
result = account.int_post_func(None, lambda : 42, "user1@jcl.test.com")
self.assertEquals(result, 42)
def test_mandatory_field_empty(self):
self.assertRaises(FieldError, \
account.mandatory_field, \
self.assertRaises(FieldError,
account.mandatory_field,
"")
def test_mandatory_field_none(self):
self.assertRaises(FieldError, \
account.mandatory_field, \
self.assertRaises(FieldError,
account.mandatory_field,
None)
def test_mandatory_field_empty(self):
self.assertEquals(account.mandatory_field("value"), \
self.assertEquals(account.mandatory_field("value"),
"value")
class Account_TestCase(unittest.TestCase):
class InheritableAccount_TestCase(unittest.TestCase):
def test_get_register_fields(self):
"""Check if post functions and default functions execute correctly.
To be validated this test only need to be executed without any
exception.
"""
for (field_name,
field_type,
field_options,
field_post_func,
field_default_func) in self.account_class.get_register_fields():
if field_name is not None:
try:
field_post_func(field_default_func(), field_default_func, "user1@jcl.test.com")
except FieldError, error:
# this type of error is OK
pass
class Account_TestCase(InheritableAccount_TestCase):
def setUp(self):
if os.path.exists(DB_PATH):
os.unlink(DB_PATH)
@@ -163,6 +183,7 @@ class Account_TestCase(unittest.TestCase):
Account.createTable(ifNotExists = True)
ExampleAccount.createTable(ifNotExists = True)
del account.hub.threadConnection
self.account_class = Account
def tearDown(self):
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
@@ -176,8 +197,8 @@ class Account_TestCase(unittest.TestCase):
def test_set_status(self):
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
account11 = Account(user_jid = "test1@test.com", \
name = "account11", \
account11 = Account(user_jid="test1@test.com",
name="account11",
jid="account11@jcl.test.com")
account11.status = account.OFFLINE
self.assertEquals(account11.status, account.OFFLINE)
@@ -186,13 +207,13 @@ class Account_TestCase(unittest.TestCase):
def test_set_status_live_password(self):
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
account11 = ExampleAccount(user_jid = "test1@test.com", \
name = "account11", \
jid = "account11@jcl.test.com", \
login = "mylogin", \
password = "mypassword", \
store_password = False, \
test_enum = "choice3", \
account11 = ExampleAccount(user_jid="test1@test.com",
name="account11",
jid="account11@jcl.test.com",
login="mylogin",
password="mypassword",
store_password=False,
test_enum="choice3",
test_int=21)
account11.waiting_password_reply = True
account11.status = account.OFFLINE
@@ -201,7 +222,7 @@ class Account_TestCase(unittest.TestCase):
self.assertEquals(account11.password, None)
del account.hub.threadConnection
class PresenceAccount_TestCase(unittest.TestCase):
class PresenceAccount_TestCase(InheritableAccount_TestCase):
def setUp(self):
if os.path.exists(DB_PATH):
os.unlink(DB_PATH)
@@ -231,9 +252,11 @@ class PresenceAccount_TestCase(unittest.TestCase):
fields = self.account_class.get_presence_actions_fields()
self.assertEquals(len(fields), 6)
(possibles_actions, chat_default_action) = fields["chat_action"]
self.assertEquals(possibles_actions, self.account_class.possibles_actions)
self.assertEquals(possibles_actions,
self.account_class.possibles_actions)
(possibles_actions, online_default_action) = fields["online_action"]
self.assertEquals(possibles_actions, self.account_class.possibles_actions)
self.assertEquals(possibles_actions,
self.account_class.possibles_actions)
def test_possibles_actions(self):
for (field_name, \
@@ -243,12 +266,13 @@ class PresenceAccount_TestCase(unittest.TestCase):
default_func) in self.account_class.get_register_fields()[1:]:
if possibles_actions is not None:
for possible_action in possibles_actions:
self.assertEquals(post_func(possible_action, default_func),
self.assertEquals(post_func(possible_action, default_func,
"user1@jcl.test.com"),
int(possible_action))
self.assertTrue(str(default_func()) in possibles_actions)
else:
try:
post_func("42", default_func)
post_func("42", default_func, "user1@jcl.test.com")
default_func()
except FieldError, error:
pass