nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

ovs: restrict conf choices and let browser fetch cities list

Browse source
This commit is contained in:
Vincent A 2013-11-07 18:24:36 +01:00 • committed by Florent
commit 4ee3460b01
3 changed files with 33 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
modules/ovs/pages.py
View file

@ -21,12 +21,13 @@
import datetime
import re
import urllib
from urlparse import urlsplit
from weboob.tools.browser import BasePage
from weboob.capabilities.messages import Message, Thread
import ovsparse
__all__ = ['PagePrivateThreadsList', 'PagePrivateThread', 'PageLogin', 'PageIndex', 'DummyPage', 'PagePostMessage', 'PageUserProfile']
__all__ = ['PagePrivateThreadsList', 'PagePrivateThread', 'PageLogin', 'PageIndex', 'DummyPage', 'PagePostMessage', 'PageUserProfile', 'PageCityList']
class OvsPage(BasePage):
@ -172,6 +173,19 @@ class PageUserProfile(OvsPage):
#~ self.browser.submit()
class PageCityList(DummyPage):
def get_cities(self, master_domain='onvasortir.com'):
cities = {}
for home_a in self.document.findAll('a', href=re.compile(r'http://(.*)\.%s/?' % master_domain)):
hostname = urlsplit(home_a['href']).hostname
code = hostname.split('.')[0]
if code == 'www':
continue
name = home_a.text
cities[name] = {'code': code, 'hostname': hostname}
return cities
def recode_dict(dict_, encoding):
for k in dict_:
dict_[k] = dict_[k].encode(encoding)