54 lines
2.4 KiB
Python
54 lines
2.4 KiB
Python
|
import os
|
|||
|
|
import zipfile
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main(zip_path=r'C:\Users\Administrator\Desktop\材料管理系统模版\造价站近两年价格数据.zip'):
|
|||
|
|
with zipfile.ZipFile(zip_path, 'r') as zip_ref:
|
|||
|
|
for file_info in zip_ref.infolist():
|
|||
|
|
filename = file_info.filename.encode('cp437').decode('gbk')
|
|||
|
|
is_excel = filename.endswith('.xlsx') or filename.endswith('.xls')
|
|||
|
|
is_fuzhou = '福州' in filename
|
|||
|
|
# 地材
|
|||
|
|
# if is_excel and is_fuzhou and '地材汇总/' in filename and '福州市交通工程地方材料价格信息汇总表(数据处理)' not in filename\
|
|||
|
|
# and '混凝土价格' not in filename:
|
|||
|
|
# year, month = get_month(filename)
|
|||
|
|
# print(year, month, filename)
|
|||
|
|
# 调查表
|
|||
|
|
if is_excel and is_fuzhou and '11地市调查表/' in filename and '福建省' not in filename and '通闽公司' not in filename and '-' not in filename:
|
|||
|
|
year, month = get_month(filename)
|
|||
|
|
print(year, month, filename)
|
|||
|
|
target_file_path = f'./{year}-{month}-调查表-福州{Path(filename).suffix}'
|
|||
|
|
with open(target_file_path, 'wb') as f:
|
|||
|
|
f.write(zip_ref.read(file_info))
|
|||
|
|
# 公路局
|
|||
|
|
# if is_excel and is_fuzhou and '4.福州公路' in filename:
|
|||
|
|
# print(filename)
|
|||
|
|
# 交通局
|
|||
|
|
# if is_excel and is_fuzhou and '1.福州交通局' in filename:
|
|||
|
|
# print(filename)
|
|||
|
|
# 网络价格
|
|||
|
|
# if is_excel and is_fuzhou and '网络价格（' in filename and '通闽公司' not in filename and '5.福州' not in filename:
|
|||
|
|
# print(filename)
|
|||
|
|
# 改性剂
|
|||
|
|
# if is_excel and '改性剂和沙钢' in filename :
|
|||
|
|
# print(filename)
|
|||
|
|
# 三明钢铁
|
|||
|
|
# if is_excel and '2.三明钢铁' in filename:
|
|||
|
|
# print(filename)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_month(filename):
|
|||
|
|
if '年信息价/' in filename:
|
|||
|
|
year = filename.split('年信息价')[0].split('/')[-1]
|
|||
|
|
month = filename.split('月发布）')[0].split('（')[-1]
|
|||
|
|
return year, month
|
|||
|
|
elif '月份地材汇总/' in filename:
|
|||
|
|
year = filename.split('年地材/')[0].split('/')[-1]
|
|||
|
|
month = filename.split('月份地材汇总')[0].split('年')[-1]
|
|||
|
|
return year, month
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
main()
|