/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/Model/transientdb.py

Diff of /branches/WIP-pyshapelib-bramz/Thuban/Model/transientdb.py

Parent Directory | Revision Log | View Patch Patch

-revision 839 by bh,
Tue May  6 15:54:18 2003 UTC
+revision 1662 by bh,
Wed Aug 27 13:51:01 2003 UTC
 Line 18 
 __version__ = "$Revision$"
  # $Source$
  # $Id$
- import os
- import weakref
  from sqlite import connect
+ from base import TitledObject
  import table
  sql_type_map = {
 Line 118 
 class TransientTableBase(table.OldTableI
          # Build the CREATE TABLE statement and create the table in the
          # database
-         table_types = []
+         table_types = ["id INTEGER PRIMARY KEY"]
          for col in self.columns:
              table_types.append("%s %s" % (col.internal_name,
                                            sql_type_map[col.type]))
 Line 160 
 class TransientTableBase(table.OldTableI
          """
          return self.column_map.has_key(col)
-     def ReadRowAsDict(self, index):
+     def RowIdToOrdinal(self, gid):
+         """Return the row ordinal given its id
+         At the moment the transient tables are only used for tables that
+         don't distinguish between row number and row id, so the value is
+         returned unchanged.
+         """
+         return gid
+     def RowOrdinalToId(self, num):
+         """Return the rowid for given its ordinal
+         At the moment the transient tables are only used for tables that
+         don't distinguish between row number and row id, so the value is
+         returned unchanged.
+         """
+         return num
+     def ReadRowAsDict(self, index, row_is_ordinal = 0):
+         """Return the entire row as a dictionary with column names as keys
+         The row_is_ordinal is ignored because at the moment the
+         transient tables are only used for DBF files where it doesn't
+         matter.
+         """
+         # Implementation Strategy: Executing a completely new select
+         # statement every time this method is called is too slow. The
+         # most important usage is to read the records more or less
+         # sequentially. This happens e.g. when drawing a layer with a
+         # classification where the shapes are drawn in order of the
+         # shape ids. Another pattern is that the same row is requested
+         # several times in a row. This happens in the table view, for
+         # instance.
+         # We can exploit this to make access faster by having one cursor
+         # open all the time and keeping the last row read around in case
+         # the same row is accessed again the next time and if the row
+         # index is larger than the row we have read last we simply fetch
+         # rows from the cursor until we've reached the requested row. If
+         # the requested row index is smaller then we start a new cursor.
+         # FIXME: So far this scheme seems to work well enough. Obvious
+         # improvements would be to start the cursor at exactly the
+         # requested row (should be efficient and easy to do now that the
+         # id is the primary key) and to perhaps to also start a new
+         # cursor if the requested index is much larger than the last row
+         # so that we don't read and discard lots of the rows.
+         # Check whether we have to start a new cursor
          if self.read_record_cursor is None or index <self.read_record_last_row:
              stmt = ("SELECT %s FROM %s;"
                      % (", ".join([c.internal_name for c in self.columns]),
-Line 184 
 class TransientTableBase(table.OldTableI
+Line 232 
 class TransientTableBase(table.OldTableI
          self.read_record_last_row = index
          return dict(zip(self.orig_names, result))
+     def ReadValue(self, row, col, row_is_ordinal = 0):
+         """Return the value of the specified row and column
+         The col parameter may be the index of the column or its name.
+         The row_is_ordinal is ignored because at the moment the
+         transient tables are only used for DBF files where it doesn't
+         matter.
+         """
+         # Depending on the actual access patterns of the table data, it
+         # might be a bit faster in some circumstances to not implement
+         # this via ReadRowAsDict, but this simple implementation should
+         # be fast enough for most purposes.
+         return self.ReadRowAsDict(row)[self.column_map[col].name]
      def ValueRange(self, col):
          col = self.column_map[col]
          iname = col.internal_name
-Line 205 
 class TransientTableBase(table.OldTableI
+Line 268 
 class TransientTableBase(table.OldTableI
              result.append(row[0])
          return result
+     def Width(self, col):
+         """Return the maximum width of values in the column
+         The return value is the the maximum length of string
+         representation of the values in the column (represented by index
+         or name).
+         """
+         max = 0
+         type  = self.column_map[col].type
+         iname = self.column_map[col].internal_name
+         cursor = self.db.cursor()
+         cursor.execute("SELECT %s FROM %s;" % (iname, self.tablename))
+         values = [ i[0] for i in cursor.fetchall()]
+         if not values:
+             return None
+         if type == table.FIELDTYPE_DOUBLE:
+             format = "%.12f"
+         elif type == table.FIELDTYPE_INT:
+             format = "%d"
+         else:
+             format = "%s"
+         for value in values:
+             if value is None: continue
+             l = len(format % value)
+             if l > max:
+                 max = l
+         return max
+     def SimpleQuery(self, left, comparison, right):
+         """Return the indices of all rows that matching a condition.
+         Parameters:
+            left -- The column object for the left side of the comparison
+            comparison -- The comparison operator as a string. It must be
+                          one of '==', '!=', '<', '<=', '>=', '>'
+            right -- The right hand side of the comparison. It must be
+                     either a column object or a value, i.e. a string,
+                     int or float.
+         The return value is a sorted list of the indices of the rows
+         where the condition is true.
+         """
+         if comparison not in ("==", "!=", "<", "<=", ">=", ">"):
+             raise ValueError("Comparison operator %r not allowed" % comparison)
+         if hasattr(right, "internal_name"):
+             right_template = right.internal_name
+             params = ()
+         else:
+             right_template = "%s"
+             params = (right,)
+         query = "SELECT id FROM %s WHERE %s %s %s ORDER BY id;" \
+                 % (self.tablename, left.internal_name, comparison,
+                    right_template)
+         cursor = self.db.cursor()
+         cursor.execute(query, params)
+         result = []
+         while 1:
+             row = cursor.fetchone()
+             if row is None:
+                 break
+             result.append(row[0])
+         return result
+     def Dependencies(self):
+         """Placeholder for a method in a derived class.
+         Return a sequence with the tables and other data objects that
+         self depends on.
+         """
+         raise NotImplementedError
- class TransientTable(TransientTableBase):
+ class TransientTable(TitledObject, TransientTableBase):
      """A Table in a transient DB that starts as the copy of a Thuban Table."""
-Line 217 
 class TransientTable(TransientTableBase)
+Line 359 
 class TransientTable(TransientTableBase)
          interface.
          """
          TransientTableBase.__init__(self, transient_db)
+         TitledObject.__init__(self, table.Title())
          self.create(table)
      def create(self, table):
-Line 227 
 class TransientTable(TransientTableBase)
+Line 370 
 class TransientTable(TransientTableBase)
          TransientTableBase.create(self, columns)
          # copy the input table to the transient db
-         insert_template = "INSERT INTO %s (%s) VALUES (%s);" \
+         # A key to insert to use for the formatting of the insert
+         # statement. The key must not be equal to any of the column
+         # names so we construct one by building a string of x's that is
+         # longer than any of the column names
+         id_key = max([len(col.name) for col in self.columns]) * "x"
+         insert_template = "INSERT INTO %s (id, %s) VALUES (%%(%s)s, %s);" \
                                 % (self.tablename,
                                    ", ".join([col.internal_name
                                               for col in self.columns]),
+                                   id_key,
                                    ", ".join(["%%(%s)s" % col.name
                                               for col in self.columns]))
          cursor = self.db.cursor()
          for i in range(table.NumRows()):
-             cursor.execute(insert_template, table.ReadRowAsDict(i))
+             row = table.ReadRowAsDict(i)
+             row[id_key] = i
+             cursor.execute(insert_template, row)
          self.db.conn.commit()
- class TransientJoinedTable(TransientTableBase):
+ class TransientJoinedTable(TitledObject, TransientTableBase):
      """A Table in the transient DB that contains a join of two tables"""
      def __init__(self, transient_db, left_table, left_field,
-                  right_table, right_field = None):
+                  right_table, right_field = None, outer_join = False):
          """Create a new table in the transient DB as a join of two tables.
          Both input tables, left_table and right_table must have a
          transient_table method that returns a table object for a table
-         in the trnsient database. The join is performed on the condition
+         in the transient database. The join is performed on the condition
          that the value of the left_field column the the left table is
          equal to the value of the right_field in the right_table.
-         The joined table contains all columns of the input tables with
+         The joined table contains all columns of the input tables,
-         one exception: Any column in the right_table with the same name
+         however, the column names of the right table may be changed
-         as one of the columns in the left_table will be omitted. This is
+         slightly to make them unique in the joined table. This is
-         somewhat of an implementation detail, but is done so that the
+         currently done by appending a sufficient number of underscores
-         column names of the joined table can be the same as the column
+         ('_').
-         names of the input tables without having to create prefixes.
          """
          TransientTableBase.__init__(self, transient_db)
+         self.dependencies = (left_table, right_table)
          self.left_table = left_table.transient_table()
          self.left_field = left_field
          self.right_table = right_table.transient_table()
-Line 269 
 class TransientJoinedTable(TransientTabl
+Line 422 
 class TransientJoinedTable(TransientTabl
              self.right_field = right_field
          else:
              self.right_field = self.left_field
+         self.outer_join = outer_join
+         title = "Join of %(left)s and %(right)s" \
+                 % {"left": self.left_table.Title(),
+                    "right": self.right_table.Title()}
+         TitledObject.__init__(self, title)
          self.create()
      def create(self):
-Line 277 
 class TransientJoinedTable(TransientTabl
+Line 437 
 class TransientJoinedTable(TransientTabl
          self.right_table.ensure_index(self.right_field)
+         # determine the internal column names to join on before
+         # coalescing the column information because if the external
+         # column names are the same they will be mapped to the same
+         # internal name afterwards.
+         internal_left_col = self.left_table.orig_to_internal[self.left_field]
+         internal_right_col =self.right_table.orig_to_internal[self.right_field]
          # Coalesce the column information
          visited = {}
          columns = []
-         for col in self.left_table.columns + self.right_table.columns:
+         newcolumns = []
-             if col.name in visited:
+         for table in (self.left_table, self.right_table):
-                 continue
+             for col in table.Columns():
-             columns.append(col)
+                 colname = col.name
-         TransientTableBase.create(self, columns)
+                 # We can't allow multiple columns with the same
+                 # original name, so append '_' to this one until
+                 # it is unique.
+                 # FIXME: There should be a better solution.
+                 while colname in visited:
+                     colname = colname + '_'
+                 columns.append((table.tablename, col))
+                 newcol = ColumnReference(colname, col.type,
+                                             "Col%03d" % (len(newcolumns)+1))
+                 newcolumns.append(newcol)
+                 visited[colname] = 1
+         TransientTableBase.create(self, newcolumns)
          # Copy the joined data to the table.
-         internal_names = [col.internal_name for col in self.columns]
+         newinternal_names = [col.internal_name for col in self.columns]
-         stmt = "INSERT INTO %s (%s) SELECT %s FROM %s JOIN %s ON %s = %s;" \
+         internal_references = ["%s.%s" % (table, col.internal_name)
-                % (self.tablename,
+                                                     for table, col in columns]
-                   ", ".join(internal_names),
+         if self.outer_join:
-                   ", ".join(internal_names),
+             join_operator = 'LEFT OUTER JOIN'
-                   self.left_table.tablename,
+         else:
-                   self.right_table.tablename,
+             join_operator = 'JOIN'
-                   self.orig_to_internal[self.left_field],
+         stmt = ("INSERT INTO %s (id, %s) SELECT %s.id, %s FROM %s"
-                   self.orig_to_internal[self.right_field])
+                 " %s %s ON %s.%s = %s.%s;"
+                 % (self.tablename,
+                    ", ".join(newinternal_names),
+                    self.left_table.tablename,
+                    ", ".join(internal_references),
+                    self.left_table.tablename,
+                    join_operator,
+                    self.right_table.tablename,
+                    self.left_table.tablename,
+                    internal_left_col,
+                    self.right_table.tablename,
+                    internal_right_col))
          self.db.execute(stmt)
+     def Dependencies(self):
+         """Return a tuple with the two tables the join depends on."""
+         return self.dependencies
+     def JoinType(self):
+         """Return the type of the join (either 'INNER' or 'LEFT OUTER')"""
+         if self.outer_join:
+             return "LEFT OUTER"
+         else:
+             return "INNER"
- class AutoTransientTable(table.OldTableInterfaceMixin):
+ class AutoTransientTable(TitledObject, table.OldTableInterfaceMixin):
      """Table that copies data to a transient table on demand.
-Line 308 
 class AutoTransientTable(table.OldTableI
+Line 508 
 class AutoTransientTable(table.OldTableI
      """
      def __init__(self, transient_db, table):
+         TitledObject.__init__(self, table.Title())
          self.transient_db = transient_db
          self.table = table
          self.t_table = None
-Line 329 
 class AutoTransientTable(table.OldTableI
+Line 530 
 class AutoTransientTable(table.OldTableI
      def NumColumns(self):
          return self.table.NumColumns()
-     def ReadRowAsDict(self, record):
+     def RowIdToOrdinal(self, gid):
+         """Return the row ordinal given its id"""
+         if self.t_table is not None:
+             return self.t_table.RowIdToOrdinal(gid)
+         else:
+             return self.table.RowIdToOrdinal(gid)
+     def RowOrdinalToId(self, num):
+         """Return the rowid for given its ordinal"""
+         if self.t_table is not None:
+             return self.t_table.RowOrdinalToId(num)
+         else:
+             return self.table.RowOrdinalToId(num)
+     def ReadRowAsDict(self, record, row_is_ordinal = 0):
          """Return the record no. record as a dict mapping field names to values
          """
          if self.t_table is not None:
-             return self.t_table.ReadRowAsDict(record)
+             return self.t_table.ReadRowAsDict(record,
+                                               row_is_ordinal = row_is_ordinal)
+         else:
+             return self.table.ReadRowAsDict(record,
+                                             row_is_ordinal = row_is_ordinal)
+     def ReadValue(self, row, col, row_is_ordinal = 0):
+         """Return the value of the specified row and column
+         The col parameter may be the index of the column or its name.
+         """
+         if self.t_table is not None:
+             return self.t_table.ReadValue(row, col,
+                                           row_is_ordinal = row_is_ordinal)
          else:
-             return self.table.ReadRowAsDict(record)
+             return self.table.ReadValue(row, col,
+                                         row_is_ordinal = row_is_ordinal)
      def copy_to_transient(self):
          """Internal: Create a transient table and copy the data into it"""
-Line 358 
 class AutoTransientTable(table.OldTableI
+Line 587 
 class AutoTransientTable(table.OldTableI
          if self.t_table is None:
              self.copy_to_transient()
          return self.t_table.UniqueValues(col)
+     def SimpleQuery(self, left, comparison, right):
+         if self.t_table is None:
+             self.copy_to_transient()
+         # Make sure to use the column object of the transient table. The
+         # left argument is always a column object so we can just ask the
+         # t_table for the right object.
+         if hasattr(right, "name"):
+             return self.t_table.SimpleQuery(self.t_table.Column(left.name),
+                                             comparison,
+                                             self.t_table.Column(right.name))
+         else:
+             return self.t_table.SimpleQuery(self.t_table.Column(left.name),
+                                             comparison, right)
+     def Dependencies(self):
+         """Return a tuple containing the original table"""
+         return (self.table,)
+     def Width(self, col):
+         return self.table.Width(col)

 Legend:



Removed from v.839
 


changed lines


 
Added in v.1662
 Legend:



Removed from v.839
 


changed lines


 
Added in v.1662
-Removed from v.839
+Added in v.1662

[email protected]	ViewVC Help
Powered by ViewVC 1.1.26