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

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

Parent Directory | Revision Log | View Patch Patch

-trunk/thuban/Thuban/Model/session.py
revision 982 by bh,
Thu May 22 12:02:15 2003 UTC
+branches/WIP-pyshapelib-bramz/Thuban/Model/session.py
revision 2734 by bramz,
Thu Mar  1 12:42:59 2007 UTC
 Line 1
- # Copyright (c) 2001, 2002, 2003 by Intevation GmbH
+ # Copyright (c) 2001, 2002, 2003, 2004 by Intevation GmbH
  # Authors:
  # Bernhard Herzog <[email protected]>
  # Jan-Oliver Wagner <[email protected]>
 Line 15 
 import weakref
  from messages import MAPS_CHANGED, EXTENSIONS_CHANGED, FILENAME_CHANGED, \
       MAP_LAYERS_CHANGED, MAP_PROJECTION_CHANGED, \
       LAYER_CHANGED, LAYER_PROJECTION_CHANGED, LAYER_VISIBILITY_CHANGED,\
-      EXTENSION_CHANGED, EXTENSION_OBJECTS_CHANGED, CHANGED
+      EXTENSION_CHANGED, EXTENSION_OBJECTS_CHANGED, CHANGED, \
+      TABLE_REMOVED, DBCONN_ADDED, DBCONN_REMOVED
  from Thuban import _
  from base import TitledObject, Modifiable
  from map import Map
  from data import ShapefileStore
+ from table import DBFTable
+ import postgisdb
  from transientdb import TransientDatabase, AutoTransientTable
-Line 131 
 class Session(TitledObject, Modifiable):
+Line 134 
 class Session(TitledObject, Modifiable):
          self.tables = []
          self.shapestores = []
          self.extensions = []
+         self.db_connections = []
          self.temp_dir = None
          self.transient_db = None
-Line 199 
 class Session(TitledObject, Modifiable):
+Line 203 
 class Session(TitledObject, Modifiable):
          The list includes all tables that are indirectly opened through
          shape stores and the tables that have been opened explicitly.
          """
-         return self.tables + [store.Table() for store in self.ShapeStores()]
+         tables = self.tables[:]
+         ids = {}
+         for t in tables:
+             ids[id(t)] = 1
+         for store in self.ShapeStores():
+             t = store.Table()
+             if id(t) not in ids:
+                 ids[id(t)] = 1
+                 tables.append(t)
+         return tables
+     def UnreferencedTables(self):
+         """Return the tables that are not referenced by other data sources"""
+         known = {}
+         for table in self.tables:
+             known[id(table)] = 0
+         for table in self.tables + self.ShapeStores():
+             for dep in table.Dependencies():
+                 known[id(dep)] = 1
+         return [table for table in self.tables if known[id(table)] == 0]
      def AddTable(self, table):
          """Add the table to the session
-Line 225 
 class Session(TitledObject, Modifiable):
+Line 248 
 class Session(TitledObject, Modifiable):
          else:
              transient_table = table
          self.tables.append(transient_table)
+         self.changed()
          return transient_table
+     def RemoveTable(self, table):
+         """Remove the table from the session.
+         The table object must be a table object previously returned by
+         the AddTable method. If the table is not part of the session
+         raise a ValueError.
+         Issue a TABLE_REMOVED message after the table has been removed.
+         The message has the removed table as the single parameter.
+         """
+         tables = [t for t in self.tables if t is not table]
+         if len(tables) == len(self.tables):
+             raise ValueError
+         self.tables = tables
+         self.changed(TABLE_REMOVED, table)
+     def DataContainers(self):
+         """Return all data containers, i.e. shapestores and tables"""
+         return self.tables + self.ShapeStores()
+     def OpenTableFile(self, filename):
+         """Open the table file filename and return the table object.
+         The filename argument must be the name of a DBF file.
+         """
+         return self.AddTable(DBFTable(filename))
      def temp_directory(self):
          """
          Return the name of the directory for session specific temporary files
-Line 246 
 class Session(TitledObject, Modifiable):
+Line 297 
 class Session(TitledObject, Modifiable):
          self._add_shapestore(store)
          return store
+     def AddShapeStore(self, shapestore):
+         """Add the shapestore to the session.
+         The session only holds a weak reference to the shapestore, so it
+         will automatically be removed from the session when the last
+         reference goes away.
+         """
+         self._add_shapestore(shapestore)
+         return shapestore
      def TransientDB(self):
          if self.transient_db is None:
              filename = os.path.join(self.temp_directory(), "transientdb")
-Line 255 
 class Session(TitledObject, Modifiable):
+Line 316 
 class Session(TitledObject, Modifiable):
                                                          self.temp_dir_remover)
          return self.transient_db
+     def AddDBConnection(self, dbconn):
+         """Add the database connection dbconn to the session
+         The argument should be an instance of PostGISConnection.
+         """
+         self.db_connections.append(dbconn)
+         self.changed(DBCONN_ADDED)
+     def DBConnections(self):
+         """
+         Return a list of all database connections registered with the session
+         """
+         return self.db_connections
+     def HasDBConnections(self):
+         """Return whether the session has open database connections"""
+         return bool(self.db_connections)
+     def CanRemoveDBConnection(self, dbconn):
+         """Return whether the database connections dbconn can be removed
+         If can be removed if none of the shapestores or tables in the
+         session references it.
+         """
+         for store in self.ShapeStores():
+             if (isinstance(store, postgisdb.PostGISShapeStore)
+                 and store.db is dbconn):
+                 return 0
+         for table in self.Tables():
+             if (isinstance(table, postgisdb.PostGISTable)
+                 and table.db is dbconn):
+                 return 0
+         return 1
+     def RemoveDBConnection(self, dbconn):
+         """Remove the database connection from the session
+         The parameter must be a connection that was registered
+         previously by a AddDBConnection() call.
+         """
+         if self.CanRemoveDBConnection(dbconn):
+             remaining = [c for c in self.db_connections if c is not dbconn]
+             if len(remaining) < len(self.db_connections):
+                 self.db_connections = remaining
+                 self.changed(DBCONN_REMOVED)
+             else:
+                 raise ValueError("DBConection %r is not registered"
+                                  " with session %r" % (dbconn, self))
+         else:
+             raise ValueError("DBConnection %r is still in use" % (dbconn,))
+     def OpenDBShapeStore(self, db, tablename, id_column = None,
+                          geometry_column = None):
+         """Create and return a shapstore for a table in the database
+         The db parameter must be a database connection previously passed
+         to AddDBConnection().
+         """
+         store = postgisdb.PostGISShapeStore(db, tablename,
+                                             id_column = id_column,
+                                             geometry_column = geometry_column)
+         self._add_shapestore(store)
+         return store
      def Destroy(self):
          for map in self.maps:
              map.Destroy()

 Legend:



Removed from v.982
 


changed lines


 
Added in v.2734
 Legend:



Removed from v.982
 


changed lines


 
Added in v.2734
-Removed from v.982
+Added in v.2734

[email protected]	ViewVC Help
Powered by ViewVC 1.1.26