mirror of
https://github.com/nikdoof/test-auth.git
synced 2025-12-14 14:52:15 +00:00
Andrew Williams
f82e211d91
Old service access methods have been removed, XMLRPC is the prefered choice due to speed and reliability of connection. Also as a bonus the Jabber API now sets the person's nickname to the same as their character name.
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
import xmlrpclib
|
|
from sso.services import BaseService
|
|
import settings
|
|
|
|
|
|
class JabberService(BaseService):
|
|
|
|
settings = { 'require_user': True,
|
|
'require_password': True,
|
|
'provide_login': False }
|
|
|
|
@staticmethod
|
|
def exec_xmlrpc(func, **kwargs):
|
|
""" Send a XMLRPC request """
|
|
server = xmlrpclib.Server(settings.JABBER_XMLRPC_URL)
|
|
params = {}
|
|
for i in kwargs:
|
|
params[i] = kwargs[i]
|
|
|
|
return getattr(server, func)(params)
|
|
|
|
def add_user(self, username, password, **kwargs):
|
|
""" Add user to service """
|
|
res = self.exec_xmlrpc('register', user=username, host=settings.JABBER_SERVER, password=password)
|
|
if res['res'] == 0:
|
|
if 'character' in kwargs:
|
|
self.exec_xmlrpc('set_nickname', user=username, host=settings.JABBER_SERVER, nickname=kwargs['character'].name)
|
|
return res['text'].split(" ")[1]
|
|
else:
|
|
return False
|
|
|
|
def delete_user(self, uid):
|
|
""" Delete a user """
|
|
username, server = uid.split("@")
|
|
res = self.exec_xmlrpc('unregister', user=username, host=server)
|
|
if res['res'] == 0:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def check_user(self, username):
|
|
""" Check if the username exists """
|
|
res = self.exec_xmlrpc('check_account', user=username, host=settings.JABBER_SERVER)
|
|
if res['res'] == 0:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def disable_user(self, uid):
|
|
""" Disable a user """
|
|
username, server = uid.split("@")
|
|
res = self.exec_xmlrpc('ban_account', host=server, user=username, reason='Auth account disable')
|
|
if res['res'] == 0:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def enable_user(self, uid, password):
|
|
""" Enable a user """
|
|
self.reset_password(uid, password)
|
|
|
|
def reset_password(self, uid, password):
|
|
""" Reset the user's password """
|
|
username, server = uid.split("@")
|
|
res = self.exec_xmlrpc('change_password', user=username, host=server, newpass=password)
|
|
if res['res'] == 0:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
ServiceClass = 'JabberService'
|