mirror of
https://github.com/osm-search/Nominatim.git
synced 2026-02-16 15:47:58 +00:00
3c05d98b80
The psycopg dialect was only added in SQLAlchemy 2.0. To avoid loading errors when SQLAlchemy 1.4 is installed together with psycopg3, check that the dialect is really available.
24 lines
674 B
Python
24 lines
674 B
Python
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
#
|
|
# This file is part of Nominatim. (https://nominatim.org)
|
|
#
|
|
# Copyright (C) 2023 by the Nominatim developer community.
|
|
# For a full list of authors see the git log.
|
|
"""
|
|
Import the base library to use with asynchronous SQLAlchemy.
|
|
"""
|
|
# pylint: disable=invalid-name, ungrouped-imports, unused-import
|
|
|
|
from typing import Any
|
|
|
|
try:
|
|
import sqlalchemy.dialects.postgresql.psycopg
|
|
import psycopg
|
|
PGCORE_LIB = 'psycopg'
|
|
PGCORE_ERROR: Any = psycopg.Error
|
|
except ModuleNotFoundError:
|
|
import sqlalchemy.dialects.postgresql.asyncpg
|
|
import asyncpg
|
|
PGCORE_LIB = 'asyncpg'
|
|
PGCORE_ERROR = asyncpg.PostgresError
|