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

backported some aum stuff

Browse source
This commit is contained in:
Romain Bignon 2010-03-11 17:40:26 +01:00
commit bcd63ee3e7
2 changed files with 17 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
weboob/backends/aum/adopte.py
View file

@ -65,6 +65,10 @@ class AdopteUnMec(Browser):
'http://www.adopteunmec.com/(\w+)': ProfilePage, # a custom profile url
}
def __init__(self, *args, **kwargs):
Browser.__init__(self, *args, **kwargs)
self.my_id = 0
def login(self):
if not self.is_on_page(LoginPage):
self.home()
@ -132,9 +136,13 @@ class AdopteUnMec(Browser):
@pageaccess
def get_my_id(self):
if self.my_id:
return self.my_id
if not self.is_on_page(HomePage):
self.home()
return self.page.get_my_id()
self.my_id = self.page.get_my_id()
return self.my_id
@pageaccess
def nb_new_mails(self):

11
weboob/tools/browser.py
View file

@ -219,6 +219,11 @@ class Browser(mechanize.Browser):
if self.__cookie:
self.__cookie.save()
def str(self, s):
if isinstance(s, unicode):
s = s.encode('iso-8859-15', 'replace')
return s
def set_field(self, args, label, field=None, value=None, is_list=False):
try:
if not field:
@ -228,15 +233,15 @@ class Browser(mechanize.Browser):
if is_list:
if isinstance(is_list, (list, tuple)):
try:
value = [str(is_list.index(args[label]))]
value = [self.str(is_list.index(args[label]))]
except ValueError, e:
if args[label]:
print '[%s] %s: %s' % (label, args[label], e)
return
else:
value = [str(args[label])]
value = [self.str(args[label])]
else:
value = str(args[label])
value = self.str(args[label])
self[field] = value
except ClientForm.ControlNotFoundError:
return