readDatabase Namespace Reference

Functions
def	getWeights (dsid)
def	getOrderedWeights (dsid)
def	main (argv)

Variables
string	SystToolsPath = "/cvmfs/atlas.cern.ch/repo/sw/database/GroupData/dev/PMGTools/SystematicsDatabase/"

Function Documentation

getOrderedWeights()

def readDatabase.getOrderedWeights

(

dsid

)

Definition at line 45 of file readDatabase.py.

def getOrderedWeights(dsid):
  res = {}
  for fn in os.listdir("%s/Weight_files" % SystToolsPath):
    print(fn, str(dsid), (str(dsid) in fn))
    if str(dsid) in fn:
      with open("%s/Weight_files/%s" % (SystToolsPath, fn)) as f:
        counter = 0
        for line in f.readlines():
          line = line.strip()
          res[line] = counter
          counter += 1
      break
  if len(res) == 0:
    print((str(dsid) + ' not in Database'))
  return res
 
 

getWeights()

def readDatabase.getWeights

(

dsid

)

Definition at line 22 of file readDatabase.py.

def getWeights(dsid):
  info = {}
  keys = []
  with open("%s/DSID_Database.yaml" % SystToolsPath) as f:
      database = yaml.load(f)
      if any(str(dsid) == k for k in database.keys()):
          keys = database.get(str(dsid))
          # print(keys)
          if keys == ['nominal']:
              info.update({"nominal": "nominal"})
          else:
              with open('%s/Weight_Database.yaml' % SystToolsPath) as d:
                  weight_data = yaml.load(d)
                  for key in keys:
                      if key in weight_data.keys():
                          info.update({key: weight_data.get(key)})
                      else:
                          print(key + ' key for ' + str(dsid) + ' not in Weight Databse')
      else:
          print((str(dsid) + ' not in Database'))
  return info, keys
 
 

main()

def readDatabase.main

(

argv

)

Definition at line 62 of file readDatabase.py.

def main(argv):
  parser = argparse.ArgumentParser(description='Variation string needed')
  parser.add_argument('-d', '--dsid', default=410425, type=int, help='six digit DSID')
  args = parser.parse_args()
  dictionary, keys = getWeights(args.dsid)
  print(keys)
  print(dictionary)
 
 

Variable Documentation

SystToolsPath

string readDatabase.SystToolsPath = "/cvmfs/atlas.cern.ch/repo/sw/database/GroupData/dev/PMGTools/SystematicsDatabase/"

Definition at line 11 of file readDatabase.py.