run.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import os, shutil, logging, getopt, sys
from assemble import assemble
from diffuse import diff_apk
from files import *
from apktool import ApkInfo
from logger import config_logging
from git_tag import *
from global_config import *
from strengthen import jiagu_360
from mailing import send_email
from resources import *

command_info = "\
Options: \n\
    -h[--help]                 Help info\n\
    -c[--script]               Target build script path\n\
    -v[--version]              Build APK version\n\
    -c[--channels]             Build APK channels, split by ',' for multiple channels, for example 'oversea,national'"

def _build_apk(bit: BitConfiguration, flavor: FlavorConfiguration) -> ApkInfo:
    '''Execute APK build flow.'''
    print(">>>> Beginning to build APK for flavor [%s] bit [%s]" % (flavor.get_name(), bit.get_name()))
    info = assemble(bit, flavor)
    diff = diff_apk(info)
    if config.strengthen_enable:
        jiagu_file_directory = os.path.join(info.output_apk_directory, 'strengthen')
        jiagu_360(info.output_apk_file_path, jiagu_file_directory)
    mail_subject = "%s(%s,%s)" % (config.output_mail_title, flavor.get_name(), bit.get_name())
    send_email(config.output_mail_receivers, mail_subject, diff, 'html')
    return info

def _show_invalid_command(info: str):
    '''Show command invalid info.'''
    print('Error: Unrecognized command: %s' % info)
    print(command_info)     

def _parse_command(argv):
    '''Parse command.'''
    try:
        opts, args = getopt.getopt(argv, "-h:-s:-v:-c:", ["help", "script=", 'version=', 'channels='])
    except BaseException as e:
        _show_invalid_command(str(e))
        sys.exit(2)
    for opt, arg in opts:
        print(arg)
        if opt in ('-s', '--script'):
            build_config.target_script = arg 
        elif opt in ("-v", "--version"):
            build_config.version = arg
        elif opt in ("-c", "--channels"):
            build_config.channels = arg
        elif opt in ('-h', '--help'):
            print(command_info)
    logi("Build Info: target script[%s], version[%s] and channels[%s]"
         % (build_config.target_script, build_config.version, build_config.channels))

def _run_main():
    '''Run main program.'''
    if len(build_config.channels) == 0:
        # _build_apk(BitConfiguration.BIT_32, FlavorConfiguration.NATIONAL)
        info = _build_apk(BitConfiguration.BIT_64, FlavorConfiguration.NATIONAL)
        info = _build_apk(BitConfiguration.BIT_64, FlavorConfiguration.OVERSEA)
    else:
        has_national = build_config.channels.find(FlavorConfiguration.NATIONAL.get_name()) >= 0
        has_oversea = build_config.channels.find(FlavorConfiguration.OVERSEA.get_name()) >= 0
        if has_national:
            _build_apk(BitConfiguration.BIT_32, FlavorConfiguration.NATIONAL)
            info = _build_apk(BitConfiguration.BIT_64, FlavorConfiguration.NATIONAL)
        if has_oversea:
            _build_apk(BitConfiguration.BIT_32, FlavorConfiguration.OVERSEA)
            info = _build_apk(BitConfiguration.BIT_64, FlavorConfiguration.OVERSEA)
        if not has_national and not has_oversea:
            print(">>>>> failed! due to no channels match!")
            return
    copy_language_resources(info.version_name)
    gen_git_log(info)
    add_new_tag(info)

if __name__ == "__main__":
    ''' python3 run.py -s config/config_product.yml -v 4.0.3 '''
    config_logging()
    _parse_command(sys.argv[1:])
    config.parse()
    _run_main()