python/utils/addToDatasets.py

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