]> rtime.felk.cvut.cz Git - linux-conf-perf.git/blobdiff - scripts/database.py
kconfig2sat: Fix single-diff problem generation
[linux-conf-perf.git] / scripts / database.py
index 2e96b44d3d4572f08ac8d73a42894422a45f4807..58fdf9ecf0b525bb1d0e898382a6326287f7797a 100644 (file)
@@ -1,3 +1,4 @@
+import os
 import datetime
 import postgresql
 import collections
@@ -5,6 +6,7 @@ import collections
 import utils
 import exceptions
 from conf import conf
+from conf import sf
 
 def __git_describe__():
        return utils.callsubprocess('git_describe',
@@ -17,18 +19,15 @@ def __git_commit__():
 def __timestamp__():
        return datetime.datetime.now().strftime('%y-%m-%d-%H-%M-%S')
 
-Config = collections.namedtuple('Config', 'id hash cfile') # Named tuple for configuration
-Measure = collections.namedtuple('Measure', 'id conf_id mfile value') # Named tuple for measurement
+Config = collections.namedtuple('Config', 'id hash config') # Named tuple for configuration
+Measure = collections.namedtuple('Measure', 'id conf_id output value') # Named tuple for measurement
 
 class database:
        "Class used for accessing PostgreSQL project database."
        def __init__(self):
                self.db = postgresql.open(database = conf.db_database,
-                               user = conf.db_user,
-                               password = conf.db_password,
-                               host = conf.db_host,
-                               port = conf.db_port
-                               )
+                                                                 unix='/var/run/postgresql/.s.PGSQL.5432',
+                                                                 password='')
                # check if tables are present
                tables = ('toolsgit', 'configurations', 'measure')
                for tab in tables:
@@ -55,36 +54,60 @@ class database:
                ps(ds, cm)
                return self.check_toolsgit()
 
-       def add_configuration(self, hash, cfile):
+       def check_linuxgit(self):
+               "Return id of linuxgit row. If missing, it is inserted."
+               wd = os.getcwd()
+               os.chdir(sf(conf.linux_sources))
+               ds = __git_describe__()
+               cm = __git_commit__()
+               os.chdir(wd)
+               ps = self.db.prepare("""SELECT id FROM linuxgit
+                                                         WHERE git_describe = $1 AND git_commit = $2
+                                                         """)
+               id = ps(ds, cm)
+               if id:
+                       return id[0][0]
+               ps = self.db.prepare("""INSERT INTO linuxgit
+                                                  (git_describe, git_commit)
+                                                  VALUES
+                                                  ($1, $2);
+                                                  """)
+               ps(ds, cm)
+               return self.check_linuxgit()
+
+       def add_configuration(self, hash, txtconfig, generator):
                "Add configuration to database."
                ps = self.db.prepare("""INSERT INTO configurations
-                                                               (hash, cfile, gtime, toolgit)
+                                                               (hash, config, gtime, toolgit, linuxgit, generator)
                                                                VALUES
-                                                               ($1, $2, $3, $4);
+                                                               ($1, $2, $3, $4, $5, $6);
                                                                """)
                gt = self.check_toolsgit()
+               lgt = self.check_linuxgit()
                tm = datetime.datetime.now()
-               ps(hash, cfile, tm, gt)
+               ps(hash, '\n'.join(txtconfig), tm, gt, lgt, generator)
 
        def get_configration(self, hash):
                "Return configration id for inserted hash."
-               ps = self.db.prepare("""SELECT id, cfile FROM configurations
+               ps = self.db.prepare("""SELECT id, config FROM configurations
                                                                WHERE hash = $1""")
                rtn = []
                for dt in ps(hash):
-                       rtn.append(Config(dt[0], hash, dt[1]))
+                       rtn.append(Config(dt[0], hash, dt[1].split('\n')))
                return rtn
 
-       def add_measure(self, mfile, conf_id, value = None):
+       def add_measure(self, output, result, conf_id, value = None):
                "Add measurement."
                ps = self.db.prepare("""INSERT INTO measure
-                                                               (conf, mfile, value, mtime, toolgit)
+                                                               (conf, output, value, mtime, toolgit,
+                                                               linuxgit, measurement, result)
                                                                VALUES
-                                                               ($1, $2, $3, $4, $5);
+                                                               ($1, $2, $3, $4, $5, $6, $7, $8);
                                                                """)
                gt = self.check_toolsgit()
+               lgt = self.check_linuxgit()
                tm = datetime.datetime.now()
-               ps(conf_id, mfile, value, tm, gt)
+               ps(conf_id, output, value, tm, gt, lgt, conf.measure_identifier, result)
 
        def update_measure(self, measure_id, value):
                "Update measured value"
@@ -97,7 +120,7 @@ class database:
 
        def get_measures(self, conf_id):
                "Get measures for configuration with conf_id id"
-               ps = self.db.prepare("""SELECT id, mfile, value FROM measure
+               ps = self.db.prepare("""SELECT id, output, value FROM measure
                                                                WHERE conf = $1;
                                                                """)
                rtn = []
@@ -107,11 +130,32 @@ class database:
 
        def get_unmeasured(self):
                "Returns list of all unmeasured configurations."
-               ps = self.db.prepare("""SELECT c.id, c.hash, c.cfile
-                                                               FROM configurations AS c, measure AS m
-                                                               WHERE c.id NOT IN m.conf;
+               # FIXME: Take into account case when we want the same
+               # configuration for either different experiment or kernel
+               # version or target.
+               ps = self.db.prepare("""SELECT id, hash, config FROM configurations
+                                                               WHERE id NOT IN
+                                                               (SELECT conf FROM measure)
                                                                """)
                rtn = []
                for dt in ps():
-                       rtn.append(Config(dt[0], dt[1], dt[2]))
+                       rtn.append(Config(dt[0], dt[1], dt[2].split('\n')))
+               return rtn
+
+       def add_configsort(self, configopt):
+               "Add configuration option to sorted list"
+               ps = self.db.prepare("""INSERT INTO configopt
+                                                               (configopt) VALUES ($1)
+                                                               """)
+               ps(configopt)
+
+       def get_configsort(self):
+               "Returns sorted list of all configuration options"
+               ps = self.db.prepare("""SELECT id, configopt FROM configopt
+                                                               ORDER BY id ASC
+                                                               """)
+               rtn = []
+               itms = ps()
+               for id, config in itms:
+                       rtn.append(config)
                return rtn