Shevraar/jovabot

import logging
import sys

import requests
from bs4 import BeautifulSoup

# White pages url
WHITE_PAGES_URL = "http://www.paginebianche.it/execute.cgi"


class wp_response(object):
    def __init__(self, name, tel, addr):
        self.name = name
        self.tel = tel
        self.addr = addr


def search_wp(name, location):
    payload = {'btt': '1', 'qs': name, 'dv': location}

    try:
        r = requests.get(WHITE_PAGES_URL, params=payload)
        logging.info('requesting url {0}'.format(r.url))
        return parse_response(r.text)
    except Exception as e:
        logging.exception('search failed {0}'.format(e))
        return None


def parse_response(text):
    soup = BeautifulSoup(text, 'html.parser')

    for d in soup.find_all("div", "vcard"):
        try:
            name = d.find('h2', 'rgs').a['title']
            phone = d.find("div", "tel").find("span", "value").text
            address = d.find("div", "address").div.text
        except:
            logging.exception(d.find('h2', 'rgs'))
            continue

        yield wp_response(name, phone, address)


def test():
    name = sys.argv[1]
    loc = sys.argv[2]

    for o in search_wp(name, loc):
        logging.debug(o.__dict__)


if __name__ == "__main__":
    test()