'''
@Description 生成文件夹
@params (path) 文件夹路径
@params (title) 文件名称
@params (content) html 文件代码
@params (zip) 是否生成压缩文件夹
'''
import time
import random
import os
import zipfile # 压缩文件
def save_html(title, content=None, zip=False,path='html'):
isExists = os.path.exists(path)
# 默认的code
str1 = 'my name is :'
str2 = '--努力才能逃离--'
other = """<html><head></head><body><p>%s</p><p>%s</p></body></html>""" % (str1, str2)
if not isExists:
os.makedirs(path)
try:
# 命名生成的html
f = open(path+'/'+title+'.html', 'w', encoding='UTF-8')
if(content):
message = content
else:
message = other
f.write(message)
f.close()
print("正在压缩文件,请稍后...")
time.sleep(2)
if zip:
zp = zipfile.ZipFile('v_1.0.zip', 'w', zipfile.ZIP_DEFLATED)
list = os.listdir(path)
file_paths = []
for i in range(0, len(list)):
file_paths.append(path+'/'+list[i])
for i in range(0, len(file_paths)):
zp.write(file_paths[i])
zp.close()
time.sleep(3)
print('压缩完成!')
else:
print()
except Exception as e:
print(e)
return True
else:
print('文件夹已存在')
return False
02-04