dev.py

#!/usr/bin/python
# coding=utf-8
import sys, getopt
import os
import tarfile
import shutil
import subprocess
rootDir = sys.path[0]+os.sep
sys.path.append(rootDir + "script" + os.sep + "dev")
import  scmProjectUtil 
import scmCmdExecutor
import scmJdkInstaller
from scmCmdExecutor import ScmCmdExecutor
from scmProjectUtil import ScmProjectUtil
from scmJdkInstaller import ScmJdkInstaller

cmdExecutor = ScmCmdExecutor(False)
scmProjectUtil = ScmProjectUtil(rootDir + "script" + os.sep + "dev" + os.sep + "projectInfo.json", cmdExecutor)
jdkInstaller = ScmJdkInstaller(rootDir)

compileScmModule = None
needArchiveRelease = False

supportedCompileProfile = ["release", "original"]

def getScmVersion():
    versionFile = rootDir + "script" + os.sep + "dev" + os.sep + "version"
    with open(versionFile) as lines:
        for line in lines:
            line = line.strip()
            if line.startswith("NewVersion="):
                return line.replace("NewVersion=", "")
    print ("Error:failed to get scm version from " + versionFile)
    sys.exit(-1)

archiveReleasePath = rootDir + "archive-target" + os.sep + 'sequoiacm-' + getScmVersion() + '-release.tar.gz'
compileTestModule = None
runTestModule = None
compileRelease = False
compileProfile = "release"
skipInstallJDK = False


def display(exit_code):
    print("config file: " + rootDir + "script" + os.sep + "dev" + os.sep + "projectInfo.json")
    print("")
    print(" --help | -h       : print help message")
    print(" --compile | -c    : option value: all " + " ".join(scmProjectUtil.getScmModules()))
    print(" --compileProfile  : profile of maven compile configuration, default is release, supported: {}".format(supportedCompileProfile))
    print(" --release         : compile release sequoiacm")
    print(" --compileTest     : option value: all " + " ".join(scmProjectUtil.getTestModules()))
    print(" --runTest         : run test by testng.xml, test module: all " + " ".join(scmProjectUtil.getTestModules()))
    print(" --archive | -a    : archive scm project, package a tar.gz file")
    print(" --tar-path | -t   : specify package scm target file path, default " + archiveReleasePath)
    print(" --skipInstallJDK  : skip install JDK")
    sys.exit(exit_code)

def parse_command():
    global compileScmModule, needArchiveRelease, archiveReleasePath, compileTestModule, runTestModule, compileRelease, compileProfile, skipInstallJDK
    try:
        options, args = getopt.getopt(sys.argv[1:], "hc:at:",
                                      ["help", "compile=", "compileProfile=", "release", "archive", "tar-path=",
                                       "compileTest=", "runTest=", "skipInstallJDK"])
    except getopt.GetoptError, e:
        print ("Error:", e)
        sys.exit(-1)
        
    for name, value in options:
        if name in ("-h", "--help"):
            display(0)
        elif name in ("-c","--compile"):
            compileScmModule = value
        elif name in ("--compileProfile"):
            if value not in supportedCompileProfile:
                print('compileProfile {} is not supported, supported: {}\n'
                      .format(value, supportedCompileProfile))
                display(1)
            compileProfile = value
        elif name in ("-a", "--archive"):
            needArchiveRelease = True;
        elif name in ("-t", "--tar-path"):
            archiveReleasePath = value
        elif name in ("--compileTest"):
            compileTestModule = value
        elif name in ("--runTest"):
            runTestModule = value
        elif name in ("--release"):
            compileRelease = True
        elif name in ("--skipInstallJDK"):
            skipInstallJDK = True


def compileScm():
    if not skipInstallJDK:
        jdkInstaller.check_and_install_jdk()
    compileArgs = ""
    if compileRelease:
        compileArgs = "-DpackageType=release"

    if compileProfile:
        compileArgs = compileArgs + " -P{} ".format(compileProfile)

    projectRootDir = os.path.join(rootDir, 'src')
    vulnerabilityFixerRootDir = os.path.join(rootDir, 'tools', 'scm-vulnerability-fixer')

    if compileScmModule == "all":
        scmProjectUtil.compileDoc(rootDir + "doc" + os.sep)
        
        # cleanup local maven repository before compile
        scmProjectUtil.cleanMavenLocalRepo()
        scmProjectUtil.compileScm(compileArgs)
    else:
        scmProjectUtil.compileScmModule(compileScmModule, compileArgs)

    if compileProfile == "release":
        scmProjectUtil.verifyScm(projectRootDir,
                                 os.path.join(vulnerabilityFixerRootDir, 'src', 'main', 'resources', 'artifacts.txt'),
                                 os.path.join(vulnerabilityFixerRootDir, 'tmp'))
        print("Success: no vulnerability jars have been detected in product archive.")


def compileTest():
    if compileTestModule == "all":
        scmProjectUtil.compileTest()
    else:
        scmProjectUtil.compileTestModule(compileTestModule)

def archive():
    scmProjectUtil.packageScm(archiveReleasePath)

def untar(filePath,dir):
    try:
        t = tarfile.open(filePath)
        t.extractall(dir)
    except Exception as e:
        print(e)

def runTest():
    if runTestModule == "all":
        scmProjectUtil.runAllTest()
    else:
        scmProjectUtil.runTest(runTestModule)

if __name__ == '__main__':
    parse_command()

    doSomething = False
    if compileScmModule != None:
        compileScm()
        doSomething = True

    if compileTestModule != None:
        compileTest()
        doSomething = True
        
    if runTestModule != None:
        runTest()
        doSomething = True
        
    if needArchiveRelease:
        archive()
        doSomething = True
    
    if doSomething == False:
        display(0)