203 lines
8.7 KiB
Python
Executable file
203 lines
8.7 KiB
Python
Executable file
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
SAT plugin for managing xep-0045
|
|
Copyright (C) 2009, 2010 Jérôme Poisson (goffi@goffi.org)
|
|
|
|
This program 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, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
|
"""
|
|
|
|
from logging import debug, info, warning, error
|
|
from twisted.words.xish import domish
|
|
from twisted.internet import protocol, defer, threads, reactor
|
|
from twisted.words.protocols.jabber import client, jid, xmlstream
|
|
from twisted.words.protocols.jabber import error as jab_error
|
|
from twisted.words.protocols.jabber.xmlstream import IQ
|
|
from twisted.web.client import getPage
|
|
import os.path
|
|
import pdb
|
|
import random
|
|
|
|
from zope.interface import implements
|
|
|
|
from wokkel import disco, iwokkel, data_form
|
|
from tools.xml_tools import XMLUI
|
|
import urllib
|
|
|
|
from BeautifulSoup import BeautifulSoup
|
|
import re
|
|
|
|
|
|
PLUGIN_INFO = {
|
|
"name": "CouchSurfing plugin",
|
|
"import_name": "CS",
|
|
"type": "Misc",
|
|
"protocols": [],
|
|
"dependencies": [],
|
|
"main": "CS_Plugin",
|
|
"handler": "no",
|
|
"description": _(u"""This plugin allow to manage your CouchSurfing account throught your SàT frontend""")
|
|
}
|
|
|
|
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'
|
|
|
|
class CS_Plugin():
|
|
|
|
params = """
|
|
<params>
|
|
<individual>
|
|
<category name="CouchSurfing">
|
|
<param name="Login" type="string" />
|
|
<param name="Password" type="password" />
|
|
</category>
|
|
</individual>
|
|
</params>
|
|
"""
|
|
|
|
def __init__(self, host):
|
|
info(_("Plugin CS initialization"))
|
|
self.host = host
|
|
#parameters
|
|
host.memory.importParams(CS_Plugin.params)
|
|
#menu
|
|
host.importMenu(_("Plugin"), "CouchSurfing", self.menuSelected, help_string = _("Launch CoushSurfing mangement interface"))
|
|
self.data=self.host.memory.getPrivate('plugin_cs_data') or {} #TODO: delete cookies/data after a while
|
|
|
|
def erroCB(self, e, id):
|
|
"""Called when something is going wrong when contacting CS website"""
|
|
message_data={"reason": "connection error", "message":_(u"Impossible to contact CS website, please check your login/password, connection or try again later")}
|
|
self.host.bridge.actionResult("ERROR", id, message_data)
|
|
|
|
def menuSelected(self, id, profile):
|
|
"""Called when the couchsurfing menu item is selected"""
|
|
login = self.host.memory.getParamA("Login", "CouchSurfing", profile_key=profile)
|
|
password = self.host.memory.getParamA("Password", "CouchSurfing", profile_key=profile)
|
|
if not login or not password:
|
|
message_data={"reason": "uncomplete", "message":_(u"You have to fill your CouchSurfing login & password in parameters before using this interface")}
|
|
self.host.bridge.actionResult("ERROR", id, message_data)
|
|
return
|
|
|
|
post_data = urllib.urlencode({'auth_login[un]':login,'auth_login[pw]':password,'auth_login[action]':'Login...'})
|
|
|
|
if not self.data.has_key(profile):
|
|
self.data[profile] = {'cookies':{}}
|
|
else:
|
|
self.data[profile]['cookies'] = {}
|
|
|
|
|
|
|
|
|
|
|
|
#tmp
|
|
f = open('/home/goffi/tmp/CS_principale.html','r')
|
|
html = f.read()
|
|
self.__connectionCB(html, id, profile)
|
|
|
|
"""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'])
|
|
d.addCallback(self.__connectionCB, id, profile)
|
|
d.addErrback(self.erroCB, id)"""
|
|
|
|
|
|
#self.host.bridge.actionResult("SUPPRESS", id, {})
|
|
|
|
|
|
#pages parsing callbacks
|
|
def savePage(self, name, html):
|
|
f = open ('/home/goffi/tmp/CS_'+name+'.html','w')
|
|
f.write(html)
|
|
f.close()
|
|
print "page [%s] sauvee" % name
|
|
#pdb.set_trace()
|
|
|
|
def __connectionCB(self, html, id, profile):
|
|
print 'Response received'
|
|
self.savePage('principale',html)
|
|
soup = BeautifulSoup(html)
|
|
self.data[profile]['user_nick'] = soup.find('a','item_link',href='/home.html').contents[0]
|
|
self.data[profile]['user_name'] = soup.html.head.title.string.split(' - ')[1]
|
|
#unread messages
|
|
try:
|
|
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))
|
|
except:
|
|
self.data[profile]['unread_messages'] = 0
|
|
#unread couchrequest messages
|
|
try:
|
|
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))
|
|
except:
|
|
self.data[profile]['unread_CR_messages'] = 0
|
|
|
|
#if we have already the list of friend, no need to make new requests
|
|
if not self.data[profile].has_key('friends'):
|
|
self.data[profile]['friends'] = {}
|
|
"""f = open('/home/goffi/tmp/CS_friends.html','r')
|
|
html = f.read()
|
|
self.__friendsPageCB(html, id, profile)"""
|
|
d = getPage('http://www.couchsurfing.org/connections.html?type=myfriends&show=10000', agent=AGENT, cookies=self.data[profile]['cookies'])
|
|
d.addCallback(self.__friendsPageCB, id=id, profile=profile)
|
|
d.addErrback(self.erroCB, id)
|
|
else:
|
|
self.host.bridge.actionResult("XMLUI", id, {"type":"window", "xml":self.__buildUI(self.data[profile])})
|
|
|
|
def __buildUI(self, data):
|
|
"""Build the XML UI of the plugin
|
|
@param data: data store for the profile"""
|
|
user_nick = data['user_nick']
|
|
user_name = data['user_name']
|
|
unread_mess = data['unread_messages']
|
|
unread_CR_mess = data['unread_CR_messages']
|
|
friends_list = data['friends'].keys()
|
|
friends_list.sort()
|
|
interface = XMLUI('window','tabs')
|
|
interface.addCategory(_("Messages"), "vertical")
|
|
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 ''})
|
|
interface.addList(friends_list, 'friends')
|
|
interface.addCategory(_("Events"), "vertical")
|
|
interface.addCategory(_("Couch search"), "vertical")
|
|
return interface.toXml()
|
|
|
|
def __meetingPageCB(self, html):
|
|
"""Called when the meeting page has been received"""
|
|
|
|
def __friendsPageCB(self, html, id, profile):
|
|
"""Called when the friends list page has been received"""
|
|
self.savePage('friends',html)
|
|
soup = BeautifulSoup(html.replace('"formtable width="400','"formtable" width="400"'))
|
|
friends = self.data[profile]['friends']
|
|
for _tr in soup.findAll('tr', {'class':re.compile("^msgRow*")}): #we parse the row with friends infos
|
|
_nobr = _tr.find('nobr') #contain the friend name
|
|
friend_name = unicode(_nobr.string)
|
|
friend_link = u'http://www.couchsurfing.org'+_nobr.parent['href']
|
|
regex_href = re.compile(r'/connections\.html\?id=([^&]+)')
|
|
a_tag = _tr.find('a',href=regex_href)
|
|
friend_id = regex_href.search(unicode(a_tag)).groups()[0]
|
|
|
|
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})
|
|
friends[friend_name] = {'link':friend_link,'id':friend_id}
|
|
a = soup.find('td','barmiddle next').a #is there several pages ?
|
|
if a:
|
|
#yes, we parse the next page
|
|
d = getPage('http://www.couchsurfing.org/'+str(a['href']), agent=AGENT, cookies=self.data[profile]['cookies'])
|
|
d.addCallback(self.__friendsPageCB, id=id, profile=profile)
|
|
d.addErrback(self.erroCB, id)
|
|
else:
|
|
#no, we show the result
|
|
self.host.bridge.actionResult("XMLUI", id, {"type":"window", "xml":self.__buildUI(self.data[profile])})
|
|
#and save the data
|
|
self.host.memory.setPrivate('plugin_cs_data', self.data)
|
|
|
|
|
|
|
|
|
|
|