Skip to content

feat (TS): cast getTransformationFilesCount at the client level #8165

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
chrisburr merged 1 commit into from
Apr 22, 2025
Merged

feat (TS): cast getTransformationFilesCount at the client level #8165

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 18 additions & 1 deletion src/DIRAC/TransformationSystem/Client/TransformationClient.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,8 @@
from DIRAC import S_OK, S_ERROR, gLogger
from DIRAC.Core.Base.Client import Client, createClient
from DIRAC.Core.Utilities.List import breakListIntoChunks
from DIRAC.Core.Utilities.JEncode import decode as jdecode
from DIRAC.Core.Utilities.ReturnValues import convertToReturnValue, returnValueOrRaise
from DIRAC.Core.Utilities.JEncode import decode as jdecode, strToIntDict
from DIRAC.ConfigurationSystem.Client.Helpers.Operations import Operations
from DIRAC.TransformationSystem.Client import TransformationStatus
from DIRAC.TransformationSystem.Client import TransformationFilesStatus
Expand Down Expand Up @@ -591,3 +592,19 @@ def isOK(self):
def addDirectory(self, path, force=False):
rpcClient = self._getRPC()
return rpcClient.addDirectory(path, force)

@convertToReturnValue
def getTransformationFilesCount(self, transName, field, selection=None, **kwargs):
if not selection:
selection = {}
rpcClient = self._getRPC(**kwargs)

files_count = returnValueOrRaise(rpcClient.getTransformationFilesCount(transName, field, selection))
# Some of these fields are LHCb specific, however:
# * it seems only LHCb uses this method
# * it will not be necessary with diracx & pydantic anymore
if field in ["TransformationID", "FileID", "TaskID", "ErrorCount", "RunNumber", "Size"]:
total = files_count.pop("Total")
files_count = strToIntDict(files_count)
files_count["Total"] = total
return files_count
Loading