| 1 |
slacapra |
1.1 |
class check_HN_name:
|
| 2 |
|
|
def init(self):
|
| 3 |
|
|
pass
|
| 4 |
|
|
|
| 5 |
|
|
def stdaloneCheck(self):
|
| 6 |
|
|
|
| 7 |
|
|
import urllib
|
| 8 |
|
|
from commands import getstatusoutput
|
| 9 |
|
|
print 'start standalone check ...\n'
|
| 10 |
|
|
status, dn = getstatusoutput('voms-proxy-info -identity')
|
| 11 |
|
|
if status == 0:
|
| 12 |
|
|
print "my DN is: %s \n"%dn
|
| 13 |
|
|
dn = dn.split('\n')[-1]
|
| 14 |
|
|
dn = urllib.urlencode({'dn':dn})
|
| 15 |
|
|
print 'Using urlencoded DN: \n\t %s '%dn
|
| 16 |
belforte |
1.2 |
f = urllib.urlopen("https://cmsweb-testbed.cern.ch/sitedb/json/index/dnUserName?%s" % dn)
|
| 17 |
slacapra |
1.1 |
print 'my HN user name is: %s \n'%str(f.read())
|
| 18 |
|
|
f.close()
|
| 19 |
|
|
print '\nend check.....................'
|
| 20 |
|
|
|
| 21 |
|
|
def crabCheck(self):
|
| 22 |
|
|
from CrabLogger import CrabLogger
|
| 23 |
|
|
from WorkSpace import *
|
| 24 |
|
|
import tempfile, urllib, os
|
| 25 |
|
|
|
| 26 |
|
|
dname = tempfile.mkdtemp( "", "crab_", '/tmp' )
|
| 27 |
|
|
os.system("mkdir %s/log"%dname )
|
| 28 |
|
|
os.system("touch %s/crab.log"%dname )
|
| 29 |
|
|
|
| 30 |
|
|
cfg_params={'USER.logdir' : dname }
|
| 31 |
|
|
common.work_space = WorkSpace(dname, cfg_params)
|
| 32 |
|
|
args = string.join(sys.argv,' ')
|
| 33 |
|
|
common.debugLevel = 0
|
| 34 |
|
|
common.logger = CrabLogger(args)
|
| 35 |
|
|
|
| 36 |
|
|
from crab_util import getDN,gethnUserNameFromSiteDB
|
| 37 |
|
|
print 'start using CRAB utils ...\n'
|
| 38 |
|
|
print "my DN is: %s \n"%getDN()
|
| 39 |
|
|
try:
|
| 40 |
|
|
print 'my HN user name is: %s \n'%gethnUserNameFromSiteDB()
|
| 41 |
|
|
except:
|
| 42 |
|
|
print '\nWARNING native crab_utils failed! '
|
| 43 |
|
|
dn=urllib.urlencode({'dn':getDN()})
|
| 44 |
|
|
print 'trying now using urlencoded DN: \n\t %s '%dn
|
| 45 |
|
|
status,hnName = self.gethnName_urlenc(dn)
|
| 46 |
|
|
if status == 1:
|
| 47 |
|
|
print '\nWARNING: failed also using urlencoded DN '
|
| 48 |
|
|
else:
|
| 49 |
|
|
print 'my HN user name is: %s \n'%name
|
| 50 |
|
|
print 'problems with crab_utils'
|
| 51 |
|
|
print '\nend check.....................'
|
| 52 |
|
|
|
| 53 |
|
|
os.system("rm -rf %s"%dname )
|
| 54 |
|
|
|
| 55 |
|
|
def gethnName_urlenc(self,dn):
|
| 56 |
|
|
from WMCore.Services.SiteDB.SiteDB import SiteDBJSON
|
| 57 |
|
|
hnUserName = None
|
| 58 |
|
|
userdn = dn
|
| 59 |
|
|
mySiteDB = SiteDBJSON()
|
| 60 |
|
|
status = 0
|
| 61 |
|
|
try:
|
| 62 |
|
|
hnUserName = mySiteDB.dnUserName(dn=userdn)
|
| 63 |
|
|
except:
|
| 64 |
|
|
status = 1
|
| 65 |
|
|
return status,hnUserName
|
| 66 |
|
|
|
| 67 |
|
|
|
| 68 |
|
|
if __name__ == '__main__' :
|
| 69 |
|
|
import sys
|
| 70 |
|
|
args = sys.argv[1:]
|
| 71 |
|
|
check = check_HN_name()
|
| 72 |
|
|
if 'crab' in args:
|
| 73 |
|
|
check.crabCheck()
|
| 74 |
|
|
else:
|
| 75 |
|
|
check.stdaloneCheck()
|
| 76 |
|
|
|
