import xlwt
import os
import xlwings as xw
import datetime
# 建行全局函数分为二部分 第一 写入数据(0创建目录和Excel并新建重命名,1 读取页面所有的数据,2开始写入表格)
# 第二部分是截图(创建时间格式的目录,然后保存图片)
#0 创建目录并新建excel重命名
def fond_create_name(fond_name):
newdir0 = r"E:/fund_data11"
if not os.path.exists(newdir0):
os.makedirs(newdir0)
workbook = xlwt.Workbook(encoding='utf-8')
# 创建sheet
data_sheet = workbook.add_sheet(fond_name)
excel_dir = newdir0 + '/' + fond_name + '.xls'
workbook.save(excel_dir)
return excel_dir
# if __name__ == "__main__":
# fond_create_name("东吴啊")
# 2 写入数据(前提先获取数据)
def write_data(fond_name):
# list1 = get_data_all_1() #获取读取数据
excel_dir = fond_create_name(fond_name) #创建目录并新建excel重命名
list1 = [1,3,4,1,1,2,3,4,5,6,6,4,3,99,775,4,3,222,35,45,546,47]
# print(list1)
iters = [iter(list1)] * 9
a_data = list(zip(*iters))
# print(a_data)
num = '3'
for v in a_data:
b_num = 'A' + str(num)
v = list(v) # 本身是个生成器 不加list是不能写入成功
# print('v',v)
wb = xw.Book(excel_dir)
sht = wb.sheets[0]
sht.range(b_num).value = v
num = str(int(num) + 1)
wb.save()
print("写入成功111")
return ('写入成功')
# write_data("王小天") #全局函数调用位置1================================================
# 3 以前一天时间命令的完整目录(重新命名目录名字和时间格式)
def time_dir(fund_dir):
now_time = datetime.datetime.now()
yesterday = now_time + datetime.timedelta(days=-1)
yesterday = yesterday.strftime('%Y%m%d')
newdir = "E:\fund_data\" + fund_dir + "\pic" + str(yesterday)
print(newdir,"-========")
if not os.path.exists(newdir):
os.makedirs(newdir)
print('====newdir====', newdir)
return newdir
# 4截图 命名保存
def move_pic(fund_dir, fund_name):
newdir = time_dir(fund_dir)
path = "E:\fund_data\imgsss\" # 截图的临时目录临时目录
if not os.path.exists(path):
os.makedirs(path)
dirs_pic = os.listdir(path) # 把获取的图片名字按字母排序
print(dirs_pic)
# 对以上获取按字母排序的目录文件列表排序 以是按照创建时间对文件排序
oldfile_list = sorted(dirs_pic, key=lambda x: os.path.getctime(os.path.join(path, x)))
# dir_list = sorted(dir_list,key=lambda x: os.path.getctime(os.path.join(DIR,x)))
oldfile = path + oldfile_list[0] #从临时目录去除按时间先后的图片 完整路径
print("1oldfile=== =========", oldfile)
# 获取临时目录下的文件列表
newfile = newdir + "\" + fund_name + ".jpg"
# 新文件名字,bankname是传入的参数
# print(oldfile)
# print(newfile)
if os.path.exists(newfile): # 必须先判断目标目录是否存在文件
os.remove(newfile)#如果存在相同的就删除
os.rename(oldfile, newfile)
move_pic("基金", "小金2")#全局函数调用处2=================================================