| 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 |
try:
|
| 17 |
f = urllib.urlopen("https://cmsweb.cern.ch/sitedb/json/index/dnUserName?%s" % dn)
|
| 18 |
username = str(f.read())
|
| 19 |
f.close()
|
| 20 |
except:
|
| 21 |
print "failed to get username via SiteDB V1 API"
|
| 22 |
username="None"
|
| 23 |
|
| 24 |
cmd = "curl -s --capath $X509_CERT_DIR --cert $X509_USER_PROXY --key $X509_USER_PROXY 'https://cmsweb.cern.ch/sitedb/data/prod/whoami'|tr ':,' '\n'|grep -A1 login|tail -1"
|
| 25 |
uname = subprocess.check_output(cmd,shell=True)
|
| 26 |
|
| 27 |
|
| 28 |
print 'v1: my HN user name is: %s' % username
|
| 29 |
print 'v2: my HN user name is: %s' % uname
|
| 30 |
print '\nend check.....................'
|
| 31 |
|
| 32 |
def crabCheck(self):
|
| 33 |
from CrabLogger import CrabLogger
|
| 34 |
from WorkSpace import WorkSpace, common
|
| 35 |
import tempfile, urllib, os, string
|
| 36 |
|
| 37 |
dname = tempfile.mkdtemp( "", "crab_", '/tmp' )
|
| 38 |
os.system("mkdir %s/log"%dname )
|
| 39 |
os.system("touch %s/crab.log"%dname )
|
| 40 |
|
| 41 |
cfg_params={'USER.logdir' : dname }
|
| 42 |
common.work_space = WorkSpace(dname, cfg_params)
|
| 43 |
args = string.join(sys.argv,' ')
|
| 44 |
common.debugLevel = 0
|
| 45 |
common.logger = CrabLogger(args)
|
| 46 |
|
| 47 |
from crab_util import getDN,gethnUserNameFromSiteDB
|
| 48 |
print 'start using CRAB utils ...\n'
|
| 49 |
print "my DN is: %s \n"%getDN()
|
| 50 |
try:
|
| 51 |
print 'my HN user name is: %s \n'%gethnUserNameFromSiteDB()
|
| 52 |
except:
|
| 53 |
print '\nWARNING native crab_utils failed! '
|
| 54 |
dn=urllib.urlencode({'dn':getDN()})
|
| 55 |
print 'trying now using urlencoded DN: \n\t %s '%dn
|
| 56 |
status,hnName = self.gethnName_urlenc(dn)
|
| 57 |
if status == 1:
|
| 58 |
print '\nWARNING: failed also using urlencoded DN '
|
| 59 |
else:
|
| 60 |
print 'my HN user name is: %s \n'%name
|
| 61 |
print 'problems with crab_utils'
|
| 62 |
print '\nend check.....................'
|
| 63 |
|
| 64 |
os.system("rm -rf %s"%dname )
|
| 65 |
|
| 66 |
def gethnName_urlenc(self,dn):
|
| 67 |
from WMCore.Services.SiteDB.SiteDB import SiteDBJSON
|
| 68 |
hnUserName = None
|
| 69 |
userdn = dn
|
| 70 |
mySiteDB = SiteDBJSON()
|
| 71 |
status = 0
|
| 72 |
try:
|
| 73 |
hnUserName = mySiteDB.dnUserName(dn=userdn)
|
| 74 |
except:
|
| 75 |
status = 1
|
| 76 |
return status,hnUserName
|
| 77 |
|
| 78 |
|
| 79 |
if __name__ == '__main__' :
|
| 80 |
import sys
|
| 81 |
import subprocess
|
| 82 |
args = sys.argv[1:]
|
| 83 |
check = check_HN_name()
|
| 84 |
|
| 85 |
# print "verify if user DN is mapped in CERN's SSO"
|
| 86 |
# cmd='curl -s --capath $X509_CERT_DIR --cert $X509_USER_PROXY --key $X509_USER_PROXY "https://cmsweb.cern.ch/sitedb/data/dev/people" | grep "`voms-proxy-info -identity`"'
|
| 87 |
# DocUrl = "\nhttps://hypernews.cern.ch/HyperNews/CMS/get/cernCompAnnounce/820.html"
|
| 88 |
# DocUrl += "\nhttps://twiki.cern.ch/twiki/bin/view/CMS/SiteDBForCRAB"
|
| 89 |
|
| 90 |
# try:
|
| 91 |
# subprocess.check_call(cmd,shell=True)
|
| 92 |
# print "OK. user ready for SiteDB switchover on March 12, 2013\n\n"
|
| 93 |
# except:
|
| 94 |
# print "********** ATTENTION !!!! **************"
|
| 95 |
# print "DN not registered in CERN's SSO"
|
| 96 |
# print "need to follow instructions in %s" % DocUrl
|
| 97 |
# print "or Crab will stop working for you on March 12, 2013"
|
| 98 |
# print "*****************************************"
|
| 99 |
# dummy=raw_input("\nHit Return to acknowlege that you read this message")
|
| 100 |
|
| 101 |
if 'crab' in args:
|
| 102 |
check.crabCheck()
|
| 103 |
else:
|
| 104 |
check.stdaloneCheck()
|
| 105 |
|
