Python Badger.reCalcCount示例

示例#1

__RCSID__ = "$Id$"
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 IHEPDIRAC.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)

示例#2

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 IHEPDIRAC.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)