Python Badger.reCalcCount Exemples

Exemple #1

Fichier : besdirac-dms-dataset-create.py Projet : besdiracgrid/BESDIRAC

from DIRAC.Core.Base import Script

Script.setUsageMessage(__doc__)

args = Script.getPositionalArgs()
# print len(args)
if len(args) < 3:
    Script.showHelp()
    exit(-1)

datasetName = args[0]
path = args[1]
strArg = args[2]

from BESDIRAC.Badger.API.Badger import Badger

badger = Badger()
prefix = badger.getDatasetNamePrefix()
datasetName = prefix + datasetName

# print datasetName
result = badger.registerDataset(datasetName, path, strArg)
if result["OK"]:
    resVal = badger.getFilesByDatasetName(datasetName)
    if resVal["OK"]:
        fileList = resVal["Value"]
        badger.reCalcCount(fileList)
# print result

exit(0)

Exemple #2

Fichier : besdirac-dms-dataset-delete.py Projet : besdiracgrid/BESDIRAC

from DIRAC import S_OK, S_ERROR, gLogger, exit
from DIRAC.Core.Base import Script

Script.setUsageMessage(__doc__)
args = Script.getPositionalArgs()

if len(args)!=1:
  Script.showHelp()
datasetName = args[0]

from BESDIRAC.Badger.API.Badger import Badger
badger = Badger()
from DIRAC.DataManagementSystem.Client.ReplicaManager import ReplicaManager
rm = ReplicaManager()

result = badger.getFilesByDatasetName(datasetName)

if result['OK']:
  fileList = result['Value'] 
  fileCountDict = badger.reCalcCount(fileList,False)
  badger.removeDataset(datasetName)

  for file in fileCountDict:
    if fileCountDict[file] == 0:
      result = rm.removeFile(file)
      if not result['OK']:
        print "Failed remove file %s"%file


exit(0)