compat.py
No OneTemporary
Actions

Size

992 B

Subscribers

None

compat.py
View Options

	"""pscypog2 Compatibility module."""
	from typing import Any, TYPE_CHECKING
	from platform import python_implementation


	__all__ = ("psycopg2", "cursor", "connection", "check_for_psycopg2")


	try:
	import psycopg2
	except ImportError:
	psycopg2 = False

	if not psycopg2:
	if not TYPE_CHECKING:
	# if there's no postgres, just go with the flow.
	cursor = Any
	connection = Any
	elif python_implementation() == "PyPy":
	from psycopg2cffi._impl.cursor import Cursor as cursor
	from psycopg2cffi._impl.connection import Connection as connection
	else:
	from psycopg2._psycopg import cursor, connection


	def check_for_psycopg2() -> None:
	"""
	Function checks whether psycopg2 was imported.

	Raises ImportError if not.
	"""
	if not psycopg2:
	raise ImportError(
	"No module named psycopg2. Please install either "
	"psycopg2 or psycopg2-binary package for CPython "
	"or psycopg2cffi for Pypy."
	)