Changeset 31345 for trunk/ippToPsps/jython/gpc1db.py

Timestamp:

Apr 22, 2011, 1:15:51 PM (15 years ago)

Author:

rhenders

Message:

Now subclass of MySql base class

File:

• trunk/ippToPsps/jython/gpc1db.py (modified) (6 diffs)

trunk/ippToPsps/jython/gpc1db.py

@@ -7 +7 @@
 import logging
 
+from mysql import MySql
 from java.sql import *
-from xml.etree.ElementTree import ElementTree
 
 
@@ -14 +14 @@
 Class for GPC1 database connectivity
 '''
-class Gpc1Db(object):
-
-    driverName="com.mysql.jdbc.Driver"
+class Gpc1Db(MySql):
 
     '''
     Constructor
-
     '''
     def __init__(self, logger):
-
-        # setup logging
-        self.logger = logger
-        self.logger.debug("Gpc1Db constructor")
-
-        # open config
-        doc = ElementTree(file="config.xml")
-
-        # set up JDBC connection
-        dbName = doc.find("gpc1database/name").text
-        dbHost = doc.find("gpc1database/host").text
-        dbUser = doc.find("gpc1database/user").text
-        dbPass = doc.find("gpc1database/password").text
-        self.url = "jdbc:mysql://"+dbHost+"/"+dbName+"?user="+dbUser+"&password="+dbPass
-        self.con = DriverManager.getConnection(self.url)
-        self.stmt = self.con.createStatement()
+        super(Gpc1Db, self).__init__(logger,"gpc1database")
 
     '''
@@ -46 +28 @@
 
         self.logger.debug("Gpc1Db destructor")
-        self.stmt.close()
-        self.con.close()
-
 
     '''
-    Gets a list of sky_ids in this DVO database
+    Gets a list of ids in this DVO database for this stage, could be cam or staticsky (so far)
     '''
-    def getSkyIDsInThisDVODb(self, dvoDb):
+    def getIDsInThisDVODbForThisStage(self, dvoDb, stage):
 
         self.logger.debug("Querying GPC1 for sky_ids in this DVO database: " + dvoDb)
@@ -59 +38 @@
         sql = "SELECT DISTINCT stage_id \
                FROM addRun \
-               WHERE stage = 'staticsky' \
+               WHERE stage = '" + stage + "' \
                AND dvodb = '" + dvoDb + "'"
-
-        print sql
 
         try:
@@ -69 +46 @@
             self.logger.exception("Can't query for sky_ids")
 
-        skyIDs = []
+        ids = []
         while (rs.next()):
-            skyIDs.append(rs.getInt(1))
+            ids.append(rs.getInt(1))
 
         rs.close()
 
-        return skyIDs
+        return ids
 
     '''
@@ -106 +83 @@
 
         return imageIDs
+
+    '''
+    Gets some camera-stage meta data for this cam_id
+    '''
+    def getCameraStageMeta(self, camID):
+
+        self.logger.debug("Querying GPC1 for camera meta data")
+
+        meta = []
+        sql = "SELECT exp_id, exp_name, camRun.dist_group \
+               FROM camRun \
+               JOIN chipRun USING(chip_id) \
+               JOIN rawExp USING(exp_id) WHERE camRun.cam_id = %d" % camID
+
+        try:
+            rs = self.stmt.executeQuery(sql)
+            rs.first()
+        except:
+            self.logger.exception("Can't query for camera meta")
+        finally:
+            meta.append(rs.getInt(1))
+            meta.append(rs.getString(2))
+            meta.append(rs.getString(3))
+
+        return meta
+
+    '''
+    Gets a camera-stage smf for this cam_id
+    '''
+    def getCameraStageSmf(self, camID):
+
+        self.logger.debug("Querying GPC1 for camera smf files")
+
+        sql = "SELECT path_base \
+               FROM camProcessedExp \
+               JOIN camRun USING(cam_id) \
+               WHERE camRun.cam_id = %d" % camID
+
+        try:
+            rs = self.stmt.executeQuery(sql)
+            rs.first()
+        except:
+            self.logger.exception("Can't query for camera smfs")
+
+        # get path to base dir of cmf files
+        path = rs.getString(1)
+        path = path[0:path.rfind("/")]
+       
+        # list all cmf files if a neb path
+        files = []
+        if path.startswith("neb"):
+
+            f=os.popen("neb-ls -p "+path+"/%smf")
+            print "neb-ls -p "+path+"/%smf"
+            for i in f.readlines():
+                files.append(i.rstrip())
+
+        # or not a neb path
+        else:
+            files = glob.glob(path + "/*.cmf")
+
+        return files[0] # TODO just returning first file - check