/[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

-revision 778 by bh,
Tue Apr 29 14:54:17 2003 UTC
+revision 1282 by bh,
Mon Jun 23 09:47:18 2003 UTC
 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
  from Thuban import _
  from base import TitledObject, Modifiable
  from map import Map
  from data import ShapefileStore
+ from table import DBFTable
- from transientdb import TransientDatabase
+ from transientdb import TransientDatabase, AutoTransientTable
  class AutoRemoveFile:
-Line 129 
 class Session(TitledObject, Modifiable):
+Line 131 
 class Session(TitledObject, Modifiable):
          self.filename = None
          self.maps = []
          self.tables = []
+         self.shapestores = []
          self.extensions = []
          self.temp_dir = None
          self.transient_db = None
-Line 178 
 class Session(TitledObject, Modifiable):
+Line 181 
 class Session(TitledObject, Modifiable):
              extension.Subscribe(channel, self.forward, channel)
          self.changed(EXTENSIONS_CHANGED)
+     def ShapeStores(self):
+         """Return a list of all ShapeStore objects open in the session"""
+         return [store() for store in self.shapestores]
+     def _add_shapestore(self, store):
+         """Internal: Add the shapestore to the list of shapestores"""
+         self.shapestores.append(weakref.ref(store,
+                                             self._clean_weak_store_refs))
+     def _clean_weak_store_refs(self, weakref):
+         """Internal: Remove the weakref from the shapestores list"""
+         self.shapestores = [store for store in self.shapestores
+                                   if store is not weakref]
+     def Tables(self):
+         """Return a list of all table objects open in the session
+         The list includes all tables that are indirectly opened through
+         shape stores and the tables that have been opened explicitly.
+         """
+         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
+         All tables associated with the session that are not implicitly
+         created by the OpenShapefile method (and maybe other Open*
+         methods in the future) have to be passed to this method to make
+         sure the session knows about it. The session keeps a reference
+         to the table. Only tables managed by the session in this way
+         should be used for layers contained in one of the session's
+         maps.
+         The table parameter may be any object implementing the table
+         interface. If it's not already one of the transient tables
+         instantiate an AutoTransientTable with it and use that instead
+         of the original table (note that the AutoTransientTable keeps a
+         reference to the original table).
+         Return the table object actually used by the session.
+         """
+         if not hasattr(table, "transient_table"):
+             transient_table = AutoTransientTable(self.TransientDB(), table)
+         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 193 
 class Session(TitledObject, Modifiable):
+Line 291 
 class Session(TitledObject, Modifiable):
      def OpenShapefile(self, filename):
          """Return a shapefile store object for the data in the given file"""
-         return ShapefileStore(self, filename)
+         store = ShapefileStore(self, filename)
+         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:

 Legend:



Removed from v.778
 


changed lines


 
Added in v.1282
 Legend:



Removed from v.778
 


changed lines


 
Added in v.1282
-Removed from v.778
+Added in v.1282

[email protected]	ViewVC Help
Powered by ViewVC 1.1.26