1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
#!/usr/bin/python
"""Simple Jabber client that sends disco#info requests."""
import ConfigParser
import sys
import gobject
import pyxmpp.jid
import pyxmpp.jabber.client
import pyxmpp.iq
import libxml2
loop = None
def indent(s):
return ''.join([' %s\n' % line for line in s.splitlines()])
def xml_pp(s):
node = libxml2.parseDoc(s).getRootElement()
print indent(node.serialize(format=1)).rstrip()
class JabberClient(pyxmpp.jabber.client.JabberClient):
def __init__(self, query_jid, query_node, **kw):
self.query_jid = query_jid
self.query_node = query_node
pyxmpp.jabber.client.JabberClient.__init__(self, **kw)
gobject.idle_add(self.cb_connect)
gobject.idle_add(self.cb_idle)
def cb_connect(self):
self.connect()
socket = self.get_socket()
flags = gobject.IO_IN | gobject.IO_ERR | gobject.IO_HUP
gobject.io_add_watch(socket, flags, self.cb_io)
gobject.timeout_add(500, self.cb_send_iq)
def cb_send_iq(self):
iq = pyxmpp.iq.Iq(
from_jid=self.stream.my_jid,
to_jid=self.query_jid,
stanza_type='get')
iq.new_query('http://jabber.org/protocol/disco#info')
if self.query_node:
query = iq.get_query()
query.setProp('node', query_node)
print 'sending:'
xml_pp(iq.serialize())
self.stream.set_response_handlers(
iq, self.cb_iq_result, self.cb_iq_error, self.cb_iq_timeout, 10)
self.stream.send(iq)
return False
def cb_idle(self):
self.idle()
return True
def cb_io(self, fd, condition):
stream = self.get_stream()
stream.process()
return True
def cb_iq_result(self, stanza):
print 'result:'
xml_pp(stanza.serialize())
loop.quit()
def cb_iq_error(self, stanza):
print 'error:'
xml_pp(stanza.serialize())
loop.quit()
def cb_iq_timeout(self, arg1, arg2):
print 'timeout'
loop.quit()
def read_config(path):
parser = ConfigParser.SafeConfigParser()
parser.read(path)
jid = parser.get('jabber', 'jid')
password = parser.get('jabber', 'password')
return jid, password
if __name__ == '__main__':
loop = gobject.MainLoop()
jid, password = read_config('config')
if len(sys.argv) > 2:
query_node = sys.argv[2]
else:
query_node = None
client = JabberClient(
query_jid=sys.argv[1],
query_node=query_node,
jid=pyxmpp.jid.JID(jid),
password=password)
loop.run()
|
