scripts/jjblib.py

   1 import argparse
   2 import os
   3
   4 import yaml
   5
   6
   7 def parse_jjb_args():
   8     parser = argparse.ArgumentParser()
   9     parser.add_argument("project", help="project")
  10     parser.add_argument("-c", "--conf", help="Config file")
  11     parser.add_argument("-d", "--dependencies",
  12                         help=("Project dependencies\n\n"
  13                               "A comma-seperated (no spaces) list of projects "
  14                               "your project depends on. "
  15                               "This is used to create an integration job that "
  16                               "will trigger when a dependent project-merge "
  17                               "job is built successfully.\n\n"
  18                               "Example: aaa,controller,yangtools"))
  19     parser.add_argument("-t", "--templates", help="Job templates to use")
  20     parser.add_argument("-b", "--branches", help="Git Branches to build")
  21     parser.add_argument("-j", "--jdks", help="JDKs to build against (for verify jobs)")  # noqa
  22     parser.add_argument("-p", "--pom", help="Path to pom.xml to use in Maven "
  23                                             "build (Default: pom.xml")
  24     parser.add_argument("-g", "--mvn-goals", help="Maven Goals")
  25     parser.add_argument("-o", "--mvn-opts", help="Maven Options")
  26     parser.add_argument("-a", "--archive-artifacts",
  27                         help="Comma-seperated list of patterns of artifacts "
  28                              "to archive on build completion. "
  29                              "See: http://ant.apache.org/manual/Types/fileset.html")  # noqa
  30     parser.add_argument("-z", "--no-cfg", action="store_true",
  31                         help=("Disable initializing the project.cfg file."))
  32     return parser.parse_args()
  33
  34
  35 def create_template_config(project_dir, args):
  36     cfg_data = dict()
  37
  38     if args.templates:
  39         cfg_data["JOB_TEMPLATES"] = args.templates
  40     if args.branches:
  41         cfg_data["BRANCHES"] = args.branches
  42     if args.jdks:
  43         cfg_data["JDKS"] = args.jdks
  44     if args.pom:
  45         cfg_data["POM"] = args.pom
  46     if args.mvn_goals:
  47         cfg_data["MAVEN_GOALS"] = args.mvn_goals
  48     if args.mvn_opts:
  49         cfg_data["MAVEN_OPTS"] = args.mvn_opts
  50     if args.dependencies:
  51         cfg_data["DEPENDENCIES"] = args.dependencies
  52     if args.archive_artifacts:
  53         cfg_data["ARCHIVE"] = args.archive_artifacts
  54
  55     if cfg_data:
  56         # Create project directory if it doesn't exist
  57         if not os.path.exists(project_dir):
  58             os.makedirs(project_dir)
  59
  60         print("Creating %s.cfg file" % args.project)
  61         cfg_file = os.path.join(project_dir, "%s.cfg" % args.project)
  62
  63         with open(cfg_file, "w") as outstream:
  64             outstream.write(yaml.dump(cfg_data, default_flow_style=False))