make_release.py

#!/usr/bin/env python
# code: utf-8

'''
A wrapper script to generate zip files for GitHub releases.

This script tends to be compatible with both Python 2 and Python 3.
'''

from __future__ import print_function

import os
import shutil


SRC_DIR = 'src'
CONTRIB_BASE = 'contrib'
BUILD_BASE = 'build'
DIST_BASE = 'dist'


def make_calibre_plugin():

    contrib_dir = os.path.join(CONTRIB_BASE, 'calibre')

    build_dir = os.path.join(BUILD_BASE, 'DeDRM_calibre_plugin')
    core_dir = os.path.join(build_dir, 'DeDRM_plugin')
    plugin_name = os.path.join(build_dir, 'DeDRM_plugin')

    shutil.copytree(contrib_dir, build_dir)
    shutil.copytree(SRC_DIR, core_dir)

    shutil.make_archive(plugin_name, 'zip', core_dir)
    shutil.rmtree(core_dir)


def make_windows_app():

    contrib_dir = os.path.join(CONTRIB_BASE, 'windows')

    build_dir = os.path.join(BUILD_BASE, 'DeDRM_Windows_Application')
    core_dir = os.path.join(build_dir, 'DeDRM_App', 'DeDRM_lib', 'lib')

    shutil.copytree(contrib_dir, build_dir)
    shutil.copytree(SRC_DIR, core_dir)


def make_macos_app():

    contrib_dir = os.path.join(CONTRIB_BASE, 'macos')
    build_dir = os.path.join(BUILD_BASE, 'DeDRM_Macintosh_Application')
    core_dir = os.path.join(build_dir, 'DeDRM.app', 'Contents', 'Resources')

    shutil.copytree(contrib_dir, build_dir)

    _, dirs, files = next(os.walk(SRC_DIR))
    for name in dirs:
        shutil.copyfile(
            os.path.join(SRC_DIR, name),
            os.path.join(core_dir, name)
        )
    for name in files:
        shutil.copy2(
            os.path.join(SRC_DIR, name),
            os.path.join(core_dir, name)
        )


def make_obok_plugin():

    build_dir = os.path.join(BUILD_BASE, 'Obok_calibre_plugin')
    core_dir = os.path.join(build_dir, 'obok_plugin')
    plugin_name = os.path.join(build_dir, 'obok_plugin')

    shutil.copytree('Obok_calibre_plugin', build_dir)
    shutil.make_archive(plugin_name, 'zip', core_dir)


def make_misc():

    for name in ('Other_Tools',):
        shutil.copytree(name, os.path.join(BUILD_BASE, name))
    for name in ('FAQs.md', 'README.md', 'ReadMe_First.txt',):
        shutil.copy2(name, os.path.join(BUILD_BASE, name))


def make_release(version):

    try:
        shutil.rmtree(BUILD_BASE)
    except OSError:
        pass

    make_calibre_plugin()
    make_windows_app()
    make_macos_app()
    make_obok_plugin()
    make_misc()

    release_name = 'DeDRM_tools_{}'.format(version)
    release_path = os.path.join(DIST_BASE, release_name)
    return shutil.make_archive(release_path, 'zip', BUILD_BASE)


if __name__ == '__main__':
    import sys
    try:
        version = sys.argv[1]
    except IndexError:
        raise SystemExit('Usage: {} version'.format(__file__))

    print(make_release(version))