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

if we're on waiting, go on /invits.php to get my own id

Browse source
This commit is contained in:
Romain Bignon 2010-11-11 16:00:43 +01:00
commit 6330ab2515
4 changed files with 27 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

7
weboob/backends/aum/pages/home.py
View file

@ -19,10 +19,8 @@
import re
from weboob.backends.aum.pages.base import PageBase
from logging import error, warning
class HomePage(PageBase):
MYID_REGEXP = re.compile("http://www.adopteunmec.com/\?mid=(\d+)")
def get_my_id(self):
@ -32,11 +30,10 @@ class HomePage(PageBase):
if m:
return m.group(1)
error("Error: Unable to find my ID")
self.browser.logger.error("Error: Unable to find my ID")
return 0
def __get_home_indicator(self, pos, what):
tables = self.document.getElementsByTagName('table')
for table in tables:
if table.hasAttribute('style') and table.getAttribute('style') == 'background-color:black;background-image:url(http://s.adopteunmec.com/img/barmec.gif);background-repeat:no-repeat':
@ -47,7 +44,7 @@ class HomePage(PageBase):
i += 1
if i == pos:
return int(font.firstChild.data)
warning(u'Could not parse number of %s' % what)
self.browser.logger.error(u'Could not parse number of %s' % what)
return 0
def nb_available_charms(self):

13
weboob/backends/aum/pages/login.py
View file

@ -123,3 +123,16 @@ class ShopPage(PageBase):
class ErrPage(PageBase):
def on_loaded(self):
raise BrowserIncorrectPassword('Incorrect login/password')
class InvitePage(PageBase):
MYID_REGEXP = re.compile("http://www.adopteunmec.com/\?mid=(\d+)")
def get_my_id(self):
fonts = self.document.getElementsByTagName('font')
for font in fonts:
m = self.MYID_REGEXP.match(font.firstChild.data)
if m:
return m.group(1)
self.browser.logger.error("Error: Unable to find my ID")
return 0