|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
#!/usr/bin/env python3
#-*- coding: utf-8 -*-
#pylint: disable-msg=
"""
File : Skeleton.py
Author : Valentin Kuznetsov <vkuznet@gmail.com>
Description:
"""
# system modules
import os
import sys
import pprint
from optparse import OptionParser
# package modules
from FWCore.Skeletons.utils import code_generator, test_env, template_directory
if sys.version_info < (2, 6):
raise Exception("This script requires python 2.6 or greater")
class SkeletonOptionParser:
"Skeleton option parser"
def __init__(self):
cname = os.environ.get('MKTMPL_CMD', 'main.py')
usage = "Usage: %s [options]\n" % cname
self.parser = OptionParser(usage=usage)
msg = "debug output"
self.parser.add_option("--debug", action="store_true",
default=False, dest="debug", help=msg)
msg = "specify template, e.g. EDProducer"
self.parser.add_option("--tmpl", action="store", type="string",
default='', dest="tmpl", help=msg)
msg = "specify package name, e.g. MyProducer"
self.parser.add_option("--name", action="store", type="string",
default="TestPkg", dest="pname", help=msg)
msg = "specify author name"
self.parser.add_option("--author", action="store", type="string",
default="", dest="author", help=msg)
msg = "specify file type to generate, "
msg += "e.g. --ftype=header, default is all files"
self.parser.add_option("--ftype", action="store", type="string",
default="all", dest="ftype", help=msg)
msg = "list examples tags which should be kept in "
msg += "generate code, e.g. "
msg += "--keep-etags='@example_trac,@example_hist'"
self.parser.add_option("--keep-etags", action="store", type="string",
default=None, dest="ketags", help=msg)
msg = "list template tags"
self.parser.add_option("--tags", action="store_true",
default=False, dest="tags", help=msg)
msg = "list template example tags"
self.parser.add_option("--etags", action="store_true",
default=False, dest="etags", help=msg)
msg = "list supported templates"
self.parser.add_option("--templates", action="store_true",
default=False, dest="templates", help=msg)
def get_opt(self):
"Returns parse list of options"
return self.parser.parse_args()
def parse_args(args):
"Parse input arguments"
output = {}
for arg in args:
key, val = arg.split('=')
key = key.strip()
val = val.strip()
if val[0] == '[' and val[-1] == ']':
val = eval(val, { "__builtins__": None }, {})
output[key] = val
return output
def generator():
"""
Code generator function, parse user arguments and load appropriate
template module. Once loaded, run its data method depending on
user requested input parameter, e.g. print_etags, print_tags or
generate template code.
"""
optmgr = SkeletonOptionParser()
opts, args = optmgr.get_opt()
test_env(os.path.join(opts.tdir, opts.tmpl), opts.tmpl)
config = {'pname': opts.pname, 'tmpl': opts.tmpl, 'author': opts.author,
'args': parse_args(args), 'debug': opts.debug,
'ftype': opts.ftype}
if opts.ketags:
etags = opts.ketags.split(',')
config.update({'tmpl_etags': etags})
else:
config.update({'tmpl_etags': []})
obj = code_generator(config)
if opts.etags:
obj.print_etags()
sys.exit(0)
elif opts.tags:
obj.print_tags()
sys.exit(0)
elif opts.templates:
for name in os.listdir(template_directory()):
print(name)
sys.exit(0)
obj.generate()
if __name__ == '__main__':
generator()
|