updateFileTab.py
134 lines
| 4.1 KiB
| text/x-python
|
PythonLexer
| r0 | #!PYTHONEXE | |||
| import sys | ||||
| import os, os.path | ||||
| import time | ||||
| import traceback | ||||
| """updateFileTab.py updates every fileTab.txt file to correctly fill in fields 6 and 7 | ||||
| (#6 - File contains at least one Catalog Record File - 0 for no, 1 for yes) | ||||
| #7 File contains at least one Header Record File - 0 for no, 1 for yes) | ||||
| """ | ||||
| # catch any exception, and write an appropriate message admin | ||||
| try: | ||||
| # check if pythonlibpath env variable exists | ||||
| # written 'PYTHON' + 'LIBPATH' to stop automatic replacement during setup | ||||
| temp = os.environ.get('PYTHON' + 'LIBPATH') | ||||
| if temp != None: | ||||
| sys.path.append(temp) | ||||
| # append path madroot/lib (needed only if python not installed by setup) | ||||
| sys.path.append('MADROOT/lib/python') | ||||
| # prepare to handle MadrigalError | ||||
| import madrigal.admin | ||||
| except ImportError: | ||||
| # Fatal error - madpy library not found | ||||
| print("Unable to import the madrigal python library - please alert the sys admin!") | ||||
| sys.exit(0) | ||||
| # try to run script, and report all errors to Madrigal sys admin | ||||
| try: | ||||
| import madrigal.metadata | ||||
| import madrigal.data | ||||
| # create MadrigalDB obj | ||||
| madDBObj = madrigal.metadata.MadrigalDB() | ||||
| fileList = madDBObj.getFileListFromMetadata() | ||||
| print('Setting values for hasCatalog and hasHeader in fileTab.txt files:') | ||||
| # loop through each file | ||||
| dirName = None | ||||
| for dataFile in fileList: | ||||
| thisDir = os.path.dirname(dataFile) | ||||
| if thisDir != dirName: | ||||
| # new fileTab.txt found - fix it | ||||
| newFileTab = os.path.join(thisDir, 'fileTab.txt') | ||||
| metaFileObj = madrigal.metadata.MadrigalMetaFile(madDBObj, newFileTab) | ||||
| # now loop through each file in the metaFileObj | ||||
| for i in range(metaFileObj.getFileCount()): | ||||
| thisFile = metaFileObj.getFilenameByPosition(i) | ||||
| thisFileName = os.path.join(thisDir, thisFile) | ||||
| try: | ||||
| catList, headList = madrigal._Madrec.cedarCatalogHeaderList(thisFileName) | ||||
| except: | ||||
| # problem reading file | ||||
| catList = [] | ||||
| headList = [] | ||||
| if len(catList) > 0: | ||||
| metaFileObj.setHasCatalogByPosition(i, True) | ||||
| else: | ||||
| metaFileObj.setHasCatalogByPosition(i, False) | ||||
| if len(headList) > 0: | ||||
| metaFileObj.setHasHeaderByPosition(i, True) | ||||
| else: | ||||
| metaFileObj.setHasHeaderByPosition(i, False) | ||||
| # write new fileTab | ||||
| metaFileObj.writeMetadata() | ||||
| print('\t%s' % (newFileTab)) | ||||
| dirName = thisDir | ||||
| except madrigal.admin.MadrigalError as e: | ||||
| # handle a MadrigalError | ||||
| errStr = '<h1> Error occurred in updateFileTab.py</h1>' | ||||
| errStr = errStr + e.getExceptionHtml() | ||||
| err = traceback.format_exception(sys.exc_info()[0], | ||||
| sys.exc_info()[1], | ||||
| sys.exc_info()[2]) | ||||
| for errItem in err: | ||||
| errStr = errStr + '<br>\n' + str(errItem) | ||||
| errStr = errStr + '\nError occurred at ' + str(time.asctime(time.localtime())) + '<br>\n' | ||||
| admin = madrigal.admin.MadrigalNotify() | ||||
| admin.sendAlert('<html>\n' + errStr + '</html>', | ||||
| 'Error running updateFileTab.py' ) | ||||
| except: | ||||
| # handle a normal error | ||||
| # if a normal SystemExit, simply terminate | ||||
| if str(sys.exc_info()[0]).find('exceptions.SystemExit') != -1: | ||||
| sys.exit(0) | ||||
| errStr = '<h1> Error occurred in running updateFileTab.py.</h1>' | ||||
| err = traceback.format_exception(sys.exc_info()[0], | ||||
| sys.exc_info()[1], | ||||
| sys.exc_info()[2]) | ||||
| for errItem in err: | ||||
| errStr = errStr + '<br>\n' + str(errItem) | ||||
| errStr = errStr + '\nError occurred at ' + str(time.asctime(time.localtime())) + '<br>\n' | ||||
| admin = madrigal.admin.MadrigalNotify() | ||||
| admin.sendAlert('<html>\n' + errStr + '</html>', | ||||
| 'Error running updateFileTab.py') | ||||
| # end script | ||||
