Reveal the appropriate exception during dbapi20.connect(); db.close chained ClientCannotConnect.

jwp · jwp · commit 25e21b29a5f1 · 2020-09-18T11:32:01.000-07:00
diff --git a/postgresql/driver/dbapi20.py b/postgresql/driver/dbapi20.py
@@ -323,6 +323,7 @@ class Connection(Connection):
 		Warning
 	DatabaseError = DatabaseError
 	NotSupportedError = NotSupportedError
+	_dbapi_connected = False
 
 	def autocommit_set(self, val):
 		if val:
@@ -354,9 +355,10 @@ def connect(self, *args, **kw):
 		super().connect(*args, **kw)
 		self._xact = self.xact()
 		self._xact.start()
+		self._dbapi_connected = True
 
 	def close(self):
-		if self.closed:
+		if self.closed and self._dbapi_connected:
 			raise Error(
 				"connection already closed",
 				source = 'CLIENT',
