addFileToExp.py
145 lines
| 4.4 KiB
| text/x-python
|
PythonLexer
| r0 | #!PYTHONEXE | |||
| #$Id: addFileToExp.py 7045 2019-10-07 19:56:46Z brideout $ | ||||
| usage = """ | ||||
| addFileToExp.py is a script used to add a new Madrigal file to an | ||||
| existing experiment. Information such as the duration of the | ||||
| experiment is updated by analyzing the file. | ||||
| With Madrigal 3, accepts either old CEDAR database format files, or CEDAR Madrigal | ||||
| Hdf5 files. | ||||
| Required arguments: | ||||
| --madFilename - full path to the complete Madrigal file. Basename will | ||||
| be maintained. If old CEDAR databse format, .hdf5 will | ||||
| be appended | ||||
| --expDir - full path to experiment directory. Example: | ||||
| "/opt/madrigal/experiments/1998/mlh/20jan98" | ||||
| --permission - 0 for public, 1 for private (restricted to certain IP range) | ||||
| --fileDesc - file decription | ||||
| Optional arguments: | ||||
| --category - 1=default, 2=variant, or 3=history If this argument is missing, | ||||
| 1 (default) used. | ||||
| --kindat - Set file kindat independently from one (or more) in file | ||||
| --skipNotify - if this flag set (no arguments), registered users will NOT be notified | ||||
| of this change. The default is to email all registered users. | ||||
| --fileAnalyst - set file analyst name for this file. This will default to blank. | ||||
| --fileAnalystEmail - set file analyst email for this file. This will default to blank. | ||||
| --createCachedText - create a cached text file in overview. Default is no cached file. | ||||
| Improves speed when text version is downloaded; uses disk space. | ||||
| --createCachedNetCDF4 - create a cached netCDF4 file in overview. Default is no cached file. | ||||
| Improves speed when netCDF4 version is downloaded; uses disk space. | ||||
| """ | ||||
| import sys | ||||
| import os, os.path | ||||
| import getopt | ||||
| import traceback | ||||
| import madrigal.admin | ||||
| # parse command line | ||||
| arglist = '' | ||||
| longarglist = ['madFilename=', | ||||
| 'expDir=', | ||||
| 'permission=', | ||||
| 'fileDesc=', | ||||
| 'category=', | ||||
| 'kindat=', | ||||
| 'skipNotify', | ||||
| 'fileAnalyst=', | ||||
| 'fileAnalystEmail=', | ||||
| 'createCachedText', | ||||
| 'createCachedNetCDF4'] | ||||
| optlist, args = getopt.getopt(sys.argv[1:], arglist, longarglist) | ||||
| # set default values | ||||
| madFilename = None | ||||
| expDir = None | ||||
| permission = None | ||||
| fileDesc = None | ||||
| category = 1 | ||||
| kindat = None | ||||
| notify = True | ||||
| fileAnalyst = '' | ||||
| fileAnalystEmail = '' | ||||
| createCachedText = False | ||||
| createCachedNetCDF4 = False | ||||
| for opt in optlist: | ||||
| if opt[0] == '--madFilename': | ||||
| madFilename = opt[1] | ||||
| elif opt[0] == '--expDir': | ||||
| expDir = opt[1] | ||||
| elif opt[0] == '--permission': | ||||
| permission = int(opt[1]) | ||||
| elif opt[0] == '--fileDesc': | ||||
| fileDesc = opt[1] | ||||
| elif opt[0] == '--category': | ||||
| category = int(opt[1]) | ||||
| elif opt[0] == '--kindat': | ||||
| kindat = int(opt[1]) | ||||
| elif opt[0] == '--skipNotify': | ||||
| notify = False | ||||
| elif opt[0] == '--fileAnalyst': | ||||
| fileAnalyst = opt[1] | ||||
| elif opt[0] == '--fileAnalystEmail': | ||||
| fileAnalystEmail = opt[1] | ||||
| elif opt[0] == '--createCachedText': | ||||
| createCachedText = True | ||||
| elif opt[0] == '--createCachedNetCDF4': | ||||
| createCachedNetCDF4 = True | ||||
| else: | ||||
| raise ValueError('Illegal option %s\n%s' % (opt[0], usage)) | ||||
| # check that all required arguments passed in | ||||
| if madFilename == None: | ||||
| print('--madFilename argument required - must be full path to madrigal file') | ||||
| print(usage) | ||||
| sys.exit(0) | ||||
| if expDir == None: | ||||
| print('--expDir argument required - must be full path to experiment directory') | ||||
| sys.exit(0) | ||||
| if permission == None: | ||||
| print('--permission argument required - must be 0 for public, 1 for private') | ||||
| sys.exit(0) | ||||
| if fileDesc == None: | ||||
| print('--fileDesc argument required - must be file description') | ||||
| sys.exit(0) | ||||
| adminObj = madrigal.admin.MadrigalDBAdmin() | ||||
| adminObj.addMadrigalFile(expDir, | ||||
| madFilename, | ||||
| permission, | ||||
| fileDesc, | ||||
| category, | ||||
| kindat, | ||||
| notify=notify, | ||||
| fileAnalyst=fileAnalyst, | ||||
| fileAnalystEmail=fileAnalystEmail, | ||||
| createCachedText=createCachedText, | ||||
| createCachedNetCDF4=createCachedNetCDF4, | ||||
| updateToMad3=True) | ||||
| print('File %s successfully added to experiment at %s - run updateMaster to register change' % (madFilename, expDir)) | ||||
