python/utils/addToDatasets.py

0001 #!/usr/bin/env python
0002 # Colin
0003 # additional layer, on top of cmsBatch.py
0004
0005 import os
0006
0007 from .castorBaseDir import getUserAndArea
0008
0009 def addToDatasets(sampleAndTier, user=os.getlogin()):
0010     user, _ = getUserAndArea(user) #for if we have a magic user name
0011     notThere = os.system( 'grep %s ~%s/public/DataSets.txt' % (sampleAndTier,user) )
0012     if notThere:
0013         os.system('echo %s >> ~%s/public/DataSets.txt' % (sampleAndTier,user) )
0014     return not notThere
0015
0016
0017 if __name__ == '__main__':
0018     import sys
0019     addToDatasets(sys.argv[1], sys.argv[2])