1 |
# Copyright (c) 2001, 2002, 2003 by Intevation GmbH |
# Copyright (c) 2001, 2002, 2003, 2004 by Intevation GmbH |
2 |
# Authors: |
# Authors: |
3 |
# Bernhard Herzog <[email protected]> |
# Bernhard Herzog <[email protected]> |
4 |
# Jan-Oliver Wagner <[email protected]> |
# Jan-Oliver Wagner <[email protected]> |
16 |
MAP_LAYERS_CHANGED, MAP_PROJECTION_CHANGED, \ |
MAP_LAYERS_CHANGED, MAP_PROJECTION_CHANGED, \ |
17 |
LAYER_CHANGED, LAYER_PROJECTION_CHANGED, LAYER_VISIBILITY_CHANGED,\ |
LAYER_CHANGED, LAYER_PROJECTION_CHANGED, LAYER_VISIBILITY_CHANGED,\ |
18 |
EXTENSION_CHANGED, EXTENSION_OBJECTS_CHANGED, CHANGED, \ |
EXTENSION_CHANGED, EXTENSION_OBJECTS_CHANGED, CHANGED, \ |
19 |
TABLE_REMOVED |
TABLE_REMOVED, DBCONN_ADDED, DBCONN_REMOVED |
20 |
|
|
21 |
from Thuban import _ |
from Thuban import _ |
22 |
|
|
24 |
from map import Map |
from map import Map |
25 |
from data import ShapefileStore |
from data import ShapefileStore |
26 |
from table import DBFTable |
from table import DBFTable |
27 |
|
import postgisdb |
28 |
|
|
29 |
from transientdb import TransientDatabase, AutoTransientTable |
from transientdb import TransientDatabase, AutoTransientTable |
30 |
|
|
134 |
self.tables = [] |
self.tables = [] |
135 |
self.shapestores = [] |
self.shapestores = [] |
136 |
self.extensions = [] |
self.extensions = [] |
137 |
|
self.db_connections = [] |
138 |
self.temp_dir = None |
self.temp_dir = None |
139 |
self.transient_db = None |
self.transient_db = None |
140 |
|
|
316 |
self.temp_dir_remover) |
self.temp_dir_remover) |
317 |
return self.transient_db |
return self.transient_db |
318 |
|
|
319 |
|
def AddDBConnection(self, dbconn): |
320 |
|
"""Add the database connection dbconn to the session |
321 |
|
|
322 |
|
The argument should be an instance of PostGISConnection. |
323 |
|
""" |
324 |
|
self.db_connections.append(dbconn) |
325 |
|
self.changed(DBCONN_ADDED) |
326 |
|
|
327 |
|
def DBConnections(self): |
328 |
|
""" |
329 |
|
Return a list of all database connections registered with the session |
330 |
|
""" |
331 |
|
return self.db_connections |
332 |
|
|
333 |
|
def HasDBConnections(self): |
334 |
|
"""Return whether the session has open database connections""" |
335 |
|
return bool(self.db_connections) |
336 |
|
|
337 |
|
def CanRemoveDBConnection(self, dbconn): |
338 |
|
"""Return whether the database connections dbconn can be removed |
339 |
|
|
340 |
|
If can be removed if none of the shapestores or tables in the |
341 |
|
session references it. |
342 |
|
""" |
343 |
|
for store in self.ShapeStores(): |
344 |
|
if (isinstance(store, postgisdb.PostGISShapeStore) |
345 |
|
and store.db is dbconn): |
346 |
|
return 0 |
347 |
|
for table in self.Tables(): |
348 |
|
if (isinstance(table, postgisdb.PostGISTable) |
349 |
|
and table.db is dbconn): |
350 |
|
return 0 |
351 |
|
return 1 |
352 |
|
|
353 |
|
def RemoveDBConnection(self, dbconn): |
354 |
|
"""Remove the database connection from the session |
355 |
|
|
356 |
|
The parameter must be a connection that was registered |
357 |
|
previously by a AddDBConnection() call. |
358 |
|
""" |
359 |
|
if self.CanRemoveDBConnection(dbconn): |
360 |
|
remaining = [c for c in self.db_connections if c is not dbconn] |
361 |
|
if len(remaining) < len(self.db_connections): |
362 |
|
self.db_connections = remaining |
363 |
|
self.changed(DBCONN_REMOVED) |
364 |
|
else: |
365 |
|
raise ValueError("DBConection %r is not registered" |
366 |
|
" with session %r" % (dbconn, self)) |
367 |
|
else: |
368 |
|
raise ValueError("DBConnection %r is still in use" % (dbconn,)) |
369 |
|
|
370 |
|
def OpenDBShapeStore(self, db, tablename, id_column = None, |
371 |
|
geometry_column = None): |
372 |
|
"""Create and return a shapstore for a table in the database |
373 |
|
|
374 |
|
The db parameter must be a database connection previously passed |
375 |
|
to AddDBConnection(). |
376 |
|
""" |
377 |
|
store = postgisdb.PostGISShapeStore(db, tablename, |
378 |
|
id_column = id_column, |
379 |
|
geometry_column = geometry_column) |
380 |
|
self._add_shapestore(store) |
381 |
|
return store |
382 |
|
|
383 |
def Destroy(self): |
def Destroy(self): |
384 |
for map in self.maps: |
for map in self.maps: |
385 |
map.Destroy() |
map.Destroy() |