Zstd DC.py

import zstd
import tkinter as tk
from tkinter import *
from tkinter import filedialog as fd
from collections import deque

class Window:

    def __init__(self, master):
        self.master = master
        self.master.option_add("*Font", "Verdana 12")
 
        self.Main = Frame(self.master)
 
        self.stack = deque(maxlen = 10)
        self.stackcursor = 0
 
        self.L1 = Label(self.Main, text = "Zstd DC - Zstandard (De-)Compresser v1.0")
        self.L1.pack(padx = 5, pady = 0)

        self.menu = Menu(self.Main)
         
        filemenu = Menu(self.menu, tearoff=0)
        filemenu.add_command(label="Compress", command = self.Compress, font = ("Verdana", 10))
        filemenu.add_command(label="Decompress", command = self.Decompress, font = ("Verdana", 10))
        filemenu.add_command(label="Mass compress", command = self.Mcompress, font = ("Verdana", 10))
        filemenu.add_command(label="Mass decompress", command = self.Mdecompress, font = ("Verdana", 10))
        filemenu.add_command(label="Exit", command = root.destroy, font = ("Verdana", 10))
         
        self.menu.add_cascade(label="File",
                             menu=filemenu)
        self.menu.add_command(label = "About", command = self.About)
        self.master.config(menu = self.menu)
 
        self.B1 = Button(self.Main, text = "Decompress", width = 12, command = self.Decompress)
        self.B1.pack(padx = 5, pady = 5, side = LEFT)

        self.B2 = Button(self.Main, text = "Compress", width = 12, command = self.Compress)
        self.B2.pack(padx = 5, pady = 5, side = RIGHT)

        self.B3 = Button(self.Main, text = "Mass decompress", width = 15, command = self.Mdecompress)
        self.B3.pack(padx = 5, pady = 5, side = LEFT)

        self.B4 = Button(self.Main, text = "Mass compress", width = 15, command = self.Mcompress)
        self.B4.pack(padx = 5, pady = 5, side = RIGHT)

        self.L2 = Label(self.Main, text="""Compression level (1-9)""")
        self.L3 = Label(self.Main, text="""Extension (mass only)""")
        self.T1 = Text(root, height=1, width = 22)
        self.T2 = Text(root, height=1, width = 22)
 
        self.Main.pack(padx = 15, pady = 15)
        self.T1.pack(padx=5, pady=5)
        self.T1.insert(tk.END, "Compression level (1-22)")
        self.T1.bind("<Button-1>", lambda args: self.T1.delete('1.0', END))
        self.T2.pack(padx=5, pady=5)
        self.T2.insert(tk.END, "Extension (mass only)")        
        self.T2.bind("<Button-1>", lambda args: self.T2.delete('1.0', END))

    
    def Compress(self):
        try:
            levelc=int(self.T1.get(1.0, "end-1c"))
        except:
            levelc=3

        if levelc <= -100:
            levelc = -100
        if levelc > 22:
            levelc = 22
        
        ftypes = [('All files', '*')]
        filename = fd.askopenfilename(filetypes=ftypes)
        if filename != "" and filename is not None:
            with open(filename, 'rb') as myfile:
                f = myfile.read()
            ftypes = [('All files', '*')]
            newfilename = fd.asksaveasfilename(filetypes=ftypes, initialfile=filename)
            z = open(newfilename, "ab")
            z.write(zstd.compress(f,levelc))
            tk.messagebox.showinfo(title='Success!', message='File compressed successfully!')
            myfile.close()

    def Decompress(self):
        ftypes = [('All files', '*')]
        filename = fd.askopenfilename(filetypes=ftypes)
        if filename != "" and filename is not None:
            with open(filename, 'rb') as myfile:
                f = myfile.read()
            ftypes = [('All files', '*')]
            newfilename = fd.asksaveasfilename(filetypes=ftypes, initialfile=filename)
            z = open(newfilename, "ab")
            z.write(zstd.decompress(f))
            tk.messagebox.showinfo(title='Success!', message='File decompressed successfully!')
            myfile.close()

    def Mdecompress(self):
        endel = None
        try:
            endi=self.T2.get(1.0, "end-1c")
        except:
            endi=".txt"

        if endi == "" or endi is None:
            endi=".txt"
        if endi.find(".")== -1 and not endi.startswith("-"):
            endi=".txt"

        if endi.startswith("-"):
            endel = True
            endi = endi.replace("-", "")
            
        ftypes = [('All files', '*')]
        filename = fd.askopenfilenames(filetypes=ftypes)
        if filename != "" and filename is not None:
            fnc = len(filename)
            while fnc != 0:
                fnz = filename[fnc-1].split(',')
                fnz2 = fnz[0]
                with open(fnz2, 'rb') as myfile:
                    f = myfile.read()
                if endel is None or endel is not True:
                    newfilename = fnz2+endi
                else:
                    newfilename = fnz2.replace(endi, "")
                z = open(newfilename, "ab")
                z.write(zstd.decompress(f))
                fnc = fnc-1
                myfile.close()
            tk.messagebox.showinfo(title='Success!', message='Files decompressed successfully!')
        else:
            return


    def Mcompress(self):
        endel = None
        try:
            endi=self.T2.get(1.0, "end-1c")
        except:
            endi=".zst"

        if endi == "" or endi is None:
            endi=".zst"

        if endi.find(".")== -1 and not endi.startswith("-"):
            endi=".zst"

        if endi.startswith("-"):
            endel = True
            endi = endi.replace("-", "")

        try:
            levelc=int(self.T1.get(1.0, "end-1c"))
        except:
            levelc=3

        if levelc <= -100:
            levelc = -100
        if levelc > 22:
            levelc = 22
            
        ftypes = [('All files', '*')]
        filename = fd.askopenfilenames(filetypes=ftypes)
        if filename != "" and filename is not None:
            fnc = len(filename)
            while fnc != 0:
                fnz = filename[fnc-1].split(',')
                fnz2 = fnz[0]
                with open(fnz2, 'rb') as myfile:
                    f = myfile.read()
                if endel is None or endel is not True:
                    newfilename = fnz2+endi
                else:
                    newfilename = fnz2.replace(endi, "")
                z = open(newfilename, "ab")
                z.write(zstd.compress(f,levelc))
                fnc = fnc-1
                myfile.close()
            tk.messagebox.showinfo(title='Success!', message='Files decompressed successfully!')
        else:
            return
        

    def About(self):
        tk.messagebox.showinfo(title='About', message='''This is a program to compress and decompress files, using Zstandard.

zstd was written by Yann Collet at Facebook.

Program author: dzhemvrot
Program version: 1.0
Program restributed using GPL-3.0 license''')

root = Tk()
root.resizable(False, False)
window = Window(root)
try:
    root.iconbitmap('icon.ico')
except:
    pass
root.title(u'Zstd DC')
root.mainloop()