ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/COMP/CRAB/python/DLSInfo.py
(Generate patch)

Comparing COMP/CRAB/python/DLSInfo.py (file contents):
Revision 1.22 by spiga, Mon Sep 8 09:47:47 2008 UTC vs.
Revision 1.24 by spiga, Tue Nov 18 11:06:20 2008 UTC

# Line 37 | Line 37 | class DLSNoReplicas(exceptions.Exception
37   class DLSInfo:
38      def __init__(self, type, cfg_params):
39          self.cfg_params = cfg_params
40 +        self.showCAF = False
41          if type=="DLS_TYPE_DBS":
42              # use dbs_url as dls_endpoint if dls_type is dbs
43              try:
# Line 48 | Line 49 | class DLSInfo:
49                  endpoint=self.cfg_params['CMSSW.dls_phedex_url']
50              except KeyError:
51                  endpoint='http://cmsweb.cern.ch/phedex/datasvc/xml/prod/'
52 +            if self.cfg_params['CRAB.scheduler'].upper() == 'CAF':  self.showCAF = True
53          else:
54              msg = "DLS type %s not among the supported DLS ( DLS_TYPE_DLI and DLS_TYPE_MYSQL ) "%type
55              raise CrabException(msg)
54 –
56          common.logger.debug(5,"DLS interface: %s Server %s"%(type,endpoint))      
57          try:
58              self.api = dlsClient.getDlsApi(dls_type=type,dls_endpoint=endpoint)
# Line 60 | Line 61 | class DLSInfo:
61              #print msg
62              raise CrabException(msg)
63  
64 +
65 +    def getReplicasBulk(self,fileblocks):
66 +        """
67 +        query DLS to get replicas
68 +        """
69 +        ##
70 +        try:
71 +            entryList=self.api.getLocations(fileblocks,longList=True)
72 +        except dlsApi.DlsApiError, inst:
73 +            raise DLSNoReplicas(fileblocks)
74 +        results = {}
75 +        for entry in entryList:
76 +            ListSites=[]
77 +            for loc in entry.locations:
78 +                ListSites.append(str(loc.host))
79 +            if len(ListSites)<=0:
80 +                raise DLSNoReplicas(fileblocks)
81 +            results[entry.fileBlock.name]=ListSites
82 +
83 +        return results        
84   # ####################################
85      def getReplicas(self,fileblocks):
86          """
# Line 67 | Line 88 | class DLSInfo:
88          """
89          ##
90          try:
91 <            entryList=self.api.getLocations([fileblocks])
91 >            entryList=self.api.getLocations([fileblocks],showCAF=self.showCAF)
92          except dlsApi.DlsApiError, inst:
72 –            #msg = "Error in the DLS query: %s." % str(inst)
73 –            #print msg
93              raise DLSNoReplicas(fileblocks)
94  
95          ListSites=[]

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines