Set postgresql scheme to postgis DSN

This commit is contained in:
Cédric Krier committed 2021-05-05 17:42:37 +02:00
1 parent f9b4336714
commit ec0766bf76
1 file changed
+8
+8
View File
@@ -7,6 +7,7 @@ from psycopg2.extensions import (register_adapter, new_type, register_type,
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 @@ class Database(PGDatabase):
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)