File: compile_zip.py

package info (click to toggle)
mnemosyne 2.6.1+ds-1
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 8,268 kB
  • sloc: python: 32,145; ansic: 837; xml: 625; makefile: 249
file content (33 lines) | stat: -rwxr-xr-x 883 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#
# compile_zip.py
#

import os
import sys
import shutil
import zipfile
import compileall

# Replaces *.py files in a zip archive with compiled versions.

def compile_zip(zip_file):
    # Extract to tmp dir.
    tmp_dirname = "MNEMOSYNE_ANDROID"
    if os.path.exists(tmp_dirname):
        shutil.rmtree(tmp_dirname)
    with zipfile.ZipFile(zip_file) as my_zip:
        my_zip.extractall(tmp_dirname)
    # Compile python files and remove originals.
    compileall.compile_dir(tmp_dirname, legacy=True)
    for dirname, subdirlist, filelist in os.walk(tmp_dirname):
        for f in filelist:
            if f.endswith("py"):
                os.remove(os.path.join(dirname, f))
    # Recompress.
    os.remove(zip_file)
    shutil.make_archive(zip_file.replace(".zip", ""), 'zip', tmp_dirname)
    shutil.rmtree(tmp_dirname)


if __name__ == "__main__":
    compile_zip(sys.argv[1])