[tryton-commits] changeset in trytond-gis:default Set postgresql scheme to postgi...

[Cédric Krier]

changeset c83498eeebae in trytond-gis:default
details: https://hg.tryton.org/trytond-gis?cmd=changeset&node=c83498eeebae
description:
        Set postgresql scheme to postgis DSN
diffstat:

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

diffs (25 lines):

diff -r 0dc685fdc3a0 -r c83498eeebae 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)