Esempio n. 1
0
    def historize(self):
        """version database"""
        uri = None
        conn_info = ""
        schema = ""
        for layer_id in self.current_layers:
            layer = QgsMapLayerRegistry.instance().mapLayer(layer_id)
            uri = QgsDataSourceURI(layer.source())
            if not conn_info:
                conn_info = uri.connectionInfo()
            else:
                assert conn_info == uri.connectionInfo()
            if not schema:
                schema = uri.schema()
            else:
                assert schema == uri.schema()

        if QMessageBox.Ok != QMessageBox.warning(
            self.iface.mainWindow(),
            "Warning",
            "This will add 4 columns to all tables in schema " + schema + " (i.e. even to tables not in this project)",
            QMessageBox.Ok,
            QMessageBox.Cancel,
        ):
            print "aborted"
            return

        versioning_base.historize(self.pg_conn_info(), schema)

        grp_name = "trunk revision head"
        grp_idx = self.iface.legendInterface().addGroup(grp_name)
        for layer_id in reversed(self.current_layers):
            layer = QgsMapLayerRegistry.instance().mapLayer(layer_id)
            new_uri = QgsDataSourceURI(layer.source())
            new_uri.setDataSource(
                schema + "_trunk_rev_head",
                new_uri.table(),
                new_uri.geometryColumn(),
                new_uri.sql(),
                new_uri.keyColumn(),
            )
            display_name = QgsMapLayerRegistry.instance().mapLayer(layer_id).name()
            src = new_uri.uri().replace("()", "")
            new_layer = self.iface.addVectorLayer(src, display_name, "postgres")
            self.iface.legendInterface().moveLayer(new_layer, grp_idx)
        self.iface.legendInterface().removeGroup(self.current_group_idx)
        self.current_layers = []
Esempio n. 2
0
    def historize(self):
        """version database"""
        uri = None
        conn_info = ''
        schema = ''
        for layer_id in self.current_layers:
            layer = QgsMapLayerRegistry.instance().mapLayer(layer_id)
            uri = QgsDataSourceURI(layer.source())
            if not conn_info:
                conn_info = uri.connectionInfo()
            else:
                assert (conn_info == uri.connectionInfo())
            if not schema:
                schema = uri.schema()
            else:
                assert (schema == uri.schema())

        if QMessageBox.Ok != QMessageBox.warning(
                self.iface.mainWindow(), "Warning",
                "This will add 4 columns to all tables in schema " + schema +
                " (i.e. even to tables not in this project)", QMessageBox.Ok,
                QMessageBox.Cancel):
            print "aborted"
            return

        versioning_base.historize(self.pg_conn_info(), schema)

        grp_name = 'trunk revision head'
        grp_idx = self.iface.legendInterface().addGroup(grp_name)
        for layer_id in reversed(self.current_layers):
            layer = QgsMapLayerRegistry.instance().mapLayer(layer_id)
            new_uri = QgsDataSourceURI(layer.source())
            new_uri.setDataSource(schema + '_trunk_rev_head', new_uri.table(),
                                  new_uri.geometryColumn(), new_uri.sql(),
                                  new_uri.keyColumn())
            display_name = QgsMapLayerRegistry.instance().mapLayer(
                layer_id).name()
            src = new_uri.uri().replace('()', '')
            new_layer = self.iface.addVectorLayer(src, display_name,
                                                  'postgres')
            self.iface.legendInterface().moveLayer(new_layer, grp_idx)
        self.iface.legendInterface().removeGroup(self.current_group_idx)
        self.current_layers = []
        roughness float,
        minor_loss_coefficient float,
        status varchar,
        geometry geometry('LINESTRING',2154)
    )""")

pcur.execute("""
    INSERT INTO epanet.pipes
        (id, start_node, end_node, length, diameter, geometry) 
        VALUES
        ('0','0','1',1,2,ST_GeometryFromText('LINESTRING(1 0,0 1)',2154))""")

pcur.commit()
pcur.close()

versioning_base.historize( pg_conn_info, 'epanet' )

failed = False
try:
    versioning_base.add_branch( pg_conn_info, 'epanet', 'trunk' )
except: 
    failed = True
assert( failed )

failed = False
try:
    versioning_base.add_branch( pg_conn_info, 'epanet', 'mybranch', 'message', 'toto' )
except:
    failed = True
assert( failed )
Esempio n. 4
0
from pyspatialite import dbapi2
import psycopg2
import os
import shutil

test_data_dir = os.path.dirname(os.path.realpath(__file__))
tmp_dir = "/tmp"

# create the test database

os.system("dropdb epanet_test_db")
os.system("createdb epanet_test_db")
os.system("psql epanet_test_db -c 'CREATE EXTENSION postgis'")
os.system("psql epanet_test_db -f "+test_data_dir+"/epanet_test_db_unversioned.sql")

versioning_base.historize("dbname=epanet_test_db","epanet")

# try the update
wc = tmp_dir+"/bug_in_branch_after_commit_wc.sqlite"
if os.path.isfile(wc): os.remove(wc) 
versioning_base.checkout("dbname=epanet_test_db", ['epanet_trunk_rev_head.junctions', 'epanet_trunk_rev_head.pipes'], wc)

scur = versioning_base.Db( dbapi2.connect( wc ) )

scur.execute("SELECT * FROM pipes")
scur.execute("UPDATE pipes_view SET length = 1 WHERE OGC_FID = 1")
scur.commit()

versioning_base.commit(wc,'test', "dbname=epanet_test_db" )

versioning_base.add_branch("dbname=epanet_test_db","epanet","mybranch","add 'branch")