nominatim/db/async_core_library.py

   1 # SPDX-License-Identifier: GPL-3.0-or-later
   2 #
   3 # This file is part of Nominatim. (https://nominatim.org)
   4 #
   5 # Copyright (C) 2023 by the Nominatim developer community.
   6 # For a full list of authors see the git log.
   7 """
   8 Import the base library to use with asynchronous SQLAlchemy.
   9 """
  10 # pylint: disable=invalid-name
  11
  12 from typing import Any
  13
  14 try:
  15     import psycopg
  16     PGCORE_LIB = 'psycopg'
  17     PGCORE_ERROR: Any = psycopg.Error
  18 except ModuleNotFoundError:
  19     import asyncpg
  20     PGCORE_LIB = 'asyncpg'
  21     PGCORE_ERROR = asyncpg.PostgresError