qualis-bigquery 0.1.0__tar.gz

qualis_bigquery-0.1.0/.github/workflows/ci.yml

@@ -0,0 +1,48 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+permissions:
+  contents: read
+
+jobs:
+  test:
+    name: test (Python ${{ matrix.python-version }})
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.12", "3.13"]
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up uv
+        uses: astral-sh/setup-uv@v3
+        with:
+          enable-cache: true
+
+      - name: Set up Python ${{ matrix.python-version }}
+        run: uv python install ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: uv sync --dev
+
+      - name: Lint with ruff
+        run: uv run ruff check src/ tests/
+
+      - name: Typecheck with mypy
+        run: uv run mypy src/
+
+      - name: Run tests
+        run: uv run pytest --cov=qualis_bigquery --cov-report=term-missing
+
+      - name: Enforce coverage threshold
+        # 55% is appropriate for a thin SDK adapter: SQL templates +
+        # adapter dispatch are fully covered via mocked tests; the
+        # uncovered paths are connection/stream lifecycle that only
+        # exercise with a live BigQuery client.
+        run: uv run pytest --cov=qualis_bigquery --cov-fail-under=55

qualis_bigquery-0.1.0/.github/workflows/release.yml

@@ -0,0 +1,60 @@
+name: Release
+
+on:
+  push:
+    tags:
+      - "v*"
+
+permissions:
+  contents: read
+
+jobs:
+  build:
+    name: Build sdist and wheel
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up uv
+        uses: astral-sh/setup-uv@v3
+
+      - name: Set up Python
+        run: uv python install 3.12
+
+      - name: Build artefacts
+        run: uv build
+
+      - name: Verify version matches tag
+        run: |
+          VERSION_IN_PYPROJECT=$(grep '^version = ' pyproject.toml | head -1 | sed 's/.*"\(.*\)"/\1/')
+          TAG_VERSION="${GITHUB_REF#refs/tags/v}"
+          if [ "$VERSION_IN_PYPROJECT" != "$TAG_VERSION" ]; then
+            echo "Version mismatch: pyproject.toml says '$VERSION_IN_PYPROJECT', tag is '$TAG_VERSION'"
+            exit 1
+          fi
+          echo "Version $VERSION_IN_PYPROJECT matches tag v$TAG_VERSION"
+
+      - name: Upload artefacts
+        uses: actions/upload-artifact@v4
+        with:
+          name: dist
+          path: dist/
+
+  publish:
+    name: Publish to PyPI
+    needs: build
+    runs-on: ubuntu-latest
+    environment:
+      name: pypi
+      url: https://pypi.org/p/qualis-bigquery
+    permissions:
+      id-token: write
+    steps:
+      - name: Download artefacts
+        uses: actions/download-artifact@v4
+        with:
+          name: dist
+          path: dist/
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

qualis_bigquery-0.1.0/.gitignore

@@ -0,0 +1,16 @@
+__pycache__/
+*.py[cod]
+*.so
+.venv/
+venv/
+dist/
+build/
+*.egg-info/
+.pytest_cache/
+.mypy_cache/
+.ruff_cache/
+.coverage
+htmlcov/
+.env
+*.log
+.DS_Store

qualis_bigquery-0.1.0/.python-version

@@ -0,0 +1 @@
+3.12

qualis_bigquery-0.1.0/LICENSE

@@ -0,0 +1,176 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship made available under
+      the License, as indicated by a copyright notice that is included in
+      or attached to the work (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean, as submitted to the Licensor for inclusion
+      in the Work by the copyright owner or by an individual or Legal Entity
+      authorized to submit on behalf of the copyright owner. For the purposes
+      of this definition, "submitted" means any form of electronic, verbal,
+      or written communication sent to the Licensor or its representatives,
+      including but not limited to communication on electronic mailing lists,
+      source code control systems, and issue tracking systems that are managed
+      by, or on behalf of, the Licensor for the purpose of recording
+      instructions given to the Licensor.
+
+      "Contributor" shall mean Licensor and any Legal Entity on behalf of
+      whom a Contribution has been received by the Licensor and is included
+      in the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those Contributions necessary to
+      make, use, sell, offer to sell, import, or otherwise transfer the Work.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or Derivative
+          Works a copy of this License; and
+
+      (b) You must reproduce the above copyright notices for use,
+          reproduction, and distribution, and a relevant copyright notice
+          in the Derivative Works; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file, the contents
+          of the NOTICE file are for informational purposes only
+          and do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or in addition to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own license statement for Your modifications and
+      may provide additional grant of rights to use, copy, modify, merge,
+      publish, distribute, sublicense, and/or sell copies of the Work.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or reproducing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or exemplary damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or all other
+      commercial damages or losses), even if such Contributor has been
+      advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may offer only
+      conditions that are consistent with this full terms and conditions,
+      including warranty disclaimers and limitation of liability contained
+      in this License, or any of the terms and conditions contained in a
+      Section 9 of the License, including without limitation
+      indemnification.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the format of the file. We also recommend that
+      a file or directory name be given, with a reference to where the
+      full text can be found.
+
+   Copyright 2024 Ahmed Ashraf
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

qualis_bigquery-0.1.0/PKG-INFO

@@ -0,0 +1,59 @@
+Metadata-Version: 2.4
+Name: qualis-bigquery
+Version: 0.1.0
+Summary: BigQuery adapter for Qualis (data quality framework)
+Project-URL: Homepage, https://github.com/ahmedashraffcih/qualis-bigquery
+Project-URL: Repository, https://github.com/ahmedashraffcih/qualis-bigquery
+Project-URL: Issues, https://github.com/ahmedashraffcih/qualis-bigquery/issues
+Author-email: Ahmed Ashraf <ahmedashraffcih@gmail.com>
+License: Apache-2.0
+License-File: LICENSE
+Keywords: bigquery,data-quality,dq,qualis
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Database
+Classifier: Topic :: Software Development :: Quality Assurance
+Classifier: Typing :: Typed
+Requires-Python: >=3.12
+Requires-Dist: google-cloud-bigquery>=3.0
+Requires-Dist: qualis>=0.3.1
+Description-Content-Type: text/markdown
+
+# qualis-bigquery
+
+BigQuery adapter for [Qualis](https://github.com/ahmedashraffcih/qualis) —
+the open-source data quality framework.
+
+## Install
+
+```bash
+pip install qualis qualis-bigquery
+```
+
+## Usage
+
+```yaml
+# qualis.yaml
+adapter: bigquery
+project: my-gcp-project
+dataset: my_dataset
+# Auth: standard GCP application default credentials
+#   gcloud auth application-default login
+```
+
+## Supported checks
+
+- not_null, unique, between, regex, in_set
+- row_count, not_negative
+- reference_lookup
+
+The adapter is read-only by contract — DML statements
+(INSERT/UPDATE/DELETE/MERGE/DROP/CREATE/ALTER) are explicitly refused.
+
+## License
+
+Apache 2.0.

qualis_bigquery-0.1.0/README.md

@@ -0,0 +1,34 @@
+# qualis-bigquery
+
+BigQuery adapter for [Qualis](https://github.com/ahmedashraffcih/qualis) —
+the open-source data quality framework.
+
+## Install
+
+```bash
+pip install qualis qualis-bigquery
+```
+
+## Usage
+
+```yaml
+# qualis.yaml
+adapter: bigquery
+project: my-gcp-project
+dataset: my_dataset
+# Auth: standard GCP application default credentials
+#   gcloud auth application-default login
+```
+
+## Supported checks
+
+- not_null, unique, between, regex, in_set
+- row_count, not_negative
+- reference_lookup
+
+The adapter is read-only by contract — DML statements
+(INSERT/UPDATE/DELETE/MERGE/DROP/CREATE/ALTER) are explicitly refused.
+
+## License
+
+Apache 2.0.

qualis_bigquery-0.1.0/pyproject.toml

@@ -0,0 +1,62 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "qualis-bigquery"
+version = "0.1.0"
+description = "BigQuery adapter for Qualis (data quality framework)"
+readme = "README.md"
+license = { text = "Apache-2.0" }
+requires-python = ">=3.12"
+authors = [
+    { name = "Ahmed Ashraf", email = "ahmedashraffcih@gmail.com" },
+]
+keywords = ["data-quality", "bigquery", "qualis", "dq"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: Apache Software License",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Database",
+    "Topic :: Software Development :: Quality Assurance",
+    "Typing :: Typed",
+]
+dependencies = [
+    "qualis>=0.3.1",
+    "google-cloud-bigquery>=3.0",
+]
+
+[project.entry-points."qualis.adapters"]
+bigquery = "qualis_bigquery.adapter:BigQueryAdapter"
+
+[project.urls]
+Homepage = "https://github.com/ahmedashraffcih/qualis-bigquery"
+Repository = "https://github.com/ahmedashraffcih/qualis-bigquery"
+Issues = "https://github.com/ahmedashraffcih/qualis-bigquery/issues"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/qualis_bigquery"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+
+[tool.mypy]
+strict = true
+python_version = "3.12"
+
+[tool.ruff]
+line-length = 100
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "B", "UP", "RUF", "TC"]
+
+[dependency-groups]
+dev = [
+    "mypy>=1.0",
+    "pytest>=8.0",
+    "pytest-cov>=4.0",
+    "ruff>=0.5",
+]

qualis_bigquery-0.1.0/src/qualis_bigquery/__init__.py

@@ -0,0 +1,9 @@
+"""BigQuery adapter for Qualis."""
+
+from __future__ import annotations
+
+__version__ = "0.1.0"
+
+from qualis_bigquery.adapter import BigQueryAdapter
+
+__all__ = ["BigQueryAdapter", "__version__"]

qualis_bigquery-0.1.0/src/qualis_bigquery/adapter.py

@@ -0,0 +1,206 @@
+"""BigQueryAdapter implementing qualis.ports.database.DatabasePort.
+
+BigQuery query execution uses parameterised queries via
+``ScalarQueryParameter`` / ``ArrayQueryParameter``. Authentication uses
+GCP Application Default Credentials.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from qualis_bigquery.sql_templates import (
+    BETWEEN_SQL,
+    IN_SET_SQL,
+    NOT_NEGATIVE_SQL,
+    NOT_NULL_SQL,
+    REFERENCE_LOOKUP_SQL,
+    REGEX_SQL,
+    ROW_COUNT_SQL,
+    UNIQUE_SQL,
+)
+
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
+
+class BigQueryAdapter:
+    """Implements DatabasePort against Google BigQuery."""
+
+    def __init__(self, project: str, location: str | None = None) -> None:
+        self._project = project
+        self._location = location
+        self._client: Any = None
+
+    def _get_client(self) -> Any:
+        if self._client is None:
+            from google.cloud import bigquery
+            self._client = bigquery.Client(
+                project=self._project, location=self._location,
+            )
+        return self._client
+
+    def _qualified(self, schema: str, table: str) -> str:
+        """Return a ``project.dataset.table`` reference (backticked at SQL-render time)."""
+        return f"{self._project}.{schema}.{table}"
+
+    def _make_scalar_param(self, name: str, value: str) -> Any:
+        from google.cloud.bigquery import ScalarQueryParameter
+        return ScalarQueryParameter(name, "STRING", value)
+
+    def _make_array_param(self, name: str, values: list[str]) -> Any:
+        from google.cloud.bigquery import ArrayQueryParameter
+        return ArrayQueryParameter(name, "STRING", values)
+
+    def _run_query(
+        self, sql: str, parameters: list[Any] | None = None,
+    ) -> list[dict[str, Any]]:
+        from google.cloud.bigquery import QueryJobConfig
+        client = self._get_client()
+        config = QueryJobConfig(query_parameters=parameters or [])
+        job = client.query(sql, job_config=config)
+        return [dict(row.items()) for row in job.result()]
+
+    # ------------------------------------------------------------------
+    # DatabasePort methods
+    # ------------------------------------------------------------------
+
+    def query(self, sql: str, params: dict[str, Any] | None = None) -> list[dict[str, Any]]:
+        bq_params = [
+            self._make_scalar_param(k, str(v))
+            for k, v in (params or {}).items()
+        ]
+        return self._run_query(sql, bq_params)
+
+    def execute(self, sql: str, params: dict[str, Any] | None = None) -> int:
+        # Safety gate: refuse DML — adapter is read-only by design.
+        first_word = sql.strip().split(None, 1)[0].upper() if sql.strip() else ""
+        if first_word in {"INSERT", "UPDATE", "DELETE", "MERGE", "DROP", "CREATE", "ALTER"}:
+            raise PermissionError(
+                f"BigQueryAdapter refuses DML statement starting with {first_word!r}"
+            )
+        rows = self.query(sql, params)
+        return len(rows)
+
+    def stream(
+        self, sql: str, params: dict[str, Any] | None = None,
+        chunk_size: int = 10_000,
+    ) -> Iterator[list[dict[str, Any]]]:
+        bq_params = [
+            self._make_scalar_param(k, str(v))
+            for k, v in (params or {}).items()
+        ]
+        from google.cloud.bigquery import QueryJobConfig
+        client = self._get_client()
+        config = QueryJobConfig(query_parameters=bq_params)
+        job = client.query(sql, job_config=config)
+        chunk: list[dict[str, Any]] = []
+        for row in job.result():
+            chunk.append(dict(row.items()))
+            if len(chunk) >= chunk_size:
+                yield chunk
+                chunk = []
+        if chunk:
+            yield chunk
+
+    def table_exists(self, schema: str, table: str) -> bool:
+        from google.cloud import bigquery
+        client = self._get_client()
+        ref = bigquery.TableReference.from_string(
+            f"{self._project}.{schema}.{table}"
+        )
+        try:
+            client.get_table(ref)
+            return True
+        except Exception:
+            return False
+
+    def check_not_null(self, schema: str, table: str, column: str) -> dict[str, int]:
+        sql = NOT_NULL_SQL.format(column=column, table=self._qualified(schema, table))
+        rows = self._run_query(sql)
+        return {
+            "null_count": int(rows[0]["null_count"]),
+            "total_count": int(rows[0]["total_count"]),
+        }
+
+    def check_unique(self, schema: str, table: str, column: str) -> dict[str, int]:
+        sql_dup = UNIQUE_SQL.format(column=column, table=self._qualified(schema, table))
+        sql_total = ROW_COUNT_SQL.format(table=self._qualified(schema, table))
+        dup_rows = self._run_query(sql_dup)
+        total_rows = self._run_query(sql_total)
+        return {
+            "duplicate_count": int(dup_rows[0]["duplicate_count"]) if dup_rows else 0,
+            "total_count": int(total_rows[0]["row_count"]) if total_rows else 0,
+        }
+
+    def check_between(
+        self, schema: str, table: str, column: str,
+        min_val: str, max_val: str,
+    ) -> dict[str, int]:
+        sql = BETWEEN_SQL.format(column=column, table=self._qualified(schema, table))
+        params = [
+            self._make_scalar_param("min", min_val),
+            self._make_scalar_param("max", max_val),
+        ]
+        rows = self._run_query(sql, params)
+        return {
+            "out_of_range_count": int(rows[0]["out_of_range_count"]),
+            "total_count": int(rows[0]["total_count"]),
+            "checked": int(rows[0]["checked"]),
+        }
+
+    def check_regex(
+        self, schema: str, table: str, column: str, pattern: str,
+    ) -> dict[str, int]:
+        sql = REGEX_SQL.format(column=column, table=self._qualified(schema, table))
+        params = [self._make_scalar_param("pattern", pattern)]
+        rows = self._run_query(sql, params)
+        return {
+            "non_matching_count": int(rows[0]["non_matching_count"]),
+            "total_count": int(rows[0]["total_count"]),
+        }
+
+    def check_in_set(
+        self, schema: str, table: str, column: str, values: list[str],
+    ) -> dict[str, int]:
+        sql = IN_SET_SQL.format(column=column, table=self._qualified(schema, table))
+        params = [self._make_array_param("values", values)]
+        rows = self._run_query(sql, params)
+        return {
+            "invalid_count": int(rows[0]["invalid_count"]),
+            "total_count": int(rows[0]["total_count"]),
+        }
+
+    def check_row_count(self, schema: str, table: str) -> dict[str, int]:
+        sql = ROW_COUNT_SQL.format(table=self._qualified(schema, table))
+        rows = self._run_query(sql)
+        return {"row_count": int(rows[0]["row_count"]) if rows else 0}
+
+    def check_not_negative(
+        self, schema: str, table: str, column: str,
+    ) -> dict[str, int]:
+        sql = NOT_NEGATIVE_SQL.format(column=column, table=self._qualified(schema, table))
+        rows = self._run_query(sql)
+        return {
+            "negative_count": int(rows[0]["negative_count"]),
+            "total_count": int(rows[0]["total_count"]),
+        }
+
+    def check_reference_lookup(
+        self, schema: str, table: str, column: str,
+        valid_values: list[str],
+    ) -> dict[str, int]:
+        sql = REFERENCE_LOOKUP_SQL.format(
+            column=column, table=self._qualified(schema, table),
+        )
+        params = [self._make_array_param("valid_values", valid_values)]
+        rows = self._run_query(sql, params)
+        return {
+            "invalid_count": int(rows[0]["invalid_count"]),
+            "total_count": int(rows[0]["total_count"]),
+        }
+
+    def close(self) -> None:
+        if self._client is not None:
+            self._client.close()
+            self._client = None