爬蟲—天眼查接口函數

from selenium import webdriver
import time
import re
from bs4 import BeautifulSoup
import urllib

#獲取企業基本信息數據
def get_enterprise_data(ename):
#搜索頁面鏈接地址
keyword = urllib.parse.quote(ename)
url = 'http://www.tianyancha.com/search/'+keyword
#獲得搜索結果頁面
driver = webdriver.PhantomJS(executable_path='/root/phantomjs-2.1.1-linux-x86_64/bin/phantomjs')
driver.maximize_window()
driver.get(url)
time.sleep(2)
#從搜索結果中點擊第一個結果
driver.find_element_by_class_name('query_name').click()
time.sleep(2)
#抓取第一個結果的網頁，匹配出需要的字段
soup = BeautifulSoup(driver.page_source,"html.parser")
basic_info_list = soup.find_all('p',class_="ng-binding ng-scope")
data = []
qiyemingcheng = driver.title.split('】')[1].split('信息查詢')[0]
data.append(qiyemingcheng)
for i in basic_info_list:
data.append(i.get_text().strip())

return data

print(get_enterprise_data('科潤智能'))