249 lines
9.5 KiB
Python
249 lines
9.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Copyright(C) 2008-2010 Romain Bignon
|
|
|
|
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, version 3 of the License.
|
|
|
|
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, write to the Free Software
|
|
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
|
|
"""
|
|
|
|
import time
|
|
from logging import warning
|
|
|
|
from weboob.tools.browser import Browser
|
|
from weboob.backends.aum.exceptions import AdopteWait
|
|
|
|
from weboob.backends.aum.pages.home import HomePage
|
|
from weboob.backends.aum.pages.contact_list import ContactListPage
|
|
from weboob.backends.aum.pages.contact_thread import ContactThreadPage
|
|
from weboob.backends.aum.pages.baskets import BasketsPage
|
|
from weboob.backends.aum.pages.profile import ProfilePage
|
|
from weboob.backends.aum.pages.search import SearchPage
|
|
from weboob.backends.aum.pages.login import LoginPage, RedirectPage, BanPage, ErrPage, RegisterPage, RegisterWaitPage, RegisterConfirmPage
|
|
from weboob.backends.aum.pages.edit import EditPhotoPage, EditPhotoCbPage, EditAnnouncePage, EditDescriptionPage, EditSexPage, EditPersonalityPage
|
|
from weboob.backends.aum.pages.wait import WaitPage
|
|
|
|
class AdopteUnMec(Browser):
|
|
DOMAIN = 'www.adopteunmec.com'
|
|
PROTOCOL = 'http'
|
|
PAGES = {'http://www.adopteunmec.com/': LoginPage,
|
|
'http://www.adopteunmec.com/index.html': LoginPage,
|
|
'http://www.adopteunmec.com/index.php': LoginPage,
|
|
'http://www.adopteunmec.com/loginErr.php.*': ErrPage,
|
|
'http://www.adopteunmec.com/bans.php\?who=auto': BanPage,
|
|
'http://www.adopteunmec.com/redirect.php\?action=login': RedirectPage,
|
|
'http://www.adopteunmec.com/wait.php': WaitPage,
|
|
'http://www.adopteunmec.com/register2.php': RegisterPage,
|
|
'http://www.adopteunmec.com/register3.php.*': RegisterWaitPage,
|
|
'http://www.adopteunmec.com/register4.php.*': RegisterConfirmPage,
|
|
'http://www.adopteunmec.com/home.php': HomePage,
|
|
'http://www.adopteunmec.com/mails.php': ContactListPage,
|
|
'http://www.adopteunmec.com/mails.php\?type=1': BasketsPage,
|
|
'http://www.adopteunmec.com/thread.php\?id=([0-9]+)': ContactThreadPage,
|
|
'http://www.adopteunmec.com/edit.php\?type=1': EditPhotoPage,
|
|
'http://s\d+.adopteunmec.com/upload\d.php\?.*': EditPhotoCbPage,
|
|
'http://www.adopteunmec.com/edit.php\?type=2': EditAnnouncePage,
|
|
'http://www.adopteunmec.com/edit.php\?type=3': EditDescriptionPage,
|
|
'http://www.adopteunmec.com/edit.php\?type=4': EditSexPage,
|
|
'http://www.adopteunmec.com/edit.php\?type=5': EditPersonalityPage,
|
|
'http://www.adopteunmec.com/search.php.*': SearchPage,
|
|
'http://www.adopteunmec.com/searchRes.php.*': SearchPage,
|
|
'http://www.adopteunmec.com/rencontres-femmes/(.*)/([0-9]+)': ProfilePage,
|
|
'http://www.adopteunmec.com/catalogue-hommes/(.*)/([0-9]+)': ProfilePage,
|
|
'http://www.adopteunmec.com/view2.php': ProfilePage, # my own profile
|
|
'http://www.adopteunmec.com/(\w+)': ProfilePage, # a custom profile url
|
|
}
|
|
|
|
def login(self):
|
|
if not self.is_on_page(LoginPage):
|
|
self.home()
|
|
self.page.login(self.username, self.password)
|
|
|
|
def is_logged(self):
|
|
return not self.is_on_page(LoginPage)
|
|
|
|
def home(self):
|
|
return self.location('http://www.adopteunmec.com/home.php')
|
|
|
|
def pageaccess(func):
|
|
def inner(self, *args, **kwargs):
|
|
if self.is_on_page(WaitPage):
|
|
if not self.page.check():
|
|
raise AdopteWait()
|
|
self.home()
|
|
if not self.page or self.is_on_page(LoginPage) and self.password:
|
|
self.home()
|
|
|
|
return func(self, *args, **kwargs)
|
|
return inner
|
|
|
|
def register(self, nickname, password, sex, birthday_d, birthday_m, birthday_y, zipcode, country, godfather=''):
|
|
if not self.is_on_page(RegisterPage):
|
|
self.location('http://www.adopteunmec.com/register2.php')
|
|
|
|
return self.page.register(nickname, password, sex, birthday_d, birthday_m, birthday_y, zipcode, country, godfather)
|
|
|
|
@pageaccess
|
|
def add_photo(self, name, f):
|
|
if not self.is_on_page(EditPhotoPage):
|
|
self.location('/edit.php?type=1')
|
|
return self.page.add_photo(name, f)
|
|
|
|
@pageaccess
|
|
def set_nickname(self, nickname):
|
|
if not self.is_on_page(EditAnnouncePage):
|
|
self.location('/edit.php?type=2')
|
|
return self.page.set_nickname(nickname)
|
|
|
|
@pageaccess
|
|
def set_announce(self, title=None, description=None, lookingfor=None):
|
|
if not self.is_on_page(EditAnnouncePage):
|
|
self.location('/edit.php?type=2')
|
|
return self.page.set_announce(title, description, lookingfor)
|
|
|
|
@pageaccess
|
|
def set_description(self, **args):
|
|
if not self.is_on_page(EditDescriptionPage):
|
|
self.location('/edit.php?type=3')
|
|
return self.page.set_description(**args)
|
|
|
|
@pageaccess
|
|
def score(self):
|
|
if time.time() - self.__last_update > 60:
|
|
self.home()
|
|
return self.page.score()
|
|
|
|
@pageaccess
|
|
def get_my_name(self):
|
|
if time.time() - self.__last_update > 60:
|
|
self.home()
|
|
return self.page.get_my_name()
|
|
|
|
@pageaccess
|
|
def get_my_id(self):
|
|
if not self.is_on_page(HomePage):
|
|
self.home()
|
|
return self.page.get_my_id()
|
|
|
|
@pageaccess
|
|
def nb_new_mails(self):
|
|
if time.time() - self.__last_update > 60:
|
|
self.home()
|
|
return self.page.nb_new_mails()
|
|
|
|
@pageaccess
|
|
def nb_new_baskets(self):
|
|
if time.time() - self.__last_update > 60:
|
|
self.home()
|
|
return self.page.nb_new_baskets()
|
|
|
|
@pageaccess
|
|
def nb_new_visites(self):
|
|
if time.time() - self.__last_update > 60:
|
|
self.home()
|
|
return self.page.nb_new_visites()
|
|
|
|
@pageaccess
|
|
def nb_available_charms(self):
|
|
self.home()
|
|
return self.page.nb_available_charms()
|
|
|
|
@pageaccess
|
|
def get_baskets(self):
|
|
self.location('/mails.php?type=1')
|
|
return self.page.get_profiles_ids_list()
|
|
|
|
@pageaccess
|
|
def flush_visits(self):
|
|
""" Does nothing, only flush new visits to increase my score """
|
|
self.openurl('/mails.php?type=3')
|
|
|
|
@pageaccess
|
|
def get_contact_list(self):
|
|
if not self.is_on_page(ContactListPage):
|
|
self.location('/mails.php')
|
|
|
|
return self.page.get_contact_list()
|
|
|
|
@pageaccess
|
|
def get_thread_mails(self, id):
|
|
self.page.open_thread_page(id)
|
|
return self.page.get_mails()
|
|
|
|
@pageaccess
|
|
def post_mail(self, id, content):
|
|
self.page.open_thread_page(id)
|
|
self.page.post(content)
|
|
|
|
@pageaccess
|
|
def send_charm(self, id):
|
|
result = self.openurl('http://www.adopteunmec.com/fajax_addBasket.php?id=%s' % id).read()
|
|
warning('Charm: %s' % result)
|
|
return result.find('noMoreFlashes') < 0
|
|
|
|
@pageaccess
|
|
def add_basket(self, id):
|
|
result = self.openurl('http://www.adopteunmec.com/fajax_addBasket.php?id=%s' % id).read()
|
|
warning('Basket: %s' % result)
|
|
# TODO check if it works (but it should)
|
|
return True
|
|
|
|
def deblock(self, id):
|
|
result = self.openurl('http://www.adopteunmec.com/fajax_postMessage.php?action=deblock&to=%s' % id).read()
|
|
warning('Deblock: %s' % result)
|
|
return True
|
|
|
|
@pageaccess
|
|
def rate(self, id, what, rating):
|
|
print 'rate "%s"' % id, what, rating
|
|
result = self.openurl('http://www.adopteunmec.com/fajax_vote.php', 'member=%s&what=%s&rating=%s' % (id, what, rating)).read()
|
|
print result
|
|
return float(result)
|
|
|
|
@pageaccess
|
|
def search_profiles(self, **kwargs):
|
|
self.location('/search.php?display=1')
|
|
self.page.search(**kwargs)
|
|
return self.page.get_profiles_ids()
|
|
|
|
@pageaccess
|
|
def get_profile(self, link):
|
|
if isinstance(link, (str,unicode)) and link.startswith('/'):
|
|
link = link[1:]
|
|
self.location('/%s' % link)
|
|
return self.page
|
|
|
|
@pageaccess
|
|
def get_slut_state(self, id):
|
|
result = self.openurl('http://www.adopteunmec.com/%s' % id).read()
|
|
if result.find('<td align="right" style="font-size:12px;font-weight:bold">en ligne</td>') >= 0:
|
|
r = 'online'
|
|
elif result.find('Cet utilisateur a quitt\xe9 le site<br />') >= 0:
|
|
r = 'removed'
|
|
elif result.find('ce profil a \xe9t\xe9 bloqu\xe9 par l\'\xe9quipe de mod\xe9ration<br />') >= 0:
|
|
r = 'removed'
|
|
elif result.find('<div align=center style="color:#ff0000;font-size:16px"><br /><br />Cette personne<br>vous a bloqu\xe9</div>') >= 0:
|
|
r = 'blocked'
|
|
else:
|
|
r = 'offline'
|
|
|
|
print 'getSlutState(%s) = %s' % (id, r)
|
|
return r
|
|
|
|
@pageaccess
|
|
def is_slut_online(self, id):
|
|
result = self.openurl('http://www.adopteunmec.com/%s' % id).read()
|
|
r = result.find('<td align="right" style="font-size:12px;font-weight:bold">en ligne</td>') >= 0
|
|
print 'isSlutOnline(%s) = %s' % (id, r)
|
|
return r
|