Makefile.py

# -*- coding: utf-8 -*-
from __future__ import print_function

import os
import sys
import subprocess
import shutil
import pprint
import zipfile

skript_dir = os.path.dirname(sys.argv[0])

args = sys.argv[1:]
plugin_name = args[0]
plugin_dir = args[1]
targets = args[2:]

print("targets: ", targets)

"""
Configuration
"""
BUILD_UI_FOR_TATGET_BUILD = False
IGNORE_PATTERNS_FOR_TARGET_INSTALL = (
    "*.pyc",
    "*.ts",
    "*.qrc",
    ".git",
    #'*.ui',
)

qgis_root = os.getenv("QGIS_BASE_DIR", None)
if qgis_root is None:
    sys.exit("QGIS_BASE_DIR env var not found!")
QGIS_PLUGINS_DIST_DIR = os.path.join(
    qgis_root, "apps", "qgis", "python", "plugins"
)

TS_FILE_NAME = "%s_ru.ts" % plugin_name
TS_FILE_NAME = "QuickMapServices_ru.ts"

"""
Work section:
"""


def process_ui():
    print("process ui ...")

    ui_dir = os.path.join(plugin_dir, "src", "ui")
    if not os.path.exists(ui_dir):
        print("there is no ui dir? ui files not found")
        return

    ui_files = [
        f for f in os.listdir(ui_dir) if os.path.splitext(f)[1] == ".ui"
    ]

    for ui_file in ui_files:
        print("Process %s ..." % ui_file)
        subprocess.call(
            [
                "pyuic4.bat",
                "-o",
                os.path.join(
                    plugin_dir,
                    "src",
                    "ui_%s.py" % os.path.splitext(ui_file)[0],
                ),
                os.path.join(ui_dir, ui_file),
            ]
        )


def process_qrc():
    print("process qrc ...")
    qrc_files = [
        f for f in os.listdir(plugin_dir) if os.path.splitext(f)[1] == ".qrc"
    ]

    for qrc_file in qrc_files:
        print("Process %s ..." % qrc_file)
        subprocess.call(
            [
                "pyrcc4",
                "-o",
                os.path.join(
                    plugin_dir, "%s_rc.py" % os.path.splitext(qrc_file)[0]
                ),
                os.path.join(plugin_dir, qrc_file),
            ]
        )


def process_ts():
    print("process ts ...")
    i18n_dir = os.path.join(plugin_dir, "i18n")
    i18n_files = [
        f for f in os.listdir(i18n_dir) if os.path.splitext(f)[1] == ".ts"
    ]

    for i18n_file in i18n_files:
        print("Process %s ..." % os.path.join(i18n_dir, i18n_file))
        subprocess.call(["lrelease", os.path.join(i18n_dir, i18n_file)])


def install_plugin():
    print("install plugin ...")
    plugin_install_dir = os.path.join(QGIS_PLUGINS_DIST_DIR, plugin_name)
    if os.path.exists(plugin_install_dir):
        shutil.rmtree(plugin_install_dir)

    print("plugin_dir: ", type(plugin_dir))
    print("plugin_install_dir: ", type(plugin_install_dir))

    shutil.copytree(
        plugin_dir,
        plugin_install_dir,
        ignore=shutil.ignore_patterns(*IGNORE_PATTERNS_FOR_TARGET_INSTALL),
    )


def make_zip():
    print("make zip ...")
    zipFile = zipfile.ZipFile(
        os.path.join(skript_dir, plugin_name.lower() + ".zip"), "w"
    )

    src_dir = os.path.join(plugin_dir)

    for root, dirs, files in os.walk(src_dir):
        if root.find(".git") != -1:
            continue

        for f in files:
            if f.endswith(
                (
                    ".gitignore",
                    ".pyc",
                    ".ts",
                    ".qrc",
                    ".sublime-project",
                    ".sublime-workspace",
                    "Makefile.py",
                    ".zip",
                    "bat",
                    "pro",
                )
            ):
                continue
            filename = os.path.join(root, f)
            zipFile.write(
                filename,
                os.path.join(
                    plugin_name.lower(), os.path.relpath(filename, src_dir)
                ),
            )


def make_ts_file():
    print("make ts file ...")
    target_files = []
    for root, dirs, files in os.walk(plugin_dir):
        target_files.extend(
            [
                os.path.join(root, f)
                for f in files
                if os.path.splitext(f)[1] in [".ui", ".py"]
            ]
        )

    cmd = ["pylupdate4"]
    cmd.extend(target_files)
    cmd.append("-ts")
    cmd.append(os.path.join(plugin_dir, "i18n", TS_FILE_NAME))

    pprint.pprint(cmd)
    subprocess.call(cmd)


if "rc" in targets:
    process_qrc()

if "ui" in targets:
    process_ui()

if "build" in targets:
    process_qrc()
    if BUILD_UI_FOR_TATGET_BUILD:
        process_ui()

if "install" in targets:
    install_plugin()

if "ts" in targets:
    make_ts_file()

if "lang" in targets:
    process_ts()

if "pack" in targets:
    make_zip()