爬虫—天眼查接口函数

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(\'科润智能\'))