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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
#!/usr/bin/python
# -*- coding:utf-8 -*-
#
# Copyright 2010 Le Coz Florent <louizatakk@fedoraproject.org>
#
# This file is part of Poezio.
#
# Poezio is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# Poezio is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Poezio. If not, see <http://www.gnu.org/licenses/>.
import sys
import xmpp
from config import config
from logging import log
from threading import Thread
from multiuserchat import MultiUserChat
from handler import Handler
class Connection(Thread):
"""
Handles all network transactions
"""
def __init__(self, server, resource):
Thread.__init__(self)
self.handler = Handler()
self.server = server
self.resource = resource
self.online = 0 # 1:connected, 2:auth confirmed
self.jid = '' # we don't know our jid yet (anon account)
if not self.server:
log.error('You should set a server in the configuration file')
self.port = int(config.get('port'))
if not self.port:
log.warning('No port set in configuration file, defaulting to 5222')
self.port = 5222
def run(self):
"""
connect to server
"""
self.client = xmpp.Client(self.server, debug=[])
if not self.connect_to_server(self.server, self.port):
log.error('Could not connect to server')
sys.exit(-1)
if not self.authenticate():
log.error('Could not authenticate to server')
sys.exit(-1)
self.client.sendInitPresence()
self.online = 1 # 2 when confirmation of auth is received
self.register_handlers()
self.muc = MultiUserChat(self.client)
while 1:
self.process()
def connect_to_server(self, server, port):
# TODO proxy stuff
return self.client.connect((server, port))
def authenticate(self, anon=True):
if anon:
return self.client.auth(None, None, self.resource)
else:
log.error('Non-anonymous connections not handled currently')
return None
def register_handlers(self):
self.client.RegisterHandler('message', self.handler_message)
self.client.RegisterHandler('presence', self.handler_presence)
self.client.RegisterHandler('iq', self.handler_iq)
def handler_message(self, connection, message):
# body = message.getTag('body').getData()
# fro = str(message.getAttr('from'))
# room, nick = fro.split('/')
# print 'Message from %s in %s :' % (nick, room), body
self.handler.emit('xmpp-message-handler', message=message)
def handler_presence(self, connection, presence):
affil, role = u'', u''
fro = presence.getFrom()#presence.getAttr('from')
room_from, nick_from = fro.getStripped().encode('utf-8'), fro.getResource().encode('utf-8')
to = presence.getAttr('to')
room_to, nick_to = to.getStripped().encode('utf-8'), to.getResource().encode('utf-8')
if fro == to: # own presence
self.online = 2
self.jid = to
print 'Authentification confirmation received!'
return
for x in presence.getTags('x'):
if x.getTag('item'):
affil = x.getTagAttr('item', 'affiliation').encode('utf-8')
role = x.getTagAttr('item', 'role').encode('utf-8')
break
# print '[%s] in room {%s}. (%s - %s)'% (nick_from, room_from, affil, role)
self.handler.emit('xmpp-presence-handler', presence=presence)
def send_join_room(self, room, nick):
self.handler.emit('join-room', room=room, nick=nick)
def handler_iq(self, connection, iq):
pass
def process(self, timeout=10):
if self.online:
self.client.Process(timeout)
else:
log.warning('disconnecting...')
if __name__ == '__main__':
resource = config.get('resource')
server = config.get('server')
connection = Connection(server, resource)
connection.start()
rooms = config.get('rooms').split(':')
from time import sleep
print connection.online
sleep(2)
print connection.online
for room in rooms:
connection.send_join_room(room.split('/')[0], room.split('/')[1])
i = 17
|
