| 5 |
|
from crab_util import * |
| 6 |
|
import common |
| 7 |
|
|
| 8 |
< |
try: |
| 9 |
< |
import dlsApi |
| 10 |
< |
import dlsClient |
| 11 |
< |
from dlsDataObjects import DlsLocation, DlsFileBlock, DlsEntry |
| 12 |
< |
except: |
| 13 |
< |
try: |
| 14 |
< |
Crabpydir=commands.getoutput('which crab') |
| 15 |
< |
Topdir=string.replace(Crabpydir,'/python/crab','') |
| 16 |
< |
sys.path.append(Topdir+'/DLSAPI') |
| 17 |
< |
import dlsApi |
| 18 |
< |
import dlsClient |
| 19 |
< |
from dlsDataObjects import DlsLocation, DlsFileBlock, DlsEntry |
| 20 |
< |
except: |
| 21 |
< |
msg="ERROR no DLS API available" |
| 22 |
< |
raise CrabException(msg) |
| 8 |
> |
import dlsApi |
| 9 |
> |
import dlsClient |
| 10 |
> |
from dlsDataObjects import DlsLocation, DlsFileBlock, DlsEntry |
| 11 |
|
|
| 12 |
|
## for python 2.2 add the pyexpat.so to PYTHONPATH |
| 13 |
|
pythonV=sys.version.split(' ')[0] |
| 28 |
|
|
| 29 |
|
class DLSNoReplicas(exceptions.Exception): |
| 30 |
|
def __init__(self, FileBlock): |
| 31 |
< |
args ="No replicas exists for fileblock: "+FileBlock+"\n" |
| 32 |
< |
exceptions.Exception.__init__(self, args) |
| 31 |
> |
self.args ="No replicas exists for fileblock: "+FileBlock+"\n" |
| 32 |
> |
exceptions.Exception.__init__(self, self.args) |
| 33 |
|
pass |
| 34 |
|
|
| 35 |
|
def getClassName(self): |
| 48 |
|
class DLSInfo: |
| 49 |
|
def __init__(self, type, cfg_params): |
| 50 |
|
self.cfg_params = cfg_params |
| 63 |
– |
jobtype = self.cfg_params['CRAB.jobtype'] |
| 51 |
|
if type=="DLS_TYPE_DLI": |
| 52 |
|
try: |
| 53 |
|
endpoint=self.cfg_params['CMSSW.dls_endpoint'] |
| 72 |
|
endpoint=self.cfg_params['CMSSW.dls_endpoint'] |
| 73 |
|
except KeyError: |
| 74 |
|
endpoint="lxgate10.cern.ch:18081" |
| 75 |
+ |
elif type=="DLS_TYPE_DBS": |
| 76 |
+ |
try: |
| 77 |
+ |
endpoint=self.cfg_params['CMSSW.dls_endpoint'] |
| 78 |
+ |
except KeyError: |
| 79 |
+ |
endpoint="http://cmsdbsprod.cern.ch/cms_dbs_prod_global/servlet/DBSServlet" |
| 80 |
|
else: |
| 81 |
|
msg = "DLS type %s not among the supported DLS ( DLS_TYPE_DLI and DLS_TYPE_MYSQL ) "%type |
| 82 |
|
raise CrabException(msg) |
| 85 |
|
try: |
| 86 |
|
self.api = dlsClient.getDlsApi(dls_type=type,dls_endpoint=endpoint) |
| 87 |
|
except dlsApi.DlsApiError, inst: |
| 88 |
< |
msg = "Error when binding the DLS interface: %s Server %s"%(str(inst),self.DLSServer_) |
| 88 |
> |
msg = "Error when binding the DLS interface: %s Server %s"%(str(inst),endpoint) |
| 89 |
|
#print msg |
| 90 |
|
raise CrabException(msg) |
| 91 |
|
|
| 98 |
|
try: |
| 99 |
|
entryList=self.api.getLocations([fileblocks]) |
| 100 |
|
except dlsApi.DlsApiError, inst: |
| 101 |
< |
msg = "Error in the DLS query: %s." % str(inst) |
| 101 |
> |
#msg = "Error in the DLS query: %s." % str(inst) |
| 102 |
|
#print msg |
| 103 |
|
raise DLSNoReplicas(fileblocks) |
| 104 |
|
|
