import csv
import pymysql
from openpyxl import load_workbook
# 读取现有Excel
class MyFuncRreadExcel(object):
def __init__(self):
self.mysqlConfig = {
"host": "127.0.0.1",
"user": "root",
"password": "!@#$qwerasdfzxcv",
"database": "dbforpymysql"
}
# 读取Excel文件，获取 工作表 最大行数 最大列数
def Read_excel(self, xls_path):
'''
:param xls_path: excel path
:return: {'sheet': sheet, 'row_max': row_max, 'column_max': column_max} 当前工作表对象| 最大行数| 最大列数
'''
# 打开一个Excel
try:
wb = load_workbook(xls_path)
print('打开 %s 成功！' % xls_path)
except Exception as e:
print('打开 %s 失败！' % xls_path)
raise e
try:
# 获取当前工作表
sheet = wb.active
print('打开工作表成功！')
except Exception as e:
print('打开sheet失败！')
raise e
# 获取表的最大行数
row_max = sheet.max_row
print('\n商品表的最大行数为：', row_max)
# 获取表格的最大列数
column_max = sheet.max_column
print('\n商品表的最大列数为：', column_max)
result = {
'sheet': sheet,
'row_max': row_max,
'column_max': column_max,
}
return result
# 数据处理函数 处理每行中所有列的数据
def process_perRow(self, sheet, row_max, process_row_min=None, process_row_max=None):
'''
数据处理函数 处理每行中所有列的数据
:param sheet: 需要处理的工作表 类型：对象
:param row_max: excel 文件的最大行数
:param process_row_max: 处理文件的最大行数，默认为None，代表总行数
:param process_row_min: 处理文件的最小行数，默认为None，代表从第一行开始处理
:return: 暂时返回True，没有含义
'''
# 函数处理结果，有备无患
res = True
# 参数格式化
# 最大行数
if not process_row_max:
process_row_max = row_max
# 最小行数
if not process_row_min:
process_row_min = 1
# 处理行数计数
row_count = 0
# 获取每行信息
for row in sheet.rows:
row_count += 1
# 判断行数是否在 设定范围内
if process_row_min <= row_count <= process_row_max:
print('正在处理第 %d 行数据:\n' % row_count)
# 打印行内容
print(row)
# 处理列数计数
cell_count = 0
# 获取此行 每列的信息
for cell in row:
cell_count += 1
print(' 正在处理第 %d 行 %d 列 的数据\n' % (row_count, cell_count))
# 获取 列值
cell_value = cell.value
# 打印 列值
print(' 第 %d 行 %d 列 的数据为：' % (row_count, cell_count), cell_value, '\n')
# 此处填写 具体的处理逻辑
print(' 结束处理第 %d 行 %d 列 的数据\n' % (row_count, cell_count))
print('结束处理第 %d 行数据\n' % row_count)
return res
# 数据处理函数 处理每行中特定的某几列数据
def process_Selected_column(self, sheet, row_max, column_index, process_row_min=None, process_row_max=None):
'''
:param sheet: 需要处理的工作表 类型：对象
:param row_max: excel 文件的最大行数
:param process_row_max: 处理文件的最大行数，默认为None，代表总行数
:param process_row_min: 处理文件的最小行数，默认为None，代表从第一行开始处理
:return: [
[columnX_value, columnX_value, columnX_value, ... columnX_value], # 行
[columnX_value, columnX_value, columnX_value, ... columnX_value],
]
'''
# 函数处理结果:
result = []
# 参数格式化
# 最大行数
if not process_row_max:
process_row_max = row_max
# 最小行数
if not process_row_min:
process_row_min = 1
# 处理行数计数
row_count = 0
# 获取每行信息
for row in sheet.rows:
row_count += 1
# 判断行数是否在 设定范围内
if process_row_min <= row_count <= process_row_max:
print('正在处理第 %d 行数据:\n' % row_count)
# 打印行内容
# print(row)
# 保存每行特定列的结果
lst = []
# 根据要求 处理特定的几列 数据
for item in column_index:
item -= 1
lst.append(row[item].value)
# 每行的结果添加到result
result.append(lst)
print('结束处理第 %d 行数据\n' % row_count)
return result
# csv保存文件
def save2csv(self, filePath, data):
'''
:param filePath: 要保存的文件 路径+文件名
:param data: 需要保存的数据 格式：[[xxx, xxx, xxx, xxx], [xxx, xxx, xxx, xxx]]
:return: 无返回值
'''
# 多行写入(writerows([(),(),()]
# newline = '':在Windows中默认会添加1个空行，所以加入newline = ''参数，避免空行出现
with open(filePath, 'w', newline='', encoding='utf8') as f:
writer = csv.writer(f)
writer.writerows(data)
print(filePath, '写入完毕！')
# 数据库 操作 pymysql
# 数据库基础：连接 关闭数据库
def MySql_operation(self):
pass
# 主函数
def main(self, filePath):
# 调用入口函数, 获取字典 sheet row_max column_max
result_sheet_dict = MyFuncRreadExcel.Read_excel(filePath)
sheet = result_sheet_dict['sheet']
row_max = result_sheet_dict['row_max']
column_max = result_sheet_dict['column_max']
# 2、设定处理行数范围
# 最小行数 默认为1
process_row_min = int(input('请输入需要处理的最小行数：'))
# 最大行数 默认为None，代表处理到最后一行
process_row_max = input('请输入需要处理的最大行数：')
if not process_row_max:
process_row_max = None
else:
process_row_max = int(process_row_max)
# 3、设定 要处理的 特定列的索引值（从1开始算起）
str_user_choice = input('请输入需要处理特定列索引，用句号（英文格式‘.’）分隔：')
lst = str_user_choice.split('.')
column_index = [int(item) for item in lst if item != '']
# 二、用户选择：
print('功能菜单如下：')
print('''
1、处理每行的所有列；
2、获取特定行的特定列的值并打印；
3、获取特定行的特定列的值并保存为csv文件；
4、对特定行的特定列 进行数据处理；
''')
user_choice = int(input('请选择：'))
if user_choice == 1:
# 调用处理方法：处理每行的所有列
result_process_perRow = MyFuncRreadExcel.process_perRow(sheet, row_max, process_row_min, process_row_max)
elif user_choice == 2:
# 调用处理方法： 处理每行特定的几列
# 得到 特定行范围内的特定列的值
result_process_Selected_column = MyFuncRreadExcel.process_Selected_column(sheet, row_max, column_index,
process_row_min, process_row_max)
print(result_process_Selected_column)
elif user_choice == 3:
# 调用处理方法： 处理每行特定的几列
# 得到 特定行范围内的特定列的值
result_process_Selected_column = MyFuncRreadExcel.process_Selected_column(sheet, row_max, column_index,
process_row_min, process_row_max)
# 获取文件保存路径
filePath2csv = input("请输入您要保存csv文件的路径+文件名:")
# 调用csv保存方法
MyFuncRreadExcel.save2csv(filePath2csv, result_process_Selected_column)
elif user_choice == 4:
print("正在快马加鞭滴开发ing！")
if __name__ == '__main__':
# 一、常量配置
# 1、excel 文件路径
filePath = './链家二手房.xlsx'
# 实例化
MyFuncRreadExcel = MyFuncRreadExcel()
# 调用 main 函数
MyFuncRreadExcel.main(filePath)
