upx-packer.py

#     Copyright 2019, Jorj McKie, mailto:<jorj.x.mckie@outlook.de>
#
#     Part of "Nuitka", an optimizing Python compiler that is compatible and
#     integrates with CPython, but also works on its own.
#
#     Licensed under the Apache License, Version 2.0 (the "License");
#     you may not use this file except in compliance with the License.
#     You may obtain a copy of the License at
#
#        http://www.apache.org/licenses/LICENSE-2.0
#
#     Unless required by applicable law or agreed to in writing, software
#     distributed under the License is distributed on an "AS IS" BASIS,
#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#     See the License for the specific language governing permissions and
#     limitations under the License.
#

from __future__ import print_function
import sys, os, subprocess as sp, time

py2 = str is bytes                    # check if Python2
# do some adjustments whether Python v2 or v3
if not py2:
    import PySimpleGUI as psg
else:
    import PySimpleGUI27 as psg

sep_line = "".ljust(80, "-")

try:
    print(sep_line)
    print("Checking availability of UPX:\n", end="", flush=True)
    rc = sp.call(("upx", "-qq"))                # test presence of upx
    print("OK: UPX is available.")
    print(sep_line)
except:
    raise SystemExit("UPX not installed or missing in path definition")

try:
    bin_dir = sys.argv[1]
except:
    bin_dir = psg.PopupGetFolder("Enter folder:",
                                 "UPX Compression of binaries",
                                )

if not bin_dir:
    raise SystemExit("Cancel requested")

bin_dir = os.path.abspath(bin_dir)
print("UPX Compression of binaries in folder '%s'" % bin_dir)

tasks = []
file_count = 0
file_sizes = {}
t0 = time.time()
for root, _, files in os.walk(bin_dir):
    root = root.lower()
    for f in files:
        f = f.lower()
        fname = os.path.join(root, f)
        file_sizes[fname] = os.stat(fname).st_size
        if "qt-plugins" in root:
            continue
        if not f.endswith((".exe", ".dll", "pyd")):   # we only handle these
            continue
        if f.endswith(".dll"):
            if f.startswith("python"):
                continue
            if f.startswith("vcruntime"):
                continue
            if f.startswith("msvcp"):
                continue
            if f.startswith("cldapi"):
                continue
            if f.startswith("edp"):
                continue
            
        # make the upx invocation command
        cmd = ('upx', '-9', fname)
        t = sp.Popen(cmd, stdout=sp.PIPE, stderr=sp.PIPE, shell=False)
        tasks.append(t)
        file_count += 1

print("Started %i compressions out of %i total files ..." % (file_count, len(file_sizes.keys())), flush=True)

for t in tasks:
    t.wait()

t1 = int(round(time.time() - t0))
print("Finished in %i seconds." % t1, flush=True)
old_size = new_size = 0.0
for f in file_sizes.keys():
    old_size += file_sizes[f]
    new_size += os.stat(f).st_size
old_size *= 1./1024/1024
new_size *= 1./1024/1024
diff_size = old_size - new_size
diff_percent = diff_size / old_size * 100
text = "\nFolder Compression Results (MB)\nbefore: %.2f\nafter: %.2f\nsavings: %.2f (%.1f%%)"
print(text % (old_size, new_size, diff_size, diff_percent))