CMS 3D CMS Logo

Functions
relval_machine Namespace Reference

Functions

def convert_keys_to_string (dictionary)
 
def fix_run (run)
 
def get_json_files ()
 
def load_steps_and_workflows ()
 

Function Documentation

◆ convert_keys_to_string()

def relval_machine.convert_keys_to_string (   dictionary)
Recursively converts dictionary keys to strings.
    Utility to help deal with unicode keys in dictionaries created from json requests.
    In order to pass dict to function as **kwarg we should transform key/value to str.

Definition at line 35 of file relval_machine.py.

References genParticles_cff.map, and str.

Referenced by load_steps_and_workflows().

35 def convert_keys_to_string(dictionary):
36  """ Recursively converts dictionary keys to strings.
37  Utility to help deal with unicode keys in dictionaries created from json requests.
38  In order to pass dict to function as **kwarg we should transform key/value to str.
39  """
40  if isinstance(dictionary, str):
41  return str(dictionary)
42  elif isinstance(dictionary, collections.Mapping):
43  return dict(map(convert_keys_to_string, dictionary.items()))
44  elif isinstance(dictionary, collections.Iterable):
45  return type(dictionary)(map(convert_keys_to_string, dictionary))
46  else:
47  return dictionary
48 
def convert_keys_to_string(dictionary)
#define str(s)

◆ fix_run()

def relval_machine.fix_run (   run)

Definition at line 25 of file relval_machine.py.

References createfilelist.int, print(), python.rootplot.root2matplotlib.replace(), and submitPVValidationJobs.split().

Referenced by load_steps_and_workflows().

25 def fix_run(run):
26  runs = run.replace(" ", "").replace("[", "").replace("]", "").split(",")
27  int_runs = []
28  for item in runs:
29  if item.isdigit():
30  int_runs.append(int(item))
31  else:
32  print("WARNING: run is in bad format: {0}".format(run))
33  return int_runs
34 
def fix_run(run)
def replace(string, replacements)
void print(TMatrixD &m, const char *label=nullptr, bool mathematicaFormat=false)
Definition: Utilities.cc:47

◆ get_json_files()

def relval_machine.get_json_files ( )

Definition at line 12 of file relval_machine.py.

Referenced by load_steps_and_workflows().

12 def get_json_files():
13  cwd = os.path.join(os.getcwd(), "json_data")
14  if not os.path.exists(cwd):
15  return []
16 
17  json_files = []
18  for f in os.listdir(cwd):
19  full_path = os.path.join(cwd, f)
20  if os.path.isfile(full_path) and f.endswith(".json"):
21  json_files.append(full_path)
22  return json_files
23 
24 

◆ load_steps_and_workflows()

def relval_machine.load_steps_and_workflows ( )

Definition at line 49 of file relval_machine.py.

References convert_keys_to_string(), fix_run(), get_json_files(), and mps_monitormerge.items.

50  data_files = get_json_files()
51  for index, data_file in enumerate(data_files):
52  with open(data_file, "r") as f: data = json.load(f)
53  data = convert_keys_to_string(data)
54  label = data["label"]
55  steps_names = []
56  for step_name, step in data["steps"].items():
57  steps_names.append((step_name, step["sequence_number"]))
58  if step_name in steps:
59  continue # this step was inserted already
60 
61  # inputInfo case
62  if "inputInfo" in step:
63  input_info = step["inputInfo"]
64  if "run" in input_info:
65  input_info["run"] = fix_run(input_info["run"])
66 
67  steps[step_name] = {
68  'INPUT': InputInfo(**input_info)
69  }
70  # step with parameters
71  elif "parameters" in step:
72  steps[step_name] = step["parameters"]
73  else:
74  raise Exception("Wrong step format in {0} file".format(data_file))
75 
76  sorted_steps = sorted(steps_names, key=lambda step: step[1]) # sort steps by sequence number
77  sorted_steps_names = [step_name[0] for step_name in sorted_steps] # filter only step names
78 
79  workflows[1000000.0 + 0.1*index] = [label, sorted_steps_names]
80 
81 
83 
84 
def fix_run(run)
def load_steps_and_workflows()
def convert_keys_to_string(dictionary)