|
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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
#!/usr/bin/env python3
#-*- coding: utf-8 -*-
#pylint: disable-msg=
"""
File : cms.py
Author : Valentin Kuznetsov <vkuznet@gmail.com>
Description: CMS-related utils
"""
# system modules
from builtins import range
import os
import sys
# package modules
from FWCore.Skeletons.utils import code_generator
def config(tmpl, pkg_help):
"Parse input arguments to mk-script"
kwds = {'author': '', 'tmpl': tmpl,
'args': {}, 'debug': False,
'working_dir': ''}
etags = []
if len(sys.argv) >= 2: # user give us arguments
if sys.argv[1] in ['-h', '--help', '-help']:
print(pkg_help)
sys.exit(0)
kwds['pname'] = sys.argv[1]
for idx in range(2, len(sys.argv)):
opt = sys.argv[idx]
if opt == '-author':
kwds['author'] = sys.argv[idx+1]
continue
if opt.find('example') != -1:
etags.append('@%s' % opt)
continue
if opt in ['-h', '--help', '-help']:
print(pkg_help)
sys.exit(0)
if opt == '-debug':
kwds['debug'] = True
continue
elif len(sys.argv) == 1:
# need to walk
msg = 'Please enter %s name: ' % tmpl.lower()
kwds['pname'] = input(msg)
else:
print(pkg_help)
sys.exit(0)
kwds['tmpl_etags'] = etags
return kwds
def config_with_parser(tmpl, args):
"""
Inject arguments parsed upstream into mk-scripts.
The arguments are parsed by the different front-ends(binaries)
and passed here via the args object.
"""
kwds = {'author': '', 'tmpl': tmpl,
'args': {}, 'debug': False}
etags = []
kwds['pname'] = args.subpackage_name
if args.author: kwds['author'] = args.author
if args.debug: kwds['debug'] = True
if args.example: etags.append('@%s' % args.example)
kwds['tmpl_etags'] = etags
return kwds
def cms_error():
"Standard CMS error message"
msg = "\nPackages must be created in a 'subsystem'."
msg += "\nPlease set your CMSSW environment and go to $CMSSW_BASE/src"
msg += "\nCreate or choose directory from there and then "
msg += "\nrun the script from that directory"
return msg
def test_cms_environment(tmpl):
"""
Test CMS environment and requirements to run within CMSSW_BASE.
Return True if we fullfill requirements and False otherwise.
"""
base = os.environ.get('CMSSW_BASE', None)
if not base:
return False, []
cdir = os.getcwd()
ldir = cdir.replace(os.path.join(base, 'src'), '')
dirs = ldir.split('/')
# test if we're within CMSSW_BASE/src/SubSystem area
if ldir and ldir[0] == '/' and len(dirs) == 2:
return 'subsystem', ldir
# test if we're within CMSSW_BASE/src/SubSystem/Pkg area
if ldir and ldir[0] == '/' and len(dirs) == 3:
return 'package', ldir
# test if we're within CMSSW_BASE/src/SubSystem/Pkg/src area
# if ldir and ldir[0] == '/' and len(dirs) == 4 and dirs[-1] == 'src':
# return 'src', ldir
# test if we're within CMSSW_BASE/src/SubSystem/Pkg/plugin area
# if ldir and ldir[0] == '/' and len(dirs) == 4 and dirs[-1] == 'plugins':
# return 'plugins', ldir
# test if we're within CMSSW_BASE/src/SubSystem/Pkg/dir area
if ldir and ldir[0] == '/' and len(dirs) == 4:
return dirs[-1], ldir
return False, ldir
def generate(kwds):
"Run generator code based on provided set of arguments"
config = dict(kwds)
tmpl = kwds.get('tmpl')
stand_alone_group = ['Record', 'Skeleton']
config.update({'not_in_dir': stand_alone_group})
if tmpl in stand_alone_group:
whereami, ldir = test_cms_environment(tmpl)
dirs = ldir.split('/')
config.update({'pkgname': kwds.get('pname')})
config.update({'subsystem': 'Subsystem'})
config.update({'pkgname': 'Package'})
if whereami:
if len(dirs) >= 3:
config.update({'subsystem': dirs[1]})
config.update({'pkgname': dirs[2]})
elif len(dirs) >= 2:
config.update({'subsystem': dirs[1]})
config.update({'pkgname': dirs[1]})
else:
whereami, ldir = test_cms_environment(tmpl)
dirs = ldir.split('/')
if not dirs or not whereami:
print(cms_error())
sys.exit(1)
config.update({'subsystem': dirs[1]})
config.update({'pkgname': kwds.get('pname')})
if whereami in ['src', 'plugins']:
config.update({'working_dir': whereami})
config.update({'tmpl_files': '.cc'})
config.update({'pkgname': dirs[2]})
elif whereami == 'test':
config.update({'working_dir': whereami})
config.update({'tmpl_files':'.cc'})
config.update({'pkgname': dirs[2]})
elif whereami == 'subsystem':
config.update({'tmpl_files': 'all'})
else:
print(cms_error())
sys.exit(1)
obj = code_generator(config)
obj.generate()
|