|
101
|
1 #!/usr/bin/python |
|
|
2 # -*- coding: utf-8 -*- |
|
|
3 |
|
|
4 """ |
|
|
5 SAT plugin for managing xep-0045 |
|
|
6 Copyright (C) 2009, 2010 Jérôme Poisson (goffi@goffi.org) |
|
|
7 |
|
|
8 This program is free software: you can redistribute it and/or modify |
|
|
9 it under the terms of the GNU General Public License as published by |
|
|
10 the Free Software Foundation, either version 3 of the License, or |
|
|
11 (at your option) any later version. |
|
|
12 |
|
|
13 This program is distributed in the hope that it will be useful, |
|
|
14 but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
|
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
|
16 GNU General Public License for more details. |
|
|
17 |
|
|
18 You should have received a copy of the GNU General Public License |
|
|
19 along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
|
20 """ |
|
|
21 |
|
|
22 from logging import debug, info, warning, error |
|
|
23 from twisted.words.xish import domish |
|
|
24 from twisted.internet import protocol, defer, threads, reactor |
|
|
25 from twisted.words.protocols.jabber import client, jid, xmlstream |
|
|
26 from twisted.words.protocols.jabber import error as jab_error |
|
|
27 from twisted.words.protocols.jabber.xmlstream import IQ |
|
|
28 from twisted.web.client import getPage |
|
|
29 import os.path |
|
|
30 import pdb |
|
|
31 import random |
|
|
32 |
|
|
33 from zope.interface import implements |
|
|
34 |
|
|
35 from wokkel import disco, iwokkel, data_form |
|
106
|
36 from tools.xml_tools import XMLUI |
|
101
|
37 import urllib |
|
|
38 |
|
102
|
39 from BeautifulSoup import BeautifulSoup |
|
106
|
40 import re |
|
101
|
41 |
|
|
42 |
|
|
43 PLUGIN_INFO = { |
|
|
44 "name": "CouchSurfing plugin", |
|
|
45 "import_name": "CS", |
|
|
46 "type": "Misc", |
|
|
47 "protocols": [], |
|
|
48 "dependencies": [], |
|
|
49 "main": "CS_Plugin", |
|
|
50 "handler": "no", |
|
|
51 "description": _(u"""This plugin allow to manage your CouchSurfing account throught your SàT frontend""") |
|
|
52 } |
|
|
53 |
|
|
54 AGENT = 'Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.2.3) Gecko/20100423 Ubuntu/10.04 (lucid) Firefox/3.6.3' |
|
|
55 |
|
|
56 class CS_Plugin(): |
|
|
57 |
|
|
58 params = """ |
|
|
59 <params> |
|
|
60 <individual> |
|
|
61 <category name="CouchSurfing"> |
|
|
62 <param name="Login" type="string" /> |
|
|
63 <param name="Password" type="password" /> |
|
|
64 </category> |
|
|
65 </individual> |
|
|
66 </params> |
|
|
67 """ |
|
|
68 |
|
|
69 def __init__(self, host): |
|
|
70 info(_("Plugin CS initialization")) |
|
|
71 self.host = host |
|
|
72 #parameters |
|
|
73 host.memory.importParams(CS_Plugin.params) |
|
|
74 #menu |
|
|
75 host.importMenu(_("Plugin"), "CouchSurfing", self.menuSelected, help_string = _("Launch CoushSurfing mangement interface")) |
|
106
|
76 self.data=self.host.memory.getPrivate('plugin_cs_data') or {} #TODO: delete cookies/data after a while |
|
102
|
77 |
|
|
78 def erroCB(self, e, id): |
|
|
79 """Called when something is going wrong when contacting CS website""" |
|
|
80 message_data={"reason": "connection error", "message":_(u"Impossible to contact CS website, please check your login/password, connection or try again later")} |
|
|
81 self.host.bridge.actionResult("ERROR", id, message_data) |
|
|
82 |
|
101
|
83 def menuSelected(self, id, profile): |
|
|
84 """Called when the couchsurfing menu item is selected""" |
|
|
85 login = self.host.memory.getParamA("Login", "CouchSurfing", profile_key=profile) |
|
|
86 password = self.host.memory.getParamA("Password", "CouchSurfing", profile_key=profile) |
|
|
87 if not login or not password: |
|
|
88 message_data={"reason": "uncomplete", "message":_(u"You have to fill your CouchSurfing login & password in parameters before using this interface")} |
|
|
89 self.host.bridge.actionResult("ERROR", id, message_data) |
|
|
90 return |
|
|
91 |
|
|
92 post_data = urllib.urlencode({'auth_login[un]':login,'auth_login[pw]':password,'auth_login[action]':'Login...'}) |
|
|
93 |
|
106
|
94 if not self.data.has_key(profile): |
|
|
95 self.data[profile] = {'cookies':{}} |
|
|
96 else: |
|
|
97 self.data[profile]['cookies'] = {} |
|
101
|
98 |
|
|
99 |
|
|
100 |
|
|
101 |
|
106
|
102 |
|
102
|
103 #tmp |
|
|
104 f = open('/home/goffi/tmp/CS_principale.html','r') |
|
|
105 html = f.read() |
|
106
|
106 self.__connectionCB(html, id, profile) |
|
102
|
107 |
|
|
108 """d = getPage('http://www.couchsurfing.org/login.html', method='POST', postdata=post_data, headers={'Content-Type':'application/x-www-form-urlencoded'} , agent=AGENT, cookies=self.data[profile]['cookies']) |
|
106
|
109 d.addCallback(self.__connectionCB, id, profile) |
|
102
|
110 d.addErrback(self.erroCB, id)""" |
|
|
111 |
|
|
112 |
|
106
|
113 #self.host.bridge.actionResult("SUPPRESS", id, {}) |
|
|
114 |
|
|
115 |
|
|
116 #pages parsing callbacks |
|
|
117 def savePage(self, name, html): |
|
|
118 f = open ('/home/goffi/tmp/CS_'+name+'.html','w') |
|
|
119 f.write(html) |
|
|
120 f.close() |
|
|
121 print "page [%s] sauvee" % name |
|
|
122 #pdb.set_trace() |
|
|
123 |
|
|
124 def __connectionCB(self, html, id, profile): |
|
|
125 print 'Response received' |
|
|
126 self.savePage('principale',html) |
|
|
127 soup = BeautifulSoup(html) |
|
|
128 self.data[profile]['user_nick'] = soup.find('a','item_link',href='/home.html').contents[0] |
|
|
129 self.data[profile]['user_name'] = soup.html.head.title.string.split(' - ')[1] |
|
|
130 #unread messages |
|
|
131 try: |
|
|
132 self.data[profile]['unread_messages'] = int(soup.find(lambda tag: tag.name=='div' and ('class','item_bubble') in tag.attrs and tag.find('a', href="/messages.html?message_status=inbox")).find(text=True)) |
|
|
133 except: |
|
|
134 self.data[profile]['unread_messages'] = 0 |
|
|
135 #unread couchrequest messages |
|
|
136 try: |
|
|
137 self.data[profile]['unread_CR_messages'] = int(soup.find(lambda tag: tag.name=='div' and ('class','item_bubble') in tag.attrs and tag.find('a', href="/couchmanager")).find(text=True)) |
|
|
138 except: |
|
|
139 self.data[profile]['unread_CR_messages'] = 0 |
|
|
140 |
|
|
141 #if we have already the list of friend, no need to make new requests |
|
|
142 if not self.data[profile].has_key('friends'): |
|
|
143 self.data[profile]['friends'] = {} |
|
|
144 """f = open('/home/goffi/tmp/CS_friends.html','r') |
|
|
145 html = f.read() |
|
|
146 self.__friendsPageCB(html, id, profile)""" |
|
|
147 d = getPage('http://www.couchsurfing.org/connections.html?type=myfriends&show=10000', agent=AGENT, cookies=self.data[profile]['cookies']) |
|
|
148 d.addCallback(self.__friendsPageCB, id=id, profile=profile) |
|
|
149 d.addErrback(self.erroCB, id) |
|
|
150 else: |
|
|
151 self.host.bridge.actionResult("XMLUI", id, {"type":"window", "xml":self.__buildUI(self.data[profile])}) |
|
|
152 |
|
|
153 def __buildUI(self, data): |
|
|
154 """Build the XML UI of the plugin |
|
|
155 @param data: data store for the profile""" |
|
|
156 user_nick = data['user_nick'] |
|
|
157 user_name = data['user_name'] |
|
|
158 unread_mess = data['unread_messages'] |
|
|
159 unread_CR_mess = data['unread_CR_messages'] |
|
|
160 friends_list = data['friends'].keys() |
|
|
161 friends_list.sort() |
|
|
162 interface = XMLUI('window','tabs') |
|
|
163 interface.addCategory(_("Messages"), "vertical") |
|
|
164 interface.addText(_("G'day %(name)s, you have %(nb_message)i unread message%(plural_mess)s and %(unread_CR_mess)s unread couch request message%(plural_CR)s") % {'name':user_name, 'nb_message':unread_mess, 'plural_mess':'s' if unread_mess>1 else '', 'unread_CR_mess': unread_CR_mess, 'plural_CR':'s' if unread_CR_mess>1 else ''}) |
|
|
165 interface.addList(friends_list, 'friends') |
|
|
166 interface.addCategory(_("Events"), "vertical") |
|
|
167 interface.addCategory(_("Couch search"), "vertical") |
|
|
168 return interface.toXml() |
|
|
169 |
|
|
170 def __meetingPageCB(self, html): |
|
|
171 """Called when the meeting page has been received""" |
|
|
172 |
|
|
173 def __friendsPageCB(self, html, id, profile): |
|
|
174 """Called when the friends list page has been received""" |
|
|
175 self.savePage('friends',html) |
|
|
176 soup = BeautifulSoup(html.replace('"formtable width="400','"formtable" width="400"')) |
|
|
177 friends = self.data[profile]['friends'] |
|
|
178 for _tr in soup.findAll('tr', {'class':re.compile("^msgRow*")}): #we parse the row with friends infos |
|
|
179 _nobr = _tr.find('nobr') #contain the friend name |
|
|
180 friend_name = unicode(_nobr.string) |
|
|
181 friend_link = u'http://www.couchsurfing.org'+_nobr.parent['href'] |
|
|
182 regex_href = re.compile(r'/connections\.html\?id=([^&]+)') |
|
|
183 a_tag = _tr.find('a',href=regex_href) |
|
|
184 friend_id = regex_href.search(unicode(a_tag)).groups()[0] |
|
|
185 |
|
|
186 debug(_("CS friend found: %(friend_name)s (id: %(friend_id)s, link: %(friend_link)s)") % {'friend_name':friend_name, 'friend_id':friend_id, 'friend_link':friend_link}) |
|
|
187 friends[friend_name] = {'link':friend_link,'id':friend_id} |
|
|
188 a = soup.find('td','barmiddle next').a #is there several pages ? |
|
|
189 if a: |
|
|
190 #yes, we parse the next page |
|
|
191 d = getPage('http://www.couchsurfing.org/'+str(a['href']), agent=AGENT, cookies=self.data[profile]['cookies']) |
|
|
192 d.addCallback(self.__friendsPageCB, id=id, profile=profile) |
|
|
193 d.addErrback(self.erroCB, id) |
|
|
194 else: |
|
|
195 #no, we show the result |
|
|
196 self.host.bridge.actionResult("XMLUI", id, {"type":"window", "xml":self.__buildUI(self.data[profile])}) |
|
|
197 #and save the data |
|
|
198 self.host.memory.setPrivate('plugin_cs_data', self.data) |
|
|
199 |
|
102
|
200 |
|
|
201 |
|
|
202 |
|
106
|
203 |
