imageXexcel.py

# -*- coding: utf-8 -*-

import os
import io
from configparser import ConfigParser
from contextlib import redirect_stdout

import pyperclip
import tkfilebrowser
import tkinter as tk
import tkinter.font as tkfont
from tkinter import filedialog, ttk

from insert_images import dirs_images_insert_excels


cfg = ConfigParser(allow_no_value=True)
cfg.read('config.ini', encoding='utf-8')

img_root_dir = cfg['DEFAULT']['img_root_dir']
img_dir_name_list = cfg['DEFAULT']['img_dir_name_list'].split()
excel_file_name = cfg['DEFAULT']['excel_file_name']
sheet_name = cfg['DEFAULT']['sheet_name']
label_template = cfg['DEFAULT']['label_template'].split()

# 开启窗口
window = tk.Tk()
window.title('imageXexcel')
window.geometry('800x550')

# 定义几个字体格式
ft = tkfont.Font(family='Arial', size=10, weight=tkfont.BOLD)
ft1 = tkfont.Font(size=20, slant=tkfont.ITALIC)
ft2 = tkfont.Font(size=30, weight=tkfont.BOLD, underline=1, overstrike=1)


def update_config_file(cfg_key, cfg_value, config_section='DEFAULT'):
    """更新配置文件"""
    cfg[config_section][cfg_key] = cfg_value

    with open('config.ini', 'w', encoding='utf-8') as configfile:
        cfg.write(configfile)


def list_to_str(ls):
    """
    将元素为tuple的list，转换为元组元素以空格隔开的一个字符串
    """
    new_tup = ()
    for i in range(len(ls)):
        tup = ls[i]
        # print(tup)
        new_tup += tup
    # print(new_tup, type(new_tup))
    new_set = set(new_tup)
    # print(new_set, type(new_set))

    new_str = ''
    for s in new_set:
        ss = s + ' '
        new_str += ss
    # print(new_str)
    return new_str


def list_to_list(ls):
    new_tup = []
    for i in range(len(ls)):
        tup = ls[i]
        new_tup += tup
    new_ls = list(set(new_tup))
    return new_ls


def list_to_str_2(ls):
    new_str = ''
    for s in ls:
        ss = s + ' '
        new_str += ss
    return new_str


# 定义GUI界面

# 定义 LabelFrame1 部件
lf_input = ttk.LabelFrame(window, text='输入文件：', height=20, width=100)
lf_input.grid(row=0, column=0, padx=1, pady=15)


# 定义excel模板文件选择部件
def select_file():
    file_path_selected = filedialog.askopenfilename()
    file_path.set(file_path_selected)
    print('选中文件：', file_path_selected, type(file_path_selected))
    update_config_file('excel_file_name', file_path_selected)


file_path = tk.StringVar()
tk.Label(window, text='Excel模板文件路径:', width=15, height=2, foreground='green').grid(row=3, column=0)
e_template = tk.Entry(window, textvariable=file_path, width=80, bd=3)
e_template.grid(row=3, column=1)
tk.Button(window, text='选择文件', command=select_file, width=12,
          foreground='green', background='lightgreen').grid(row=3, column=2)


# 定义excel模板文件sheet name选择部件
def input_worksheet_name():
    ws_name = worksheet_name.get()
    # worksheet_name.set(ws_name)
    print('工作表名：', ws_name, type(ws_name))
    update_config_file('sheet_name', ws_name)


worksheet_name = tk.StringVar()
worksheet_name.set('勘察照片')

tk.Label(window, text='Excel模板工作表名:', width=15, height=2, foreground='green').grid(row=4, column=0)
e_sheet = tk.Entry(window, textvariable=worksheet_name, width=80, foreground='gray', bd=3)
e_sheet.grid(row=4, column=1)
tk.Button(window, text='确认工作表名', command=input_worksheet_name,
          width=12, foreground='green', background='lightgreen').grid(row=4, column=2)


# 定义照片文件夹选择部件
dir_path_selected = []


def select_dir_path():
    win_user_desktop = os.path.join(os.path.expanduser('~'), 'Desktop')
    dir_path_selected.append(tkfilebrowser.askopendirnames(title='选择照片文件夹', initialdir=win_user_desktop,
                                                           okbuttontext='打开', cancelbuttontext='取消'))

    transform_dir_path_selected_ls = list_to_list(dir_path_selected)

    print('选中文件夹：', transform_dir_path_selected_ls, type(transform_dir_path_selected_ls))
    dir_path.set(transform_dir_path_selected_ls)

    # 选中文件夹为一个列表(列表元素为每次选择的元组)，需要将其转换为字符串，再更新到配置文件
    transform_dir_path_selected_str = list_to_str(dir_path_selected)

    update_config_file('img_dir_name_list', transform_dir_path_selected_str)


dir_path = tk.StringVar()
tk.Label(window, text='图片文件夹路径:', width=15, height=2, foreground='green').grid(row=1, column=0)
lb_excel_dir = tk.Listbox(window, listvariable=dir_path, width=80, height=3, bd=3)
lb_excel_dir.grid(row=1, column=1)
tk.Button(window, text='选择文件夹', command=select_dir_path, width=12,
          foreground='green', background='lightgreen').grid(row=1, column=2)


# 定义删除选中文件夹路径部件
def delete_selection_path():
    # 获取选中的路径
    selection_path_cursor = lb_excel_dir.curselection()

    # 更新配置文件
    content = lb_excel_dir.get(selection_path_cursor)
    contents_tuple = lb_excel_dir.get(0, 'end')

    contents_list = list(contents_tuple)
    contents_list.remove(content)

    print('目前所有项：', contents_tuple, type(contents_tuple))
    print('删除选中项：', content, type(content))
    print('剩余所有项：', contents_list, type(contents_list))

    update_config_file('img_dir_name_list', list_to_str_2(contents_list))

    # 删除excel dir ListBox中的显示
    lb_excel_dir.delete(selection_path_cursor)


tk.Button(window, text='删除选中文件夹', command=delete_selection_path, width=12,
          foreground='orange').grid(row=2, column=1)


# 定义运行脚本部件
def run():
    # 需要在使用前即时从配置文件读取最新值
    img_dir_name_list = cfg['DEFAULT']['img_dir_name_list'].split()
    print('获取的照片文件夹列表：', img_dir_name_list)

    f_in = io.StringIO()
    with redirect_stdout(f_in):
        # 执行图片插入excel脚本
        excel_files_list = \
            dirs_images_insert_excels(img_dir_name_list, label_template,
                                      sheet_name, excel_file_name)
    f_out = f_in.getvalue()

    # 显示输出日志
    # t_log.delete(0.0, 'end')
    t_log.insert('end', f_out)
    t_log.see('end')

    print('生成的excel list返回值：', excel_files_list)
    # 清空excel listbox内容
    t_new_excel.delete(0, 'end')
    # 将生成的excel文件填入excel部件
    excel_files_list.reverse()
    for f in excel_files_list:
        t_new_excel.insert('end', f)


run_button = tk.Button(window, text='开始运行', command=run, width=12, foreground='red', background='pink')
run_button.grid(row=5, column=2, padx=0, pady=15)

# 定义 LabelFrame2 部件
lf_output = ttk.LabelFrame(window, text='输出文件：', height=20, width=100)
lf_output.grid(row=6, column=0, padx=1, pady=10)


# 定义生成的excel文件部件
def cd_excel_files_dir():
    # 获取t_new_excel中第一个excel路径
    excel_file_path = t_new_excel.get(0)
    excel_root_path = os.path.dirname(excel_file_path)

    # 打开系统路径
    arg = 'start explorer' + ' ' + excel_root_path
    print('进入目录：', arg)
    os.startfile(excel_root_path)


tk.Label(window, text='输出Excel文件:', width=15, height=2, foreground='blue').grid(row=7, column=0)
t_new_excel = tk.Listbox(window, width=80, height=3, bd=3)
t_new_excel.grid(row=7, column=1, padx=1, pady=10)
tk.Button(window, text='进入文件夹', command=cd_excel_files_dir, width=12,
          foreground='blue', background='lightblue').grid(row=7, column=2)


# 定义输出日志部件
def copy_output_log():
    log = t_log.get('1.0', 'end')
    pyperclip.copy(log)


tk.Label(window, text='输出日志:', width=15, height=2, foreground='blue').grid(row=8, column=0)

t_log = tk.Text(window, width=80, height=6, bd=3)
t_log.grid(row=8, column=1, padx=1, pady=10)

tk.Button(window, text='复制日志', command=copy_output_log, width=12,
          foreground='blue', background='lightblue').grid(row=8, column=2)


window.mainloop()