| 4 |
|
""" |
| 5 |
|
|
| 6 |
|
from crab_exceptions import * |
| 7 |
< |
from crab_util import runCommand |
| 7 |
> |
from crab_util import runCommand, getUserName |
| 8 |
|
import common |
| 9 |
< |
import os, string |
| 9 |
> |
import os, string, time |
| 10 |
|
|
| 11 |
|
|
| 12 |
< |
def LFNBase(ProcessedDataset,merged=True): |
| 12 |
> |
def LFNBase(forced_path, PrimaryDataset='',ProcessedDataset='',merged=True,publish=False): |
| 13 |
|
""" |
| 14 |
|
""" |
| 15 |
< |
lfnbase = "/store" |
| 16 |
< |
if not merged: |
| 17 |
< |
lfnbase = os.path.join(lfnbase,"tmp") |
| 18 |
< |
|
| 19 |
< |
lfnbase = os.path.join(lfnbase, "user", gethnUserName(), ProcessedDataset ) |
| 15 |
> |
if (PrimaryDataset == 'null'): |
| 16 |
> |
PrimaryDataset = ProcessedDataset |
| 17 |
> |
if PrimaryDataset != '': |
| 18 |
> |
if ( PrimaryDataset[0] == '/' ): PrimaryDataset=PrimaryDataset[1:] |
| 19 |
> |
if forced_path.find('store/group')>0: |
| 20 |
> |
lfnbase = os.path.join(forced_path, PrimaryDataset, ProcessedDataset) |
| 21 |
> |
else: |
| 22 |
> |
lfnbase = os.path.join(forced_path, getUserName(), PrimaryDataset, ProcessedDataset) |
| 23 |
> |
if (publish == True): |
| 24 |
> |
checkSlash(ProcessedDataset) |
| 25 |
> |
checkLength(lfnbase, forced_path, PrimaryDataset, ProcessedDataset) |
| 26 |
|
return lfnbase |
| 27 |
|
|
| 28 |
< |
def PFNportion(ProcessedDataset): |
| 29 |
< |
pfnpath = os.path.join(gethnUserName(), ProcessedDataset ) |
| 30 |
< |
return pfnpath |
| 31 |
< |
|
| 32 |
< |
def getDN(): |
| 33 |
< |
""" |
| 34 |
< |
extract DN from user proxy's identity |
| 35 |
< |
""" |
| 36 |
< |
try: |
| 37 |
< |
userdn = runCommand("voms-proxy-info -identity") |
| 38 |
< |
userdn = string.strip(userdn) |
| 39 |
< |
except: |
| 40 |
< |
msg = "Error. Problem with voms-proxy-info -identity command" |
| 41 |
< |
raise CrabException(msg) |
| 42 |
< |
return userdn |
| 43 |
< |
|
| 44 |
< |
def gethnUserName(): |
| 45 |
< |
""" |
| 46 |
< |
extract user name from SiteDB |
| 47 |
< |
""" |
| 48 |
< |
import urllib |
| 49 |
< |
hnUserName = None |
| 50 |
< |
userdn = getDN() |
| 51 |
< |
try: |
| 52 |
< |
sitedburl="https://cmsweb.cern.ch/sitedb/sitedb/json/index/dnUserName" |
| 53 |
< |
params = urllib.urlencode({'dn': userdn }) |
| 54 |
< |
f = urllib.urlopen(sitedburl,params) |
| 55 |
< |
udata = f.read() |
| 56 |
< |
try: |
| 57 |
< |
userinfo= eval(udata) |
| 58 |
< |
except StandardError, ex: |
| 59 |
< |
msg = "Error. Problem extracting user name from %s : %s"%(sitedburl,ex) |
| 60 |
< |
raise CrabException(msg) |
| 61 |
< |
hnUserName = userinfo['user'] |
| 62 |
< |
except: |
| 63 |
< |
msg = "Error. Problem extracting user name from %s"%sitedburl |
| 64 |
< |
msg += "Check that you are registered in SiteDB, see https://twiki.cern.ch/twiki/bin/view/CMS/SiteDBForCRAB" |
| 65 |
< |
raise CrabException(msg) |
| 66 |
< |
if not hnUserName: |
| 67 |
< |
msg = "Error. There is no user name associated to DN %s in %s. You need to register in SiteDB with the instructions at https://twiki.cern.ch/twiki/bin/view/CMS/SiteDBForCRAB"%(userdn,sitedburl) |
| 68 |
< |
print msg |
| 69 |
< |
raise CrabException(msg) |
| 70 |
< |
return hnUserName |
| 28 |
> |
def checkSlash(ProcessedDataset): |
| 29 |
> |
""" |
| 30 |
> |
check if [USER].publish_data_name contains not allowed slash |
| 31 |
> |
""" |
| 32 |
> |
import re |
| 33 |
> |
|
| 34 |
> |
#if (str(ProcessedDataset).find("/") > -1): |
| 35 |
> |
# raise CrabException("Warning: [USER] publication_data_name contains a '/' characters that is not allowed. Please change the publication_data_name value") |
| 36 |
> |
m = re.findall("[^-\w_\.%]+", str(ProcessedDataset)) |
| 37 |
> |
if (m != []): |
| 38 |
> |
common.logger.debug("check not allowed characters in the publication_data_name = " + str(m)) |
| 39 |
> |
raise CrabException("Warning: [USER] publication_data_name contains not allowed characters %s . Please change the publication_data_name value" % str(m)) |
| 40 |
> |
|
| 41 |
> |
def checkLength(lfnbase, forced_path, PrimaryDataset, ProcessedDataset): |
| 42 |
> |
""" |
| 43 |
> |
""" |
| 44 |
> |
|
| 45 |
> |
len_primary = len(PrimaryDataset) |
| 46 |
> |
|
| 47 |
> |
common.logger.debug("CheckLength of LFN and User DatasetName") |
| 48 |
> |
common.logger.debug("max length for complete LFN is 500 characters, max length for primary dataset is 100 characters") |
| 49 |
> |
common.logger.debug("PrimaryDataset = " + PrimaryDataset) |
| 50 |
> |
common.logger.debug("len_primary = " + str(len_primary)) |
| 51 |
> |
|
| 52 |
> |
if (len_primary > 100): |
| 53 |
> |
raise CrabException("Warning: you required user_data_publication. The PrimaryDatasetName has to be < 100 characters") |
| 54 |
> |
|
| 55 |
> |
if (PrimaryDataset != ProcessedDataset): |
| 56 |
> |
common.logger.debug("ProcessedDataset = " + ProcessedDataset) |
| 57 |
> |
common.logger.debug("len(ProcessedDataset) = " + str(len(ProcessedDataset))) |
| 58 |
> |
|
| 59 |
> |
common.logger.debug("forced_path = " + forced_path) |
| 60 |
> |
common.logger.debug("len(forced_path) = " + str(len(forced_path))) |
| 61 |
> |
|
| 62 |
> |
user = getUserName() |
| 63 |
> |
len_user_name = len(user) |
| 64 |
> |
common.logger.debug("user = " + user) |
| 65 |
> |
common.logger.debug("len_user_name = " + str(len_user_name)) |
| 66 |
> |
|
| 67 |
> |
common.logger.debug("lfnbase = " + lfnbase) |
| 68 |
> |
common.logger.debug("len(lfnbase) = " + str(len(lfnbase))) |
| 69 |
> |
|
| 70 |
> |
### we suppose a output_file_name of 50 characters ### |
| 71 |
> |
if len(lfnbase)>450: |
| 72 |
> |
if (PrimaryDataset != ProcessedDataset): |
| 73 |
> |
#500 - len_user_name - len_primary - len(forced) - len(PSETHASH = 32) - 4(/) - output(50) |
| 74 |
> |
#~400 - len_user_name - len_primary - len(forced) |
| 75 |
> |
if (len(ProcessedDataset) > (400 - len_user_name - len_primary - len(forced_path))): |
| 76 |
> |
raise CrabException("Warning: publication name too long. USER.publish_data_name has to be < " + str(400 - len_user_name - len_primary - len(forced_path)) + " characters") |
| 77 |
> |
else: |
| 78 |
> |
raise CrabException("Warning: LFN > 500 characters") |
| 79 |
> |
else: |
| 80 |
> |
if (len(ProcessedDataset) > (400 - len_user_name - len(forced_path)) / 2): |
| 81 |
> |
raise CrabException("Warning: publication name too long. USER.publish_data_name has to be < " + str((400 - len_user_name - len(forced_path)) / 2) + " characters") |
| 82 |
> |
else: |
| 83 |
> |
raise CrabException("Warning: LFN > 500 characters") |
| 84 |
> |
|
| 85 |
|
|
| 86 |
|
if __name__ == '__main__' : |
| 87 |
|
""" |
| 88 |
|
""" |
| 89 |
< |
from crab_logger import Logger |
| 90 |
< |
from WorkSpace import * |
| 71 |
< |
continue_dir="/bohome/fanfani/CRAB" |
| 72 |
< |
cfg_params={'USER.logdir' : continue_dir } |
| 73 |
< |
common.work_space = WorkSpace(continue_dir, cfg_params) |
| 74 |
< |
log = Logger() |
| 75 |
< |
common.logger = log |
| 89 |
> |
import logging |
| 90 |
> |
common.logger = logging |
| 91 |
|
|
| 92 |
+ |
print "xx %s xx"%getUserName() |
| 93 |
|
baselfn = LFNBase("datasetstring") |
| 94 |
< |
print baselfn |
| 94 |
> |
print baselfn |
| 95 |
|
|
| 96 |
|
unmergedlfn = LFNBase("datasetstring",merged=False) |
| 97 |
< |
print unmergedlfn |
| 97 |
> |
print unmergedlfn |
| 98 |
|
print PFNportion("datasetstring") |
