| 4 |
|
""" |
| 5 |
|
|
| 6 |
|
from crab_exceptions import * |
| 7 |
< |
from crab_util import runCommand |
| 7 |
> |
from crab_util import runCommand, UnixUserName |
| 8 |
|
import common |
| 9 |
< |
import os, string |
| 10 |
< |
from ProdCommon.SiteDB.SiteDB import SiteDBJSON |
| 9 |
> |
import os, string, time |
| 10 |
> |
from WMCore.Services.SiteDB.SiteDB import SiteDBJSON |
| 11 |
|
|
| 12 |
|
|
| 13 |
< |
def LFNBase(ProcessedDataset,merged=True,LocalUser=False): |
| 13 |
> |
def LFNBase(forced_path, PrimaryDataset='',ProcessedDataset='',merged=True,LocalUser=False,publish=False): |
| 14 |
|
""" |
| 15 |
|
""" |
| 16 |
< |
lfnbase = "/store" |
| 17 |
< |
if not merged: |
| 18 |
< |
lfnbase = os.path.join(lfnbase,"tmp") |
| 19 |
< |
lfnbase = os.path.join(lfnbase, "user", getUserName(LocalUser=LocalUser), ProcessedDataset ) |
| 16 |
> |
#lfnbase = "/store" |
| 17 |
> |
#if not merged: |
| 18 |
> |
#lfnbase = os.path.join(lfnbase,"tmp") |
| 19 |
> |
if (PrimaryDataset == 'null'): |
| 20 |
> |
PrimaryDataset = ProcessedDataset |
| 21 |
> |
if ( PrimaryDataset[0] == '/' ): PrimaryDataset=PrimaryDataset[1:] |
| 22 |
> |
# lfnbase = os.path.join(lfnbase, "user", getUserName(LocalUser=LocalUser), PrimaryDataset, ProcessedDataset ) |
| 23 |
> |
lfnbase = os.path.join(forced_path, getUserName(LocalUser=LocalUser), PrimaryDataset, ProcessedDataset) |
| 24 |
|
|
| 25 |
|
return lfnbase |
| 26 |
|
|
| 23 |
– |
def PFNportion(ProcessedDataset,LocalUser=False): |
| 24 |
– |
pfnpath = os.path.join(getUserName(LocalUser=LocalUser), ProcessedDataset ) |
| 25 |
– |
return pfnpath |
| 26 |
– |
|
| 27 |
|
def getUnixUserName(): |
| 28 |
|
""" |
| 29 |
|
extract username from whoami |
| 30 |
|
""" |
| 31 |
< |
try: |
| 32 |
< |
UserName = runCommand("whoami") |
| 33 |
< |
UserName = string.strip(UserName) |
| 34 |
< |
except: |
| 35 |
< |
msg = "Error. Problem with whoami command" |
| 36 |
< |
raise CrabException(msg) |
| 37 |
< |
return UserName |
| 31 |
> |
return UnixUserName() |
| 32 |
|
|
| 33 |
|
def getDN(): |
| 34 |
|
""" |
| 44 |
|
raise CrabException(msg) |
| 45 |
|
return userdn.split('\n')[0] |
| 46 |
|
|
| 47 |
< |
def gethnUserName(): |
| 47 |
> |
def gethnUserNameFromSiteDB(): |
| 48 |
|
""" |
| 49 |
|
extract user name from SiteDB |
| 50 |
|
""" |
| 51 |
|
hnUserName = None |
| 52 |
|
userdn = getDN() |
| 53 |
|
mySiteDB = SiteDBJSON() |
| 60 |
– |
|
| 54 |
|
try: |
| 55 |
|
hnUserName = mySiteDB.dnUserName(dn=userdn) |
| 56 |
|
except: |
| 63 |
|
raise CrabException(msg) |
| 64 |
|
return hnUserName |
| 65 |
|
|
| 66 |
+ |
|
| 67 |
+ |
def gethnUserName(): |
| 68 |
+ |
""" |
| 69 |
+ |
cache the username extracted from SiteDB for 24hours |
| 70 |
+ |
""" |
| 71 |
+ |
userconfigFileName="SiteDBusername.conf" |
| 72 |
+ |
if not os.path.exists(userconfigFileName): |
| 73 |
+ |
common.logger.debug(5,"Downloading from SiteDB username into %s"%userconfigFileName) |
| 74 |
+ |
nameuser = gethnUserNameFromSiteDB() |
| 75 |
+ |
userfile = open(userconfigFileName, 'w') |
| 76 |
+ |
userfile.write(nameuser) |
| 77 |
+ |
userfile.close() |
| 78 |
+ |
else: |
| 79 |
+ |
statinfo = os.stat(userconfigFileName) |
| 80 |
+ |
## if the file is older then 24 hours it is re-downloaded to update the configuration |
| 81 |
+ |
oldness = 24*3600 |
| 82 |
+ |
if (time.time() - statinfo.st_ctime) > oldness: |
| 83 |
+ |
common.logger.debug(5,"Downloading from SiteDB username into %s"%userconfigFileName) |
| 84 |
+ |
nameuser = gethnUserNameFromSiteDB() |
| 85 |
+ |
userfile = open(userconfigFileName, 'w') |
| 86 |
+ |
userfile.write(nameuser) |
| 87 |
+ |
userfile.close() |
| 88 |
+ |
else: |
| 89 |
+ |
userfile = open(userconfigFileName, 'r') |
| 90 |
+ |
for line in userfile.readlines(): |
| 91 |
+ |
nameuser = line |
| 92 |
+ |
userfile.close() |
| 93 |
+ |
nameuser = string.strip(nameuser) |
| 94 |
+ |
return nameuser |
| 95 |
+ |
|
| 96 |
|
def getUserName(LocalUser=False): |
| 97 |
|
""" |
| 98 |
|
extract user name from either SiteDB or Unix |
| 99 |
|
""" |
| 100 |
|
if LocalUser: |
| 101 |
< |
common.logger.message("==> Using as username the Unix user name") |
| 101 |
> |
common.logger.debug(10,"Using as username the Unix user name") |
| 102 |
|
UserName=getUnixUserName() |
| 103 |
|
return UserName |
| 104 |
|
|
