flatisfy/modules/explorimmo/browser.py

# -*- coding: utf-8 -*-

# Copyright(C) 2014      Bezleputh
#
# This file is part of a woob module.
#
# This woob module is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This woob module 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this woob module. If not, see <http://www.gnu.org/licenses/>.

from woob.browser import PagesBrowser, URL
from woob.capabilities.housing import (TypeNotSupported, POSTS_TYPES,
                                         HOUSE_TYPES)
from woob.tools.compat import urlencode
from .pages import CitiesPage, SearchPage, HousingPage, HousingPage2, PhonePage


class ExplorimmoBrowser(PagesBrowser):
    BASEURL = 'https://immobilier.lefigaro.fr'

    cities = URL('/rest/locations\?q=(?P<city>.*)', CitiesPage)
    search = URL('/annonces/resultat/annonces.html\?(?P<query>.*)', SearchPage)
    housing_html = URL('/annonces/annonce-(?P<_id>.*).html', HousingPage)
    phone = URL('/rest/classifieds/(?P<_id>.*)/phone', PhonePage)
    housing = URL('/rest/classifieds/(?P<_id>.*)',
                  '/rest/classifieds/\?(?P<js_datas>.*)', HousingPage2)

    TYPES = {POSTS_TYPES.RENT: 'location',
             POSTS_TYPES.SALE: 'vente',
             POSTS_TYPES.FURNISHED_RENT: 'location',
             POSTS_TYPES.VIAGER: 'vente'}

    RET = {HOUSE_TYPES.HOUSE: 'Maison',
           HOUSE_TYPES.APART: 'Appartement',
           HOUSE_TYPES.LAND: 'Terrain',
           HOUSE_TYPES.PARKING: 'Parking',
           HOUSE_TYPES.OTHER: 'Divers'}

    def get_cities(self, pattern):
        return self.cities.open(city=pattern).get_cities()

    def search_housings(self, type, cities, nb_rooms, area_min, area_max,
                        cost_min, cost_max, house_types, advert_types):

        if type not in self.TYPES:
            raise TypeNotSupported()

        ret = []
        if type == POSTS_TYPES.VIAGER:
            ret = ['Viager']
        else:
            for house_type in house_types:
                if house_type in self.RET:
                    ret.append(self.RET.get(house_type))

        data = {'location': ','.join(cities).encode('iso 8859-1'),
                'furnished': type == POSTS_TYPES.FURNISHED_RENT,
                'areaMin': area_min or '',
                'areaMax': area_max or '',
                'priceMin': cost_min or '',
                'priceMax': cost_max or '',
                'transaction': self.TYPES.get(type, 'location'),
                'recherche': '',
                'mode': '',
                'proximity': '0',
                'roomMin': nb_rooms or '',
                'page': '1'}

        query = u'%s%s%s' % (urlencode(data), '&type=', '&type='.join(ret))

        return self.search.go(query=query).iter_housings(
            query_type=type,
            advert_types=advert_types
        )

    def get_housing(self, _id, housing=None):
        return self.housing.go(_id=_id).get_housing(obj=housing)

    def get_phone(self, _id):
        return self.phone.go(_id=_id).get_phone()

    def get_total_page(self, js_datas):
        return self.housing.open(js_datas=js_datas).get_total_page()