GooglePoiCrawler/run.py

# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.by import By

from bs4 import BeautifulSoup

from utility import database_access as DA
from utility.parseutils import *
from utility.connect import *

import pandas as pd
import time
import json
import re
# import pyautogui as pag

def serive_create(profilepath):
    option = webdriver.ChromeOptions()

    option.add_argument('--disable-web-security')
    option.add_argument('--allow-running-insecure-content') 
    option.add_argument("--user-data-dir=C:\\Users\\noodles\\AppData\\Local\\Google\\Chrome\\User Data")
    option.add_argument("profile-directory="+profilepath)

    driver = webdriver.Chrome('./utility/chromedriver_20211103/chromedriver', options=option)
    executor_url = driver.command_executor._url
    session_id = driver.session_id
    print (session_id)
    print (executor_url)
    time.sleep(3)
    
    return driver

def brower_start():
    options = webdriver.ChromeOptions()
    browser = webdriver.Remote(
        command_executor='http://192.53.174.202:4444/wd/hub',
        desired_capabilities=options.to_capabilities()
    )
    return browser


def keyin_keyword(driver, keyword):
    button = driver.find_element_by_id("searchbox")
    driver.implicitly_wait(30)
    ActionChains(driver).move_to_element(button).send_keys(keyword).send_keys(Keys.RETURN).perform()
    time.sleep(3)

    element = driver.find_element_by_class_name("V0h1Ob-haAclf")
    driver.implicitly_wait(30)
    ActionChains(driver).move_to_element(element).click(element).perform()


def open_time(driver):
    element = driver.find_element_by_xpath('//*[@id="pane"]/div/div[1]/div/div/div[9]/div[2]')
    if element.text.find('預訂') == -1:
        element = driver.find_element_by_xpath('//*[@id="pane"]/div/div[1]/div/div/div[9]/div[2]')
        driver.implicitly_wait(20)
        ActionChains(driver).move_to_element(element).click(element).perform()
        return 1
    else:
        return 0


def get_shop_info(driver, output, shop_soup):
    current_url_split = driver.current_url.split('@')[1].split(',')
    output['lon'] = current_url_split[1]
    output['lat'] = current_url_split[0]
    
    location = shop_soup.find('button',{'data-item-id':'oloc'})['aria-label'].split(' ')
    output['city'] = location[-1]
    output['area'] = location[-2]
    print(location)

    output['addr'] = shop_soup.find('button',{'data-item-id':'address'})['aria-label'].replace('地址:', '')
    output['tel'] = blank_check(shop_soup.find('button',{'data-tooltip':'複製電話號碼'})['aria-label'].split(':')[1])
    print(output['addr'], output['tel'])

    for key in element_list:
        element = element_list[key]
        if len(element) == 3:
            value = shop_soup.find(element[0],element[1])[element[2]]
            
        else:
            tmp_value = shop_soup.find(element[0],element[1])
            if tmp_value:
                value = tmp_value.text
            else:
                value = ''

        output[key] = value_check(key, value)

    return output


def get_intro_info(driver, output):
    element = driver.find_element_by_xpath('//*[@id="pane"]/div/div[1]/div/div/div[6]')
    driver.implicitly_wait(20)
    ActionChains(driver).move_to_element(element).click(element).perform()

    wait = WebDriverWait(driver, 30)
    item_xpath = "div[aria-label='{}簡介']".format(output['name'])
    wait.until(
        EC.element_to_be_clickable((By.CSS_SELECTOR, item_xpath))
    )
    time.sleep(1)
    intro_soup = BeautifulSoup(driver.page_source, 'html.parser')

    for key in intro_list:
        elements = intro_soup.find('div',{'aria-label':key})
        # print(elements)
        if elements:
            element = elements.find_all('li',{'class':'LQjNnc-p83tee-JNdkSc-ibnC6b'})
            # print(element)
            count = 0
            tmp = []
            for ele in element:
                if ele.find('img',{'src':"//www.gstatic.com/images/icons/material/system_gm/2x/check_black_18dp.png"}):
                    tmp += [{
                        'id':count,
                        intro_list[key][1]: blank_check(ele.text)
                    }]
                    count += 1
            output[intro_list[key][0]] = str(tmp)
        else:
            output[intro_list[key][0]] = []
    driver.back()
    return output


def get_time_list(shop_soup, output):
    open_now = blank_check(shop_soup.find('span', {'class':'LJKBpe-Tswv1b-hour-text'}).text.split('\xa0')[0])
    if open_now == '永久停業' or open_now == '暫時關閉':
        output['open_now'] = 'False'
    else:
        output['open_now'] = 'True'
    periods = []
    weekday_text = []
    for tr_ in shop_soup.find_all('tr'):
        if tr_.find('div').text.replace(' ','') != '':
            week = tr_.find('div').text
            time_list = [blank_check(i.text) for i in tr_.find_all('li')]
            for time_ in time_list:
                if time_ == '24 小時營業':
                    periods += [{
                        "open":{
                            "day": week_list[week], 
                            "time": 0000
                        },
                        "close":{
                            "day": week_list[week], 
                            "time": ''
                        }
                    }]
                elif time_ == '休息':
                    periods += [{
                        "open":{
                            "day": week_list[week], 
                            "time": ''
                        },
                        "close":{
                            "day": week_list[week], 
                            "time": ''
                        }
                    }]
                else:
                    start, end = time_.split('–')
                    end_hour, end_min = end.split(':')
                    start_hour, start_min = start.split(':')

                    if end_hour < start_hour:
                        end_day = week_list[week] + 1 
                    else:
                        end_day = week_list[week]

                    periods += [{
                        "open":{
                            "day": week_list[week], 
                            "time": start.replace(':','')
                        },
                        "close":{
                            "day": end_day, 
                            "time": end.replace(':','')
                        }
                    }]
            weekday_text += ["{}: {}".format(week, ', '.join(time_list))]
    output['periods'] = str(periods)
    output['weekday_text'] = str(weekday_text)

    return output


def get_reviews(driver, output):
    wait = WebDriverWait(driver, 30)
    more_reviews_css = "button[jsaction='pane.rating.moreReviews']"
    wait.until(
        EC.element_to_be_clickable((By.CSS_SELECTOR, more_reviews_css))
    )
    element = driver.find_element_by_css_selector(more_reviews_css)
    driver.implicitly_wait(20)
    ActionChains(driver).move_to_element(element).click(element).perform()
    time.sleep(2)

    all_photo = driver.find_elements_by_class_name('ODSEW-ShBeI-xJzy8c-bF1uUb')
    for ap in all_photo:
        ap.click()

    all_review = driver.find_elements_by_css_selector('button[aria-label="顯示更多"')
    for ap in all_review:
        ap.click()

    comment_soup = BeautifulSoup(driver.page_source, 'html.parser')
    count = 0
    reviews = []
    for comment in comment_soup.find_all('div',{'class':'ODSEW-ShBeI'}):
        comment_a_tag = comment.find_all('a')
        author_name = blank_check(comment_a_tag[1].find('div', class_= 'ODSEW-ShBeI-title').text)
        profile_photo_url = comment_a_tag[0].find('img')['src']
        rating = blank_check(comment.find('span',{'role':'img'})['aria-label'].replace('顆星', ''))
        text = comment.find('div', class_='ODSEW-ShBeI-ShBeI-content').text
        created_at = comment.find('span', class_='ODSEW-ShBeI-RgZmSc-date').text
        photos = []
        c = 0
        for i in comment.find_all('button', class_='ODSEW-ShBeI-xJzy8c'):
            path = i['style'].split(';')[0].split('url')[1].replace('\"','').replace('(','').replace(')','') 
            photos += [path]
            c += 1
            
        reviews += [{
            'id': comment.find('a')['href'].split('/')[5],
            'author_name': author_name,
            'profile_photo_url': profile_photo_url,
            'rating': int(rating), 
            'text': text,
            'created_at': created_at,
            'photos': photos
        }]
        count += 1

    output['reviews'] = reviews
    driver.back()
    return output

# def get_photo(output, shop_soup):
#     shop_photo = {}
#     for i in shop_soup.find('div',{'aria-label':'{}的相片'.format(output['name'])}).find_all('button'):
#         try:
#             if i['aria-label'] == '街景服務和 360 度相片' or i['aria-label'] == '影片':
#                 continue
                
#             shop_photo[i['aria-label']] = i.find('img')['src']
#         except:
#             pass
#     output['shop_photo'] = shop_photo
#     return output


def find_photo_list(driver):
    wait = WebDriverWait(driver, 30)
    wait.until(
        EC.element_to_be_clickable((By.CSS_SELECTOR, 'a[data-photo-index="5"]'))
    )
    for photo_id in range(6):
        driver.find_element(By.CSS_SELECTOR,'a[data-photo-index="{}"]'.format(photo_id)).send_keys(Keys.DOWN)
        time.sleep(1)
        
    photo_soup = BeautifulSoup(driver.page_source, 'html.parser')
    photo_url = []
    for photo_id in range(5):
        for i in photo_soup.select('a[data-photo-index="{}"]'.format(photo_id))[0].find_all('div'):
            if i['style'].find('width') != -1:
                sentence = i['style']
                photo = re.search(r'https:(.*)\"', sentence)
                photo_url += [photo.group(0).replace('\"','')]
                break
    return photo_url


def find_big_photo(output, driver):
    element = driver.find_element(By.CSS_SELECTOR, "div[aria-label='{}的相片']".format(output['name']))
    ActionChains(driver).move_to_element(element).click(element).perform()
    
    photo_map = {
        0: 'shop_photo',
        2: 'menu_photo'
    }
    
    for tab_index in [0, 2]:
        wait = WebDriverWait(driver, 30)
        wait.until(
            EC.element_to_be_clickable((By.CSS_SELECTOR, "button[data-tab-index='{}']".format(tab_index)))
        )
        element = driver.find_element(By.CSS_SELECTOR, "button[data-tab-index='{}']".format(tab_index))
        ActionChains(driver).move_to_element(element).click(element).perform()
        photo_list = find_photo_list(driver)
        output[photo_map[tab_index]] = photo_list
        
    for i in range(2):
        driver.back()
        time.sleep(1)
    return output

def get_url_list(driver):
    wait = WebDriverWait(driver, 10)
    wait.until(
        EC.element_to_be_clickable((By.XPATH, '//*[@id="sGi9mc-m5SR9c-bottom-pane"]/div/div[1]/div/div/div/div[1]/div[2]/div[2]'))
    )
    driver.back()

    time.sleep(2)
    for i in range(5, 43, 2):
        driver.find_element(By.XPATH,'//*[@id="pane"]/div/div[1]/div/div/div[2]/div[1]/div[{}]/div/a'.format(i)).send_keys(Keys.DOWN)
    url_soup = BeautifulSoup(driver.page_source, 'html.parser')
    url_list = []
    for i in url_soup.find_all('a'):
        try:
            if i['href'].find('maps/place') != -1:
                url_list += [[i['href'], i['aria-label']]]
        except:
            pass
    
    return url_list


def main():
    data = pd.read_csv('lat_long_location.csv', index_col = 0)
    tmp = data.iloc[0]
    latitude = tmp['latitude'] #緯度
    longitude = tmp['longitude'] #精度

    url = 'https://www.google.com.tw/maps/@{},{},15z?hl=zh-TW'.format(latitude,longitude)
    # driver = serive_create('Profile 1')

    db = DA.mysql_connect(MYSQL_CONFIG, DB_NAME)
    print('drvier start...')
    driver = brower_start()
    driver.get(url)
    keyin_keyword(driver, '咖啡')
    url_list = get_url_list(driver)

    result = []
    # try:
    for item_url, name in url_list:
        print(name, ': ' ,item_url)
        driver.get(item_url)

        wait = WebDriverWait(driver, 120)
        time_css =  "span[aria-label='顯示本週營業時間']"
        wait.until(
            EC.element_to_be_clickable((By.CSS_SELECTOR, time_css))
        )
        element = driver.find_element_by_css_selector(time_css)
        driver.implicitly_wait(30)
        ActionChains(driver).move_to_element(element).click(element).perform()

        time.sleep(1)
        shop_soup = BeautifulSoup(driver.page_source, 'html.parser')

        output = {
            'name': blank_check(shop_soup.find('h1', class_='x3AX1-LfntMc-header-title-title').text)
        }
        print(output['name'])

        output = get_shop_info(driver, output, shop_soup)

        output = get_intro_info(driver, output)

        output = get_time_list(shop_soup, output)

        output = get_reviews(driver, output)

        output = find_big_photo(output, driver)

        output_name = output['name'].replace('（','').replace('）', '')
        output['google_url'] = 'https://www.google.com.tw/search?q={}+{}'.format(output_name, output['addr'])
        time.sleep(2)
        result += [output]
        with open('result/20211207_{}.json'.format(name), 'w') as f:
            json.dump(output, f)

        DA.data_select_insert(db, SHOP_LIST_TABLE, SHOP_LIST_TABLE_COL, output)
        break
    # except:
        # shop_soup = BeautifulSoup(driver.page_source, 'html.parser')
        # print("error {}".format(id_))
        # print(blank_check(shop_soup.find('h1', class_='x3AX1-LfntMc-header-title-title').text))
   


if __name__ == '__main__':
    main()