[tryton-commits] changeset in trytond-gis:6.0 Set postgresql scheme to postgis DSN

[Cédric Krier]

changeset 669333b31d21 in trytond-gis:6.0
details: https://hg.tryton.org/trytond-gis?cmd=changeset&node=669333b31d21
description:
        Set postgresql scheme to postgis DSN
        (grafted from c83498eeebae115b38fce215501c49acd0d716f3)
diffstat:

 trytond_gis/postgis/database.py |  8 ++++++++
 1 files changed, 8 insertions(+), 0 deletions(-)

diffs (25 lines):

diff -r bc5711db0883 -r 669333b31d21 trytond_gis/postgis/database.py
--- a/trytond_gis/postgis/database.py   Mon May 03 15:34:01 2021 +0200
+++ b/trytond_gis/postgis/database.py   Wed May 05 17:42:37 2021 +0200
@@ -7,6 +7,7 @@
     Binary)
 
 from trytond.backend.postgresql.database import Database as PGDatabase
+from trytond.config import parse_uri
 
 from trytond_gis import _GeoJSON
 
@@ -30,6 +31,13 @@
             cursor = db_connection.cursor()
             cursor.execute("CREATE EXTENSION postgis")
 
+    @classmethod
+    def _connection_params(cls, name):
+        params = super()._connection_params(name)
+        uri = parse_uri(params['dsn'])
+        params['dsn'] = uri._replace(scheme='postgresql').geturl()
+        return params
+
     def get_connection(self, autocommit=False, readonly=False):
         conn = super(Database, self).get_connection(autocommit, readonly)