Changeset 34041 for branches/meh_branches/ppstack_test/ippToPsps/jython/setupScratchDb.py

Timestamp:

Jun 19, 2012, 5:24:19 PM (14 years ago)

Author:

mhuber

Message:

merging latest r34040 trunk changes to branch

Location:

branches/meh_branches/ppstack_test

Files:

• . (modified) (1 prop)
• ippToPsps/jython/setupScratchDb.py (modified) (1 diff)

branches/meh_branches/ppstack_test/ippToPsps/jython/setupScratchDb.py

@@ -1 +1 @@
 #!/usr/bin/env jython
 
-import os.path
 import sys
-import time
 import stilts
 from java.lang import *
 from java.sql import *
-import logging.config
 
-from java.lang import Math
-from org.apache.commons.math.special import Erf
-from xml.etree.ElementTree import ElementTree, Element, tostring
 
-from pslogger import PSLogger
+from ipptopsps import IppToPsps
 from scratchdb import ScratchDb
 
 '''
-Start of program.
+SetupScratchDb class
 '''
+class SetupScratchDb(IppToPsps):
 
-if len(sys.argv) > 1: CONFIG = sys.argv[1]
-else:
-    print "** Usage: " + sys.argv[0] + " <configPath> [init]"
-    sys.exit(1)
+    '''
+    Constructor
+    '''
+    def __init__(self, argv):
+        super(SetupScratchDb, self).__init__(argv)
 
-# open config file
-configDoc = ElementTree(file=CONFIG)
+        if len(argv) < 3:
+            self.printUsage()
+            self.exitProgram("incorrect args")
 
-# set up logging
-logging.setLoggerClass(PSLogger)
-logger = logging.getLogger("setupScratchDb")
-logger.setup(configDoc, "setupScratchDb")
+        try:
+            self.scratchDb = ScratchDb(self.logger, self.config, argv[2])
+        except:
+            self.exitProgram("Could not connect to a scratch Db")
+            raise
 
-scratchDb = ScratchDb(logger, configDoc)
 
-# install stored procedure
-logger.info("Installing psfLikelihood() stored procedure")
-scratchDb.createPsfLikelihoodFunction()
+    def run(self):
 
-# install IN tables
-logger.info("Installing initialization tables")
-tables = stilts.treads("../config/IN/tables.vot")
-for table in tables:
-    logger.debug("Creating IN table: " + table.name)
-    table.write(scratchDb.url + '#' + table.name)
+        self.logger.infoSeparator()
 
-logger.info("Done")
+        # install stored procedure
+        self.logger.infoPair("Installing",  "psfLikelihood() stored procedure")
+        self.scratchDb.createPsfLikelihoodFunction()
 
+        # install IN tables
+        self.logger.infoPair("Installing", "initialization tables")
+        tables = stilts.treads("../config/IN/tables.vot")
+        for table in tables:
+            self.logger.debug("Creating IN table: " + table.name)
+            table.write(self.scratchDb.url + '#' + table.name)
+
+    '''
+    Overrides base-class version
+    '''
+    def printUsage(self):
+        super(SetupScratchDb, self).printUsage("<scratchdb_name>")
+
+'''
+Program starts here
+'''
+try:
+    setupScratchDb = SetupScratchDb(sys.argv)
+    setupScratchDb.run()
+    setupScratchDb.exitProgram("completed")
+except: pass
+
+