-
Notifications
You must be signed in to change notification settings - Fork 0
/
basic_info.py
147 lines (143 loc) · 6.26 KB
/
basic_info.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
import re
from config import spider_result_file_name
from excel_util import *
# 导出企业基本信息
def export_basic_inf(data_list, workbook, is_new):
start_row = 1 # 数据起始excel行号
if is_new:
worksheet = workbook.add_sheet("基本信息", cell_overwrite_ok=True)
worksheet.write(0, 0, "序号", style_title)
worksheet.write(0, 1, "公司名称", style_title)
worksheet.write(0, 2, "法人", style_title)
worksheet.write(0, 3, "电话", style_title)
worksheet.write(0, 4, "地址", style_title)
worksheet.write(0, 5, "邮箱", style_title)
worksheet.write(0, 6, "注册号", style_title)
worksheet.write(0, 7, "统一社会信用代码", style_title)
worksheet.write(0, 8, "注册资本", style_title)
worksheet.write(0, 9, "成立日期", style_title)
worksheet.write(0, 10, "企业类型", style_title)
worksheet.write(0, 11, "经营范围", style_title)
worksheet.write(0, 12, "公司住所", style_title)
worksheet.write(0, 13, "营业期限", style_title)
worksheet.write(0, 14, "企业状态", style_title)
# 设置表格列宽
for i in range(15):
if i == 0:
continue
elif i == 11:
worksheet.col(i).width = 256 * 50 # 设置第二列宽度，256为衡量单位，50表示50个字符宽度
else:
worksheet.col(i).width = 256 * 20
else:
start_row = read_excel_rows(spider_result_file_name,0)
worksheet = workbook.get_sheet(0)
for _response in data_list:
worksheet.write(start_row, 0, start_row, style)
# 公司名称
company_name = re.findall('<div class="company-name">(.*?\s*)<', _response)
if len(company_name) > 0:
company_name = company_name[0].strip()
else:
company_name = '--'
print('公司名称：' + company_name)
worksheet.write(start_row, 1, company_name, style) # 将信息输入表格
# 法人
legal_person = re.findall('<a class="oper" href=".*?">(.*?\s*)</a>', _response)
if len(legal_person) > 0:
legal_person = legal_person[0].strip()
else:
legal_person = '--'
print('法人：' + legal_person)
worksheet.write(start_row, 2, legal_person, style) # 将信息输入表格
# 电话
telephone = re.findall('<a href="tel:.*?" class="phone a-decoration">(.*?\s*)</a>', _response)
if len(telephone) > 0:
telephone = telephone[0].strip()
else:
telephone = '--'
print('电话：' + telephone)
worksheet.write(start_row, 3, telephone, style) # 将信息输入表格
# 地址
address = re.findall('</div> <div class="address">(\s*.*?\s*)</div> </div>', _response)
if len(address) > 0:
address = address[0].strip()
else:
address = '--'
print('地址：' + address)
worksheet.write(start_row, 4, address, style) # 将信息输入表格
# 邮箱
email = re.findall('<a href="javascript:;" class="email a-decoration">(.*?\s*)</a>', _response)
if len(email) > 0:
email = email[0].strip()
else:
email = '--'
print('邮箱：' + email)
worksheet.write(start_row, 5, email, style) # 将信息输入表格
basic_list = re.findall('</div> <div class="basic-item-right">(\s*.*?\s*)</div>', _response)
if len(basic_list) > 0:
# 注册号
registration_number = basic_list[0].strip()
else:
registration_number = '--'
print('注册号：' + registration_number)
worksheet.write(start_row, 6, registration_number, style) # 将信息输入表格
if len(basic_list) > 1:
# 统一社会信用代码
credit_ode = basic_list[1].strip()
else:
credit_ode = '--'
print('统一社会信用代码：' + credit_ode)
worksheet.write(start_row, 7, credit_ode, style) # 将信息输入表格
if len(basic_list) > 2:
# 注册资本
registration_capital = basic_list[2].strip()
else:
registration_capital = '--'
print('注册资本：' + registration_capital)
worksheet.write(start_row, 8, registration_capital, style) # 将信息输入表格
if len(basic_list) > 3:
# 成立日期
registration_date = basic_list[3].strip()
else:
registration_date = '--'
print('成立日期：' + registration_date)
worksheet.write(start_row, 9, registration_date, style) # 将信息输入表格
if len(basic_list) > 4:
# 企业类型
company_type = basic_list[4].strip()
else:
company_type = '--'
print('企业类型：' + company_type)
worksheet.write(start_row, 10, company_type, style) # 将信息输入表格
if len(basic_list) > 5:
# 经营范围
business_scope = basic_list[5].strip()
else:
business_scope = '--'
print('经营范围：' + business_scope)
worksheet.write(start_row, 11, business_scope, style) # 将信息输入表格
if len(basic_list) > 6:
# 公司住所
company_address = basic_list[6].strip()
else:
company_address = '--'
print('公司住所：' + company_address)
worksheet.write(start_row, 12, company_address, style) # 将信息输入表格
if len(basic_list) > 7:
# 营业期限
business_term = basic_list[7].strip()
else:
business_term = '--'
print('营业期限：' + business_term)
worksheet.write(start_row, 13, business_term, style) # 将信息输入表格
if len(basic_list) > 8:
# 企业状态
enterprise_state = basic_list[8].strip()
else:
enterprise_state = '--'
print('企业状态：' + enterprise_state)
worksheet.write(start_row, 14, enterprise_state, style) # 将信息输入表格
start_row += 1
# print('----------------------------------------------------------------------')
return worksheet