checkGeoUpdate.py
142 lines
| 4.0 KiB
| text/x-python
|
PythonLexer
| r0 | #!PYTHONEXE | |||
| import sys | ||||
| import os, os.path | ||||
| import traceback | ||||
| import urllib.request, urllib.error, urllib.parse | ||||
| # 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: | ||||
| # path to OpenMadrigal download site | ||||
| openMadUrl = 'http://madrigal.haystack.mit.edu/madrigal/distributionFiles/' | ||||
| openMad3Url = 'http://madrigal.haystack.mit.edu/madrigal/distributionFiles/mad3/' | ||||
| # create MadrigalDB obj | ||||
| madDBObj = madrigal.metadata.MadrigalDB() | ||||
| # get ig_rz.dat | ||||
| f = urllib.request.urlopen(openMadUrl + 'ig_rz.dat') | ||||
| f2 = open(os.path.join(madDBObj.getMadroot(), 'bin/ig_rz.dat'), 'w') | ||||
| f2.write(f.read().decode('utf-8')) | ||||
| f.close() | ||||
| f2.close() | ||||
| # default is we need to update | ||||
| needUpdate = 1 | ||||
| # try to open geoStatus.file and compare to open madrigal version | ||||
| try: | ||||
| localStatusFile = open(madDBObj.getMadroot() + '/geoStatus.dat') | ||||
| localLines = localStatusFile.readlines() | ||||
| localStatusFile.close() | ||||
| if len(localLines) == 0: | ||||
| raise ValueError('') | ||||
| except: | ||||
| # no such file exists yet | ||||
| localLines = ('','','','','') | ||||
| openMadFile = urllib.request.urlopen(openMad3Url + 'status.dat') | ||||
| openMadText = openMadFile.read().decode('utf-8') | ||||
| openMadLines = openMadText.split('\n') | ||||
| openMadFile.close() | ||||
| for i in range(5): | ||||
| if localLines[i] != openMadLines[i]: | ||||
| break | ||||
| if i == 4: | ||||
| # no update needed - exit | ||||
| sys.exit(0) | ||||
| # write new geoStatus.dat | ||||
| print('downloading latest geo files from OpenMadrigal') | ||||
| localStatusFile = open(madDBObj.getMadroot() + '/geoStatus.dat', 'w') | ||||
| for line in openMadLines: | ||||
| localStatusFile.write(line) | ||||
| localStatusFile.close() | ||||
| try: | ||||
| os.chmod(madDBObj.getMadroot() + '/geoStatus.dat', 0o666) | ||||
| except: | ||||
| pass | ||||
| # get the updated files | ||||
| f = urllib.request.urlopen('http://madrigal.haystack.mit.edu/madrigal/distributionFiles/mad3/geofil.tar.gz') | ||||
| f2 = open(madDBObj.getMadroot() + '/geofil.tar.gz', 'wb') | ||||
| f2.write(f.read()) | ||||
| f.close() | ||||
| f2.close() | ||||
| os.chdir(madDBObj.getMadroot()) | ||||
| os.system('gunzip -f ' + madDBObj.getMadroot() + '/geofil.tar.gz') | ||||
| os.system('tar -xf ' + madDBObj.getMadroot() + '/geofil.tar') | ||||
| os.system(madDBObj.getMadroot() + '/configureExperiments') | ||||
| os.remove(madDBObj.getMadroot() + '/geofil.tar') | ||||
| except SystemExit: | ||||
| # nothing to be done | ||||
| pass | ||||
| except madrigal.admin.MadrigalError as e: | ||||
| # handle a MadrigalError | ||||
| errStr = 'Error occurred in checkGeoUpdate.py\n' | ||||
| 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 + '\n' + str(errItem) | ||||
| print(errStr) | ||||
| 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 = 'Error occurred in running checkGeoUpdate.py.\n' | ||||
| err = traceback.format_exception(sys.exc_info()[0], | ||||
| sys.exc_info()[1], | ||||
| sys.exc_info()[2]) | ||||
| for errItem in err: | ||||
| errStr = errStr + '\n' + str(errItem) | ||||
| print(errStr) | ||||
| # end script | ||||
