inji.cli.main()

Complexity

Conditions

11

Size

Total Lines	72
Code Lines	38

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	36
CRAP Score	11

Importance

Changes

0

#!/usr/bin/env python3
def main():
  """ Our main method """

  # cleanly handle ctrl-c's
  signal.signal(signal.SIGINT, sigint_handler)

  assert sys.version_info >= (3,5), 'Python version ({}.{} !>= 3.5)'.format(
    sys.version_info.major,
    sys.version_info.minor
  )

  args = cli_args()

  # this holds all the possible vars files we are told about or imply
  vars_files = []

  # context in the local configuration files - p5
  vars_files += fnmatch.filter(os.listdir('.'), "*inji.y*ml")

  # context in the overlay directories - p4
  for d in args.overlay_dir:
    files = list(utils.recursive_iglob(d, '*.y*ml'))
    if len(files):
      loc = locale.getlocale()
      locale.setlocale(locale.LC_ALL, 'C') # Using LC_ALL=C POSIX locale
      files.sort()                         # We force the sort collation of files
      locale.setlocale(locale.LC_ALL, loc) # And then reset it back
      vars_files += files

  # context from named vars files - p3
  vars_files += args.vars_file

  # This will hold the final vars dict merged from various available sources
  context = {}
  for file in vars_files:
    context.update(utils.read_context(file))

  # context from environment variables - p2
  context.update(os.environ)

  # context at the command line (either JSON or KV type) - p1
  if args.json_string:
    context.update(args.json_string)

  if args.kv_pair:
    # we've appended dicts into args.kv_pair .. unpack them in order
    for d in args.kv_pair:
      context.update(d)

  if '-' in args.template:
    # Template passed in via stdin. Create template as a tempfile and use it
    # instead but since includes are possible (though not likely), we have to do
    # this in an isolated tmpdir container to prevent inadvertent reading of
    # includes not meant to be read.
    tmpdir = tempfile.mkdtemp(prefix=__name__)
    atexit.register(shutil.rmtree, tmpdir)

    _, tmpfile = tempfile.mkstemp(prefix='stdin-', dir=tmpdir, text=True)
    atexit.register(os.remove, tmpfile)

    with open(tmpfile, "a+") as f:
      f.write(sys.stdin.read())

    # Yes, even if user specifies multiple other templates, the fact he
    # specified '-' just once means we only deal with one template i.e. '-'
    args.template = [tmpfile]

  engine = TemplateEngine( undefined_variables_mode_behaviour=args.undefined_variables_mode )

  for template in args.template:
    for block in engine.render( template=template, context=context ):
      print(block)