|
|
#!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
|
|
|
|
|
|
|
|
|
|
