#!usr/bin/python3

"""

@author: Jokin

@link: https://github.com/jokin1999/SMMS_Uploader

"""

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

import os

import re

import time

import base64

import tempfile

import threading

import tkinter as tk

import tkinter.ttk

import tkinter.messagebox

import tkinter.filedialog

import win32clipboard as wcb

import ico

from scanner import scanner

from smms import smms

from cloud import smcloud

# 消息盒子定义

def msgbox(message, title='提示', type='info'):

tk.messagebox.showerror(title=title, message=message)

# 文件选择器

def selector(mode=0):

lsbox_files.set(_files)

def listboxRenew(Listbox_var, list):

Listbox_var.set(list)

def listRemove(list, removalList):

for li in removalList:

list.remove(li)

return list

def getListboxValueByList(listbox, list):

return values

def _lsbox_remove():

global _files

listboxRenew(lsbox_files, listRemove(_files, getListboxValueByList(lsbox, list(lsbox.curselection()))))

lsbox.select_clear(0, len(_files)-1)

def _treeview_copy(type):

wcb.CloseClipboard()

def _treeview_delete():

show_status('删除时出错，删除失败')

def _lsbx_rb(event):

if not (len(lsbox.curselection()) == 0):

lsbx_rbmenu.post(event.x_root, event.y_root)

def _treeview_rb(event):

if not (len(treeview.selection()) == 0):

treeview_rbmenu.post(event.x_root, event.y_root)

# 读取上传列表

def readSuccessList(filename='./save.txt'):

return list

def switch_list(filename='./save.txt'):

treeview.insert('', list_data.index(sup), value=sup)

def operating_area(state=0):

btn_pause['state'] = tk.NORMAL

# 开始上传

def _start_upload():

t_upload['thread'].start()

def show_status(text):

label3_bottom['text'] = text

# 停止上传

def _stop_upload():

show_status('就绪')

# 变量调整器

def _t_upload(var, value):

global t_upload

t_upload[var] = value

# 上传队列中的文件

def upload(Listbox_var):

operating_area(0)

if __name__ == '__main__':

# 版本定义

VERSION = '1.0.7'

# 上传延迟

upload_delay = 0

# 多线程定义

t_upload = {}

# 文件选择器返回值

_files = []

# 当前Treeview listfile

TVFILE = './save.txt'

# 上传模式定义

upload_modes = [('单个上传', 0, 'normal'), ('群组上传', 1, 'disable')]

# 选择模式定义

selector_modes = [('单文件选择', 0, 'normal'), ('多文件选择', 1, 'normal'), ('文件夹遍历', 2, 'normal'), ('单层文件夹', 3, 'disable')]

# 创建窗口

win = tk.Tk()

win.resizable(width=False, height=False)

tempfile = tempfile.mktemp()

icon = open(tempfile, 'wb')

icon.write(base64.b64decode(ico.ico))

icon.close()

win.iconbitmap(tempfile)

os.remove(tempfile)

win.title('SMMS图床上传工具 ' + VERSION)

# 读取已上传列表

sUpload = readSuccessList()

mainFrame = tk.Frame(win)

mainFrame.grid()

# 模式标签框架

lf_mode = tk.LabelFrame(mainFrame, text='上传模式', fg='blue')

lf_mode.grid(row=1, column=0, padx=10, pady=10, sticky=tk.N+tk.S+tk.E+tk.W)

# 创建RadioButton

vUpload = tk.IntVar()

for mode, num, state in upload_modes:

rb = tk.Radiobutton(lf_mode, text=mode, variable=vUpload, value=num, state=state)

rb.pack()

vUpload.set(0)

# 其他选项框架

lf_others = tk.LabelFrame(mainFrame, text='其他', fg='blue')

lf_others.grid(row=0, column=3, rowspan=2, padx=10, pady=10, sticky=tk.N+tk.S+tk.E+tk.W)

vDR = tk.IntVar()

vSL = tk.IntVar()

vSL.set(1)

cb_duplicateRemoval = tk.Checkbutton(lf_others, text='检查上传重复（本地）', variable=vDR, state=tk.DISABLED)

cb_duplicateRemoval.grid(row=0, column=0, sticky=tk.W)

cb_sizeLimitation = tk.Checkbutton(lf_others, text='过滤大于5M的文件', variable=vSL, state=tk.NORMAL)

cb_sizeLimitation.grid(row=1, column=0, sticky=tk.W)

btn_cloud = tk.Button(lf_others, text='切换中继', command=smcloud)

btn_cloud.grid(row=2, column=0, sticky=tk.W+tk.E, padx=5)

btn_relay_list = tk.Button(lf_others, text='切换已转换列表', command=lambda: switch_list('./save2.txt'))

btn_relay_list.grid(row=3, column=0, sticky=tk.W+tk.E, padx=5)

btn_origin_list = tk.Button(lf_others, text='切换已上传列表', command=lambda: switch_list('./save.txt'))

btn_origin_list.grid(row=4, column=0, sticky=tk.W+tk.E, padx=5)

# 选择器模式标签框架

lf_selector_mode = tk.LabelFrame(mainFrame, text='选择器模式', fg='blue')

lf_selector_mode.grid(row=0, column=0, padx=10, pady=10, sticky=tk.N+tk.S+tk.E+tk.W)

# RadioButton

vSelector = tk.IntVar()

for mode, num, state in selector_modes:

rb = tk.Radiobutton(lf_selector_mode, text=mode, variable=vSelector, value=num, state=state)

rb.pack()

# 默认多文件选择

vSelector.set(1)

# 操作区

lf_operator = tk.LabelFrame(mainFrame, text='操作区', fg='blue')

lf_operator.grid(row=0, rowspan=2, column=1, padx=10, pady=10, sticky=tk.N+tk.S+tk.E+tk.W)

# RadioButton

btn_selector = tk.Button(lf_operator, text='选择', width=10, command=lambda: selector(vSelector.get()))

btn_selector.grid(padx=10, pady=10, sticky=tk.N+tk.S+tk.E+tk.W)

btn_upload = tk.Button(lf_operator, text='上传', width=10, command=lambda: _start_upload())

btn_upload.grid(padx=10, pady=10, sticky=tk.N+tk.S+tk.E+tk.W)

btn_pause = tk.Button(lf_operator, text='暂停', width=10, command=lambda: _stop_upload())

btn_pause['state'] = tk.DISABLED

btn_pause.grid(padx=10, pady=10, sticky=tk.N+tk.S+tk.E+tk.W)

# 已选文件列表

lf_lsbox = tk.LabelFrame(mainFrame, text='等待上传文件列表', fg='red')

lf_lsbox.grid(row=0, rowspan=2, column=2, padx=10, pady=10, sticky=tk.N+tk.S+tk.E+tk.W)

lsbox_files = tk.StringVar()

lsbox = tk.Listbox(lf_lsbox, selectmode=tk.EXTENDED, listvariable=lsbox_files, width=75, relief=tk.FLAT)

lsbox.grid(row=0, column=0)

# 右键菜单

lsbx_rbmenu = tk.Menu(lsbox, tearoff=False)

lsbx_rbmenu.add_command(label='删除', command=_lsbox_remove)

# lsbox.bind('<Button-3>', func=lambda event: lsbx_rbmenu.post(event.x_root, event.y_root))

lsbox.bind('<ButtonRelease-3>', func=lambda event: _lsbx_rb(event))

# 滚动条

lsbox_yscrollbar = tk.Scrollbar(lf_lsbox, command=lsbox.yview)

lsbox_yscrollbar.grid(row=0, column=1, sticky=tk.N+tk.S+tk.E+tk.W)

lsbox.config(yscrollcommand=lsbox_yscrollbar.set)

lsbox_xscrollbar = tk.Scrollbar(lf_lsbox, command=lsbox.xview, orient='horizontal')

lsbox_xscrollbar.grid(row=1, column=0, sticky=tk.N+tk.S+tk.E+tk.W)

lsbox.config(xscrollcommand=lsbox_xscrollbar.set)

# 已上传文件列表

lf_treeview = tk.LabelFrame(mainFrame, text='已上传文件列表（右键复制 | CTRL多选）', fg='green')

lf_treeview.grid(row=2, column=0, columnspan=4, padx=10, pady=5, sticky=tk.N+tk.S+tk.E+tk.W)

treeview = tk.ttk.Treeview(lf_treeview, columns=['path', 'cdn', 'delete'], show='headings', height=10)

treeview.grid(sticky=tk.N+tk.S+tk.E+tk.W)

treeview.column('path', width=550, anchor='w')

treeview.column('cdn', width=200, anchor='w')

treeview.column('delete', width=200, anchor='w')

treeview.heading('path', text='本地路径')

treeview.heading('cdn', text='CDN链接')

treeview.heading('delete', text='删除链接')

for sup in sUpload:

treeview.insert('', sUpload.index(sup), value=sup)

# id = treeview.insert('', sUpload.index(sup), value=sup)

# 滚动条

treeview_sby = tk.Scrollbar(lf_treeview, orient=tk.VERTICAL, command=treeview.yview)

treeview.configure(yscrollcommand=treeview_sby.set)

treeview_sby.grid(row=0, column=1, sticky=tk.N+tk.S)

# 右键菜单

treeview_rbmenu = tk.Menu(treeview, tearoff=False)

treeview_rbmenu.add_command(label='复制CDN地址', command=lambda: _treeview_copy(1))

treeview_rbmenu.add_command(label='复制删除地址', command=lambda: _treeview_copy(2))

treeview_rbmenu.add_command(label='复制本地路径', command=lambda: _treeview_copy(0))

treeview_rbmenu.add_separator()

treeview_rbmenu.add_command(label='删除记录', command=lambda: _treeview_delete())

treeview.bind('<ButtonRelease-3>', func=lambda event: _treeview_rb(event))

# Footer

label_bottom = tk.Label(mainFrame, text='使用须知', fg='#878787')

label_bottom_info = '1、不得将本工具用作任何商业用途\n2、请严格遵守SM.MS图床的相关使用规定\n3、上传的内容与本工具及本工具作者无关'

label_bottom.bind('<Button-1>', lambda t: msgbox(label_bottom_info, title='使用须知'))

label_bottom.grid(row=3, column=0, padx=10, sticky=tk.W)

label2_bottom = tk.Label(mainFrame, text='Made by Joe', fg='#878787')

label2_bottom_info = '作者：Joe\n鸣谢：SM.MS图床\n本程序开源免费，若有不良商家售卖，给差评！'

label2_bottom.bind('<Double-Button-1>', lambda t: msgbox(label2_bottom_info, title='关于作者'))

label2_bottom.grid(row=3, column=3, padx=5, sticky=tk.E)

label3_bottom = tk.Label(mainFrame, text='就绪', fg='#fa6333')

label3_bottom.grid(row=3, column=1, columnspan=2, padx=10, sticky=tk.W+tk.S)

# 循环

tk.mainloop()