From 672022a1671f345d7bb0f06ca0ff4d5eb4e8eeaf Mon Sep 17 00:00:00 2001 From: Jo-Ann Meunier Date: Thu, 13 Jul 2023 13:26:41 -0400 Subject: [PATCH] more wip --- .../datadog_checks/base/data/agent_requirements.in | 1 + postgres/datadog_checks/postgres/postgres.py | 1 + postgres/pyproject.toml | 1 + 3 files changed, 3 insertions(+) diff --git a/datadog_checks_base/datadog_checks/base/data/agent_requirements.in b/datadog_checks_base/datadog_checks/base/data/agent_requirements.in index 120f7c0e832ab..0f5e2f3f615b1 100644 --- a/datadog_checks_base/datadog_checks/base/data/agent_requirements.in +++ b/datadog_checks_base/datadog_checks/base/data/agent_requirements.in @@ -57,6 +57,7 @@ protobuf==3.17.3; python_version < '3.0' protobuf==3.20.2; python_version > '3.0' psutil==5.9.0 psycopg[binary]==3.1.9; python_version > '3.0' +psycopg[pool]==3.1.7; python_version > '3.0' pyasn1==0.4.6 pycryptodomex==3.10.1 pydantic==1.10.8; python_version > '3.0' diff --git a/postgres/datadog_checks/postgres/postgres.py b/postgres/datadog_checks/postgres/postgres.py index 29433d4a422ef..760e1d85a65a1 100644 --- a/postgres/datadog_checks/postgres/postgres.py +++ b/postgres/datadog_checks/postgres/postgres.py @@ -9,6 +9,7 @@ import psycopg from psycopg.rows import dict_row +from psycopg_pool import ConnectionPool from six import iteritems from datadog_checks.base import AgentCheck diff --git a/postgres/pyproject.toml b/postgres/pyproject.toml index 3ebab24ffef69..cca5eb5ba5651 100644 --- a/postgres/pyproject.toml +++ b/postgres/pyproject.toml @@ -42,6 +42,7 @@ deps = [ "cachetools==5.3.1; python_version > '3.0'", "psycopg[binary]==3.1.9; python_version > '3.0'", "semver==3.0.1; python_version > '3.0'", + "psycopg[pool]==3.1.7; python_version > '3.0'", ] [project.urls]