This commit is contained in:
glenux
parent
623d584382
commit
3eff1b83b9
2 changed files with 156 additions and 123 deletions
280
account.py
280
account.py
|
|
@ -1,137 +1,169 @@
|
||||||
#/usr/bin/env python
|
#/usr/bin/env python
|
||||||
|
# vim: set tw=72 ts=4 sw=4 st et sts si at:
|
||||||
|
|
||||||
import jabber
|
import jabber
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
def _gdisconnectHD(param=None):
|
def _gdisconnectHD(param=None):
|
||||||
if param:
|
if param:
|
||||||
print " Disconnecting : "+param
|
print " Disconnecting : "+param
|
||||||
return None
|
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
class Account:
|
class Account:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self._host=None
|
self._host = None
|
||||||
self._login=None
|
self._login = None
|
||||||
self._password=None
|
self._password = None
|
||||||
self._rosterHash={}
|
self._rosterHash = {}
|
||||||
self._proto2transport={}
|
self._proto2transport = {}
|
||||||
self._transport2proto={}
|
self._transport2proto = {}
|
||||||
self._cnx=None
|
self._cnx = None
|
||||||
|
|
||||||
def getDesc(self):
|
|
||||||
return (self._hlogin + "@" + self._host)
|
|
||||||
|
|
||||||
def setHost(self,hostName):
|
|
||||||
self._host=hostName
|
|
||||||
|
|
||||||
def setLogin(self,userName):
|
|
||||||
self._login=userName
|
|
||||||
|
|
||||||
def setPassword(self,passWord):
|
|
||||||
self._password=passWord
|
|
||||||
|
|
||||||
def setProtocol(self,protocolName,transportName):
|
|
||||||
self._proto2transport[protocolName]=transportName
|
|
||||||
|
|
||||||
def setTransport(self,transportName,protocolName):
|
|
||||||
self._transport2proto[transportName]=protocolName
|
|
||||||
|
|
||||||
def getTransport(self,protocolName):
|
|
||||||
if self._proto2transport.has_key(protocolName):
|
|
||||||
return self._proto2transport[protocolName]
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
|
|
||||||
def getProtocol(self,hostName):
|
|
||||||
if self._transport2proto.has_key(hostName):
|
|
||||||
return self._transport2proto[hostName]
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
|
|
||||||
def connect(self):
|
|
||||||
# se connecte et choppe le roster
|
|
||||||
## connexion au serveur
|
|
||||||
self._cnx=jabber.Client(host=self._host)
|
|
||||||
self._cnx.setDisconnectHandler(_gdisconnectHD)
|
|
||||||
if self._cnx:
|
|
||||||
try:
|
|
||||||
self._cnx.connect()
|
|
||||||
sys.stderr.write("* Connected to "+self._host+"\n")
|
|
||||||
## authentification
|
|
||||||
if self._login and self._password \
|
|
||||||
and self._cnx.auth(self._login, self._password, 'default'):
|
|
||||||
sys.stderr.write(" Authenticated\n")
|
|
||||||
else:
|
|
||||||
sys.stderr.write(" Not Athenticated. \n")
|
|
||||||
sys.stderr.write(" Disconnecting from "+self._host+"\n")
|
|
||||||
self._cnx=None
|
|
||||||
except IOError, e:
|
|
||||||
print e
|
|
||||||
self._cnx=None
|
|
||||||
|
|
||||||
def disconnect(self):
|
|
||||||
if self._cnx:
|
|
||||||
self._cnx.disconnect()
|
|
||||||
sys.stderr.write(" Disconnected from"+self._host+"\n")
|
|
||||||
|
|
||||||
|
|
||||||
def getAgents(self):
|
def getDesc(self):
|
||||||
if self._cnx:
|
return (self._hlogin + "@" + self._host)
|
||||||
agentsList=self._cnx.requestAgents()
|
|
||||||
sys.stderr.write(" Agents list downloaded:\n")
|
|
||||||
print agentsList
|
|
||||||
for agent in agentsList.keys():
|
|
||||||
if agent:
|
|
||||||
self.setProtocol(agentsList[agent]["service"],agent)
|
|
||||||
self.setTransport(agent,agentsList[agent]["service"])
|
|
||||||
print " - "+agent+" @ "+agentsList[agent]["service"]
|
|
||||||
return self._transport2proto
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
|
|
||||||
def getRoster(self):
|
|
||||||
if self._cnx:
|
|
||||||
roster=self._cnx.requestRoster()
|
|
||||||
sys.stderr.write(" Roster downloaded\n")
|
|
||||||
jid_list=roster.getJIDs()
|
|
||||||
for jid in jid_list:
|
|
||||||
transport=jid.getDomain()
|
|
||||||
node=jid.getNode()
|
|
||||||
if not self._rosterHash.has_key(transport):
|
|
||||||
self._rosterHash[transport]={}
|
|
||||||
if not self._rosterHash[transport].has_key(node):
|
|
||||||
self._rosterHash[transport][node]=1
|
|
||||||
return roster
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
|
|
||||||
def setRoster(self,rosterHash):
|
def setHost(self, hostName):
|
||||||
# on se connecte
|
self._host = hostName
|
||||||
if self._cnx:
|
|
||||||
for trans_proto in rosterHash.keys():
|
|
||||||
trans_host=self.getTransport(trans_proto)
|
def setLogin(self, userName):
|
||||||
if not trans_host: trans_host=trans_proto
|
self._login = userName
|
||||||
# pour chaque id connue dans ce proto
|
|
||||||
for proto_id in rosterHash[trans_proto]:
|
|
||||||
if proto_id:
|
def setPassword(self, passWord):
|
||||||
jid=jabber.JID()
|
self._password = passWord
|
||||||
jid.setNode(proto_id)
|
|
||||||
jid.setDomain(trans_host)
|
|
||||||
jid.setResource("")
|
def setProtocol(self, protocolName, transportName):
|
||||||
jid_name,jid_groups=rosterHash[trans_proto][proto_id]
|
self._proto2transport[protocolName] = transportName
|
||||||
sys.stdout.write(" -> "+proto_id)
|
|
||||||
sys.stdout.write(" @ "+trans_host)
|
|
||||||
sys.stdout.write(" = "+jid_name.encode('iso-8859-1','replace')+"...")
|
def setTransport(self, transportName, protocolName):
|
||||||
sys.stdout.flush()
|
self._transport2proto[transportName] = protocolName
|
||||||
if not(self._rosterHash.has_key(trans_host) \
|
|
||||||
and self._rosterHash[trans_host].has_key(proto_id)):
|
|
||||||
self._cnx.addRosterItem(jid)
|
def getTransport(self, protocolName):
|
||||||
sys.stdout.write(" [done]\n")
|
if self._proto2transport.has_key(protocolName):
|
||||||
else:
|
return self._proto2transport[protocolName]
|
||||||
#ne pas faire la mise a jour, le contact existe deja la-bas
|
|
||||||
sys.stdout.write(" [exists]\n")
|
else:
|
||||||
# on met d'abord a jour le nom
|
return None
|
||||||
self._cnx.updateRosterItem(jid,name=jid_name)
|
|
||||||
# puis le groupe
|
|
||||||
self._cnx.updateRosterItem(jid,name=jid_name,groups=jid_groups)
|
def getProtocol(self,hostName):
|
||||||
|
if self._transport2proto.has_key(hostName):
|
||||||
|
return self._transport2proto[hostName]
|
||||||
|
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def connect(self):
|
||||||
|
# se connecte et choppe le roster
|
||||||
|
## connexion au serveur
|
||||||
|
self._cnx = jabber.Client(host=self._host)
|
||||||
|
self._cnx.setDisconnectHandler(_gdisconnectHD)
|
||||||
|
if self._cnx:
|
||||||
|
try:
|
||||||
|
self._cnx.connect()
|
||||||
|
sys.stderr.write("* Connected to "+self._host+"\n")
|
||||||
|
## authentification
|
||||||
|
if self._login and self._password \
|
||||||
|
and self._cnx.auth(self._login, self._password, 'default'):
|
||||||
|
sys.stderr.write(" Authenticated\n")
|
||||||
|
|
||||||
|
else:
|
||||||
|
sys.stderr.write(" Not Athenticated. \n")
|
||||||
|
sys.stderr.write(" Disconnecting from " + self._host+"\n")
|
||||||
|
self._cnx = None
|
||||||
|
|
||||||
|
except IOError, e:
|
||||||
|
print e
|
||||||
|
self._cnx = None
|
||||||
|
|
||||||
|
|
||||||
|
def disconnect(self):
|
||||||
|
if self._cnx:
|
||||||
|
self._cnx.disconnect()
|
||||||
|
sys.stderr.write(" Disconnected from"+self._host+"\n")
|
||||||
|
|
||||||
|
|
||||||
|
def getAgents(self):
|
||||||
|
if self._cnx:
|
||||||
|
agentsList=self._cnx.requestAgents()
|
||||||
|
sys.stderr.write(" Agents list downloaded:\n")
|
||||||
|
print agentsList
|
||||||
|
for agent in agentsList.keys():
|
||||||
|
if agent:
|
||||||
|
self.setProtocol(agentsList[agent]["service"],agent)
|
||||||
|
self.setTransport(agent,agentsList[agent]["service"])
|
||||||
|
print " - "+agent+" @ "+agentsList[agent]["service"]
|
||||||
|
return self._transport2proto
|
||||||
|
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def getRoster(self):
|
||||||
|
if self._cnx:
|
||||||
|
roster=self._cnx.requestRoster()
|
||||||
|
sys.stderr.write(" Roster downloaded\n")
|
||||||
|
jid_list=roster.getJIDs()
|
||||||
|
for jid in jid_list:
|
||||||
|
transport=jid.getDomain()
|
||||||
|
node=jid.getNode()
|
||||||
|
if not self._rosterHash.has_key(transport):
|
||||||
|
self._rosterHash[transport]={}
|
||||||
|
|
||||||
|
if not self._rosterHash[transport].has_key(node):
|
||||||
|
self._rosterHash[transport][node]=1
|
||||||
|
return roster
|
||||||
|
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
def setRoster(self,rosterHash):
|
||||||
|
# on se connecte
|
||||||
|
if self._cnx:
|
||||||
|
for trans_proto in rosterHash.keys():
|
||||||
|
trans_host = self.getTransport(trans_proto)
|
||||||
|
if not trans_host:
|
||||||
|
trans_host = trans_proto
|
||||||
|
|
||||||
|
# pour chaque id connue dans ce proto
|
||||||
|
for proto_id in rosterHash[trans_proto]:
|
||||||
|
if proto_id:
|
||||||
|
jid = jabber.JID()
|
||||||
|
jid.setNode(proto_id)
|
||||||
|
jid.setDomain(trans_host)
|
||||||
|
jid.setResource("")
|
||||||
|
jid_name, jid_groups = rosterHash[trans_proto][proto_id]
|
||||||
|
sys.stdout.write(" -> " + proto_id)
|
||||||
|
sys.stdout.write(" @ " + trans_host)
|
||||||
|
sys.stdout.write(" = " + \
|
||||||
|
jid_name.encode('iso-8859-1','replace') + \
|
||||||
|
"...")
|
||||||
|
|
||||||
|
sys.stdout.flush()
|
||||||
|
if not(self._rosterHash.has_key(trans_host) \
|
||||||
|
and self._rosterHash[trans_host].has_key(proto_id)):
|
||||||
|
self._cnx.addRosterItem(jid)
|
||||||
|
|
||||||
|
sys.stdout.write(" [done]\n")
|
||||||
|
|
||||||
|
else:
|
||||||
|
#ne pas faire la mise a jour, le contact existe deja la-bas
|
||||||
|
sys.stdout.write(" [exists]\n")
|
||||||
|
# on met d'abord a jour le nom
|
||||||
|
|
||||||
|
self._cnx.updateRosterItem(jid, name = jid_name)
|
||||||
|
# puis le groupe
|
||||||
|
self._cnx.updateRosterItem(jid, \
|
||||||
|
name = jid_name, \
|
||||||
|
groups=jid_groups)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,5 @@
|
||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
|
# vim: set sw=4 ts=4 si et:
|
||||||
|
|
||||||
from xml.dom.ext import *
|
from xml.dom.ext import *
|
||||||
from xml.dom.ext.reader.Sax import *
|
from xml.dom.ext.reader.Sax import *
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue