hdx-python-database 1.3.5__tar.gz → 1.3.7__tar.gz

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/.github/workflows/publish.yaml

@@ -8,6 +8,13 @@ jobs:
   publish:
     runs-on: ubuntu-latest
 
+    environment:
+      name: pypi
+      url: https://pypi.org/p/hdx-python-database
+
+    permissions:
+      id-token: write  # IMPORTANT: mandatory for trusted publishing
+
     steps:
     - uses: actions/checkout@v4
     - name: Get history and tags for versioning to work
@@ -26,9 +33,5 @@ jobs:
     - name: Build with hatch
       run: |
         hatch build
-    - name: Publish with hatch
-      env:
-        HATCH_INDEX_USER: ${{secrets.HATCH_INDEX_USER}}
-        HATCH_INDEX_AUTH: ${{secrets.HATCH_INDEX_AUTH}}
-      run: |
-        hatch publish
+    - name: Publish distribution 📦 to PyPI
+      uses: pypa/gh-action-pypi-publish@release/v1

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/.github/workflows/run-python-tests.yaml

@@ -17,6 +17,10 @@ jobs:
   build:
     runs-on: ubuntu-latest
 
+    permissions:
+      checks: write
+      pull-requests: write
+
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python

hdx_python_database-1.3.5/.config/pre-commit-config.yaml → hdx_python_database-1.3.7/.pre-commit-config.yaml

@@ -1,5 +1,5 @@
 default_language_version:
-    python: python3.12
+    python: python3.13
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v5.0.0
@@ -8,16 +8,15 @@ repos:
       - id: end-of-file-fixer
       - id: check-ast
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.9.6
+    rev: v0.12.0
     hooks:
       # Run the linter.
-      - id: ruff
-        args: [--config, .config/ruff.toml, --fix]
+      - id: ruff-check
+        args: [ --fix ]
       # Run the formatter.
       - id: ruff-format
-        args: [--config, .config/ruff.toml]
   - repo: https://github.com/astral-sh/uv-pre-commit
-    rev: 0.5.29
+    rev: 0.7.14
     hooks:
       # Run the pip compile
       - id: pip-compile

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hdx-python-database
-Version: 1.3.5
+Version: 1.3.7
 Summary: HDX Python database utilities
 Project-URL: Homepage, https://github.com/OCHA-DAP/hdx-python-database
 Author-email: Michael Rans <rans@email.com>
@@ -24,17 +24,21 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.8
+Requires-Python: >=3.10
 Requires-Dist: sqlalchemy>=2
 Provides-Extra: dev
 Requires-Dist: pre-commit; extra == 'dev'
+Provides-Extra: docs
+Requires-Dist: mkapi; extra == 'docs'
 Provides-Extra: postgresql
 Requires-Dist: psycopg[binary]; extra == 'postgresql'
 Provides-Extra: sshtunnel
 Requires-Dist: sshtunnel; extra == 'sshtunnel'
 Provides-Extra: test
+Requires-Dist: psycopg[binary]; extra == 'test'
 Requires-Dist: pytest; extra == 'test'
 Requires-Dist: pytest-cov; extra == 'test'
+Requires-Dist: sshtunnel; extra == 'test'
 Description-Content-Type: text/markdown
 
 [![Build Status](https://github.com/OCHA-DAP/hdx-python-database/actions/workflows/run-python-tests.yaml/badge.svg)](https://github.com/OCHA-DAP/hdx-python-database/actions/workflows/run-python-tests.yaml)

hdx_python_database-1.3.7/documentation/.readthedocs.yaml

@@ -0,0 +1,14 @@
+# Required
+version: 2
+
+# Set the version of Python and other tools you might need
+build:
+  os: ubuntu-24.04
+  tools:
+    python: "3.12"
+  jobs:
+    pre_build:
+      - pip install .[docs]
+
+mkdocs:
+  configuration: documentation/mkdocs.yaml

hdx_python_database-1.3.5/documentation/main.md → hdx_python_database-1.3.7/documentation/index.md

@@ -61,6 +61,7 @@ Versions from 1.0.6 no longer support Python 2.7.
 A fresh schema can be created using the function `recreate_schema`. Any
 existing schema with that name will be dropped eg.
 
+    from hdx.database import Database
     Database.recreate_schema(engine, db_uri)
 
 Your SQLAlchemy database tables must inherit from `Base` in
@@ -117,6 +118,16 @@ If a prepare function is supplied in prepare_fn, it will be executed before
 Base.metadata.create_all and the results of it returned in instance variable
 prepare_results.
 
+Batch population of rows (list of dictionaries) can be achieved with the
+batch_populate method of Database, for example:
+
+        now = datetime(2023, 10, 20, 22, 35, 55, tzinfo=timezone.utc)
+        rows = [{"test_date": now}]
+        dbdatabase.batch_populate(rows, DBTestDate)
+        dbtestdate = (
+            dbsession.execute(select(DBTestDate)).all()[1][0].test_date
+        )
+
 
 ## Connection URI
 

hdx_python_database-1.3.7/documentation/mkdocs.yaml

@@ -0,0 +1,18 @@
+site_name: HDX Python Database
+repo_url: https://github.com/OCHA-DAP/hdx-python-database/
+repo_name: OCHA-DAP/hdx-python-database
+docs_dir: .
+site_dir: ../site
+theme:
+  name: material
+  highlightjs: true
+plugins:
+  - search
+  - mkapi
+nav:
+  - Home: index.md
+  - API Documentation:
+    - Database:
+      - $src/hdx.database.Database
+      - $src/hdx.database.DatabaseError
+    - PostgreSQL specific: $src/hdx.database.postgresql.*

hdx_python_database-1.3.7/hatch.toml

@@ -0,0 +1,34 @@
+# Build
+
+[build.targets.wheel]
+packages = ["src/hdx"]
+
+[build.hooks.vcs]
+version-file = "src/hdx/database/_version.py"
+
+# Versioning
+
+[version]
+source = "vcs"
+
+[version.raw-options]
+local_scheme = "no-local-version"
+version_scheme = "python-simplified-semver"
+
+# Tests
+
+[envs.hatch-test]
+features = ["test"]
+
+[[envs.hatch-test.matrix]]
+python = ["3.13"]
+
+[envs.hatch-test.scripts]
+run = """
+       pytest --rootdir=. --junitxml=test-results.xml --cov --no-cov-on-fail \
+       --cov-report=lcov --cov-report=term-missing
+       """
+
+[envs.hatch-static-analysis]
+config-path = "none"
+dependencies = ["ruff==0.12.0"]

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/pyproject.toml

@@ -31,7 +31,7 @@ classifiers = [
     "Operating System :: MacOS",
     "Operating System :: Microsoft :: Windows",
 ]
-requires-python = ">=3.8"
+requires-python = ">=3.10"
 
 dependencies = [
     "sqlalchemy>=2",
@@ -48,51 +48,6 @@ Homepage = "https://github.com/OCHA-DAP/hdx-python-database"
 [project.optional-dependencies]
 sshtunnel = ["sshtunnel"]
 postgresql = ["psycopg[binary]"]
-test = ["pytest", "pytest-cov"]
+test = ["sshtunnel", "psycopg[binary]", "pytest", "pytest-cov"]
 dev = ["pre-commit"]
-
-
-#########
-# Hatch #
-#########
-
-# Build
-
-[tool.hatch.build.targets.wheel]
-packages = ["src/hdx"]
-
-[tool.hatch.build.hooks.vcs]
-version-file = "src/hdx/database/_version.py"
-
-# Versioning
-
-[tool.hatch.version]
-source = "vcs"
-
-[tool.hatch.version.raw-options]
-local_scheme = "no-local-version"
-version_scheme = "python-simplified-semver"
-
-# Tests
-
-[tool.hatch.envs.hatch-test]
-features = ["sshtunnel", "postgresql", "test"]
-
-[[tool.hatch.envs.hatch-test.matrix]]
-python = ["3.12"]
-
-[tool.hatch.envs.hatch-test.scripts]
-run = """
-       pytest -c .config/pytest.ini --rootdir=. --junitxml=test-results.xml \
-       --cov --cov-config=.config/coveragerc --no-cov-on-fail \
-       --cov-report=lcov --cov-report=term-missing
-       """
-
-[tool.hatch.envs.hatch-static-analysis]
-dependencies = ["ruff==0.9.6"]
-
-[tool.hatch.envs.hatch-static-analysis.scripts]
-format-check = ["ruff format --config .config/ruff.toml --check --diff {args:.}",]
-format-fix = ["ruff format --config .config/ruff.toml {args:.}",]
-lint-check = ["ruff check --config .config/ruff.toml {args:.}",]
-lint-fix = ["ruff check --config .config/ruff.toml --fix {args:.}",]
+docs = ["mkapi"]

{hdx_python_database-1.3.5/.config → hdx_python_database-1.3.7}/pytest.ini

@@ -1,4 +1,4 @@
 [pytest]
-pythonpath = ../src
+pythonpath = src
 addopts = "--color=yes"
 log_cli = 1

hdx_python_database-1.3.7/requirements.txt

@@ -0,0 +1,141 @@
+# This file was autogenerated by uv via the following command:
+#    uv pip compile pyproject.toml --resolver=backtracking --all-extras -o requirements.txt
+astdoc==1.3.0
+    # via mkapi
+babel==2.17.0
+    # via mkdocs-material
+backrefs==5.9
+    # via mkdocs-material
+bcrypt==4.3.0
+    # via paramiko
+certifi==2025.7.14
+    # via requests
+cffi==1.17.1
+    # via
+    #   cryptography
+    #   pynacl
+cfgv==3.4.0
+    # via pre-commit
+charset-normalizer==3.4.2
+    # via requests
+click==8.2.1
+    # via mkdocs
+colorama==0.4.6
+    # via mkdocs-material
+coverage==7.9.2
+    # via pytest-cov
+cryptography==45.0.5
+    # via paramiko
+distlib==0.4.0
+    # via virtualenv
+filelock==3.18.0
+    # via virtualenv
+ghp-import==2.1.0
+    # via mkdocs
+greenlet==3.2.3
+    # via sqlalchemy
+identify==2.6.12
+    # via pre-commit
+idna==3.10
+    # via requests
+iniconfig==2.1.0
+    # via pytest
+jinja2==3.1.6
+    # via
+    #   mkapi
+    #   mkdocs
+    #   mkdocs-material
+markdown==3.8.2
+    # via
+    #   mkdocs
+    #   mkdocs-material
+    #   pymdown-extensions
+markupsafe==3.0.2
+    # via
+    #   jinja2
+    #   mkdocs
+mergedeep==1.3.4
+    # via
+    #   mkdocs
+    #   mkdocs-get-deps
+mkapi==4.4.4
+    # via hdx-python-database (pyproject.toml)
+mkdocs==1.6.1
+    # via
+    #   mkapi
+    #   mkdocs-material
+mkdocs-get-deps==0.2.0
+    # via mkdocs
+mkdocs-material==9.6.15
+    # via mkapi
+mkdocs-material-extensions==1.3.1
+    # via mkdocs-material
+nodeenv==1.9.1
+    # via pre-commit
+packaging==25.0
+    # via
+    #   mkdocs
+    #   pytest
+paginate==0.5.7
+    # via mkdocs-material
+paramiko==3.5.1
+    # via sshtunnel
+pathspec==0.12.1
+    # via mkdocs
+platformdirs==4.3.8
+    # via
+    #   mkdocs-get-deps
+    #   virtualenv
+pluggy==1.6.0
+    # via
+    #   pytest
+    #   pytest-cov
+pre-commit==4.2.0
+    # via hdx-python-database (pyproject.toml)
+psycopg==3.2.9
+    # via hdx-python-database (pyproject.toml)
+psycopg-binary==3.2.9
+    # via psycopg
+pycparser==2.22
+    # via cffi
+pygments==2.19.2
+    # via
+    #   mkdocs-material
+    #   pytest
+pymdown-extensions==10.16
+    # via mkdocs-material
+pynacl==1.5.0
+    # via paramiko
+pytest==8.4.1
+    # via
+    #   hdx-python-database (pyproject.toml)
+    #   pytest-cov
+pytest-cov==6.2.1
+    # via hdx-python-database (pyproject.toml)
+python-dateutil==2.9.0.post0
+    # via ghp-import
+pyyaml==6.0.2
+    # via
+    #   mkdocs
+    #   mkdocs-get-deps
+    #   pre-commit
+    #   pymdown-extensions
+    #   pyyaml-env-tag
+pyyaml-env-tag==1.1
+    # via mkdocs
+requests==2.32.4
+    # via mkdocs-material
+six==1.17.0
+    # via python-dateutil
+sqlalchemy==2.0.41
+    # via hdx-python-database (pyproject.toml)
+sshtunnel==0.4.0
+    # via hdx-python-database (pyproject.toml)
+typing-extensions==4.14.1
+    # via sqlalchemy
+urllib3==2.5.0
+    # via requests
+virtualenv==20.32.0
+    # via pre-commit
+watchdog==6.0.0
+    # via mkdocs

{hdx_python_database-1.3.5/.config → hdx_python_database-1.3.7}/ruff.toml

@@ -1,4 +1,3 @@
-line-length = 79
 exclude = ["_version.py"]
 
 [lint]

hdx_python_database-1.3.7/src/hdx/database/__init__.py

@@ -0,0 +1,3 @@
+from ._version import version as __version__  # noqa: F401
+from .database import Database as Database
+from .database import DatabaseError as DatabaseError

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/src/hdx/database/_version.py

@@ -1,8 +1,13 @@
-# file generated by setuptools_scm
+# file generated by setuptools-scm
 # don't change, don't track in version control
+
+__all__ = ["__version__", "__version_tuple__", "version", "version_tuple"]
+
 TYPE_CHECKING = False
 if TYPE_CHECKING:
-    from typing import Tuple, Union
+    from typing import Tuple
+    from typing import Union
+
     VERSION_TUPLE = Tuple[Union[int, str], ...]
 else:
     VERSION_TUPLE = object
@@ -12,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.3.5'
-__version_tuple__ = version_tuple = (1, 3, 5)
+__version__ = version = '1.3.7'
+__version_tuple__ = version_tuple = (1, 3, 7)

hdx_python_database-1.3.5/src/hdx/database/__init__.py → hdx_python_database-1.3.7/src/hdx/database/database.py

@@ -3,14 +3,13 @@
 import logging
 from typing import Any, Dict, List, Optional, Tuple, Type, Union
 
-from sqlalchemy import Engine, TableClause, create_engine
+from sqlalchemy import Engine, TableClause, create_engine, insert
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy.ext.automap import automap_base
 from sqlalchemy.orm import DeclarativeBase, Session
 from sqlalchemy.pool import NullPool
 from sqlalchemy.sql.ddl import CreateSchema, DropSchema
 
-from ._version import version as __version__  # noqa: F401
 from .dburi import get_connection_uri
 from .no_timezone import Base as NoTZBase
 from .postgresql import restore_from_pgfile, wait_for_postgresql
@@ -123,16 +122,16 @@ class Database:
                 del kwargs["ssh_port"]
             else:
                 ssh_port = 22
-            self.server = sshtunnel.SSHTunnelForwarder(
+            self._server = sshtunnel.SSHTunnelForwarder(
                 (ssh_host, ssh_port),
                 remote_bind_address=(host, port),
                 **kwargs,
             )
-            self.server.start()
-            host = self.server.local_bind_host
-            port = self.server.local_bind_port
+            self._server.start()
+            host = self._server.local_bind_host
+            port = self._server.local_bind_port
         else:
-            self.server = None
+            self._server = None
         if not engine and not db_uri:
             db_uri = get_connection_uri(
                 database,
@@ -158,19 +157,19 @@ class Database:
                 table_base = NoTZBase
         if not engine:
             engine = create_engine(db_uri, poolclass=NullPool, echo=False)
-        self.engine: Engine = engine
+        self._engine: Engine = engine
         if recreate_schema:
             self.recreate_schema(engine, schema_name)
-        self.prepare_results = prepare_fn()
-        self.session, self.base = self.create_session(
+        self._prepare_results = prepare_fn()
+        self._session, self._base = self.create_session(
             engine,
             table_base=table_base,
             reflect=reflect,
         )
         if reflect:
-            self.reflected_classes = self.base.classes
+            self._reflected_classes = self._base.classes
         else:
-            self.reflected_classes = None
+            self._reflected_classes = None
 
     def cleanup(self) -> None:
         """Cleanup SQLAlchemy.
@@ -178,10 +177,10 @@ class Database:
         Returns:
             sqlalchemy.Engine: SQLAlchemy engine
         """
-        self.session.close()
-        self.engine.dispose()
-        if self.server is not None:
-            self.server.stop()
+        self._session.close()
+        self._engine.dispose()
+        if self._server is not None:
+            self._server.stop()
 
     def __enter__(self) -> "Database":
         return self
@@ -195,7 +194,7 @@ class Database:
         Returns:
             None
         """
-        self.base.metadata.drop_all(self.engine)
+        self._base.metadata.drop_all(self._engine)
 
     def get_engine(self) -> Engine:
         """Returns SQLAlchemy engine.
@@ -203,7 +202,7 @@ class Database:
         Returns:
             sqlalchemy.Engine: SQLAlchemy engine
         """
-        return self.engine
+        return self._engine
 
     def get_session(self) -> Session:
         """Returns SQLAlchemy session.
@@ -211,7 +210,7 @@ class Database:
         Returns:
             sqlalchemy.orm.Session: SQLAlchemy session
         """
-        return self.session
+        return self._session
 
     def get_reflected_classes(self) -> Any:
         """Gets reflected classes.
@@ -219,7 +218,7 @@ class Database:
         Returns:
             Any: Reflected classes
         """
-        return self.reflected_classes
+        return self._reflected_classes
 
     def get_prepare_results(self) -> Any:
         """Returns results from prepare function.
@@ -227,7 +226,31 @@ class Database:
         Returns:
             Any: Results from prepare function
         """
-        return self.prepare_results
+        return self._prepare_results
+
+    def batch_populate(
+        self,
+        rows: List[Dict],
+        dbtable: Type[DeclarativeBase],
+        batch_size: int = 1000,
+    ) -> None:
+        """Batch populate database table.
+
+        Args:
+            rows (List[Dict]): List of rows
+            dbtable (Type[DeclarativeBase]): Database table
+            batch_size (int): Batch size. Defaults to 1000.
+
+        Returns:
+            Tuple[Session, Any]: (SQLAlchemy session, base)
+        """
+        batches = range(len(rows) // batch_size + 1)
+        for batch in batches:
+            start_row = batch * batch_size
+            end_row = start_row + batch_size
+            batch_rows = rows[start_row:end_row]
+            self._session.execute(insert(dbtable), batch_rows)
+        self._session.commit()
 
     @staticmethod
     def create_session(
@@ -258,17 +281,15 @@ class Database:
             engine = create_engine(db_uri, poolclass=NullPool, echo=False)
         if reflect:
             Base = automap_base(declarative_base=table_base)
-            Base.prepare(
-                autoload_with=engine, reflection_options={"views": True}
-            )
+            Base.prepare(autoload_with=engine, reflection_options={"views": True})
             table_base = Base
         else:
             table_base.metadata.create_all(engine)
         return Session(engine), table_base
 
     @staticmethod
-    def recreate_schema(engine: Engine, schema_name: str = "public") -> bool:
-        """Wipe and create empty schema in database using SQLAlchemy.
+    def drop_schema(engine: Engine, schema_name: str = "public") -> bool:
+        """Wipe schema in database using SQLAlchemy.
 
         Args:
             engine (Engine): SQLAlchemy engine to use.
@@ -284,10 +305,30 @@ class Database:
                     DropSchema(schema_name, cascade=True, if_exists=True)
                 )
                 connection.commit()
+                return True
+        except SQLAlchemyError:
+            return False
+
+    @staticmethod
+    def recreate_schema(engine: Engine, schema_name: str = "public") -> bool:
+        """Wipe and create empty schema in database using SQLAlchemy.
+
+        Args:
+            engine (Engine): SQLAlchemy engine to use.
+            schema_name (str): Schema name. Defaults to "public".
+
+        Returns:
+            bool: True if all successful, False if not
+        """
+        # Wipe and create an empty schema
+        try:
+            with engine.connect() as connection:
                 connection.execute(
-                    CreateSchema(schema_name, if_not_exists=True)
+                    DropSchema(schema_name, cascade=True, if_exists=True)
                 )
                 connection.commit()
+                connection.execute(CreateSchema(schema_name, if_not_exists=True))
+                connection.commit()
                 return True
         except SQLAlchemyError:
             return False

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/src/hdx/database/postgresql.py

@@ -59,6 +59,12 @@ def restore_from_pgfile(db_uri: str, pg_restore_file: str) -> str:
         str: Output from the pg_restore command
     """
     db_params = get_params_from_connection_uri(db_uri)
+    if pg_restore_file[-3:] == ".xz":
+        decompress = subprocess.Popen(
+            ("unxz", "-c", "-d", pg_restore_file), stdout=subprocess.PIPE
+        ).stdout
+    else:
+        decompress = None
     subprocess_params = ["pg_restore", "-c"]
     for key, value in db_params.items():
         match key:
@@ -74,13 +80,18 @@ def restore_from_pgfile(db_uri: str, pg_restore_file: str) -> str:
                 continue
         subprocess_params.append(f"{value}")
 
-    subprocess_params.append(f"{pg_restore_file}")
+    if not decompress:
+        subprocess_params.append(pg_restore_file)
     env = environ.copy()
     password = db_params.get("password")
     if password:
         env["PGPASSWORD"] = password
     process = subprocess.run(
-        subprocess_params, env=env, capture_output=True, encoding="utf-8"
+        subprocess_params,
+        env=env,
+        capture_output=True,
+        encoding="utf-8",
+        stdin=decompress,
     )
     try:
         process.check_returncode()

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/src/hdx/database/utils.py

@@ -13,7 +13,5 @@ def camel_to_snake_case(string: str) -> str:
     Returns:
         str: String in snake case
     """
-    string = re.sub(
-        r"((?<=[a-z0-9])[A-Z]|(?!^)[A-Z](?=[a-z]))", r"_\1", string
-    )
+    string = re.sub(r"((?<=[a-z0-9])[A-Z]|(?!^)[A-Z](?=[a-z]))", r"_\1", string)
     return string.lower().lstrip("_")

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/tests/hdx/database/test_database.py

@@ -31,10 +31,9 @@ class TestDatabase:
         ) as dbdatabase:
             dbsession = dbdatabase.get_session()
             assert str(dbsession.bind.engine.url) == f"sqlite:///{dbpath}"
-            assert dbdatabase.base == NoTZBase
+            assert dbdatabase._base == NoTZBase
             now = datetime(2022, 10, 20, 22, 35, 55, tzinfo=timezone.utc)
-            input_dbtestdate = DBTestDate()
-            input_dbtestdate.test_date = now
+            input_dbtestdate = DBTestDate(test_date=now)
             dbsession.add(input_dbtestdate)
             dbsession.commit()
             dbtestdate = dbsession.execute(select(DBTestDate)).scalar_one()
@@ -42,6 +41,13 @@ class TestDatabase:
             date_view = dbdatabase.get_prepare_results()[0]
             dbtestdate = dbsession.execute(select(date_view)).scalar_one()
             assert dbtestdate == now
+
+            now = datetime(2023, 10, 20, 22, 35, 55, tzinfo=timezone.utc)
+            rows = [{"test_date": now}]
+            dbdatabase.batch_populate(rows, DBTestDate)
+            dbtestdate = dbsession.execute(select(DBTestDate)).all()[1][0].test_date
+            assert dbtestdate == now
+
         remove(dbpath)
 
     def test_errors(self):

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/tests/hdx/database/test_dburi.py

@@ -17,9 +17,7 @@ class TestDBURI:
         "dialect": "postgresql",
         "driver": "psycopg",
     }
-    connection_uri_pg = (
-        "postgresql+psycopg://myuser:mypass@myserver:1234/mydatabase"
-    )
+    connection_uri_pg = "postgresql+psycopg://myuser:mypass@myserver:1234/mydatabase"
     params_pg_no_driver = {
         "database": "mydatabase",
         "host": "myserver",
@@ -37,9 +35,7 @@ class TestDBURI:
         "dialect": "postgresql",
         "driver": None,
     }
-    connection_uri_pg_no_driver = (
-        "postgresql://myuser:mypass@myserver:1234/mydatabase"
-    )
+    connection_uri_pg_no_driver = "postgresql://myuser:mypass@myserver:1234/mydatabase"
     params_sq_no_driver = {
         "database": "mydatabase",
         "host": "myserver",
@@ -67,9 +63,7 @@ class TestDBURI:
             include_driver=False,
         )
         assert result == TestDBURI.params_pg_no_driver
-        result = get_params_from_connection_uri(
-            TestDBURI.connection_uri_pg_no_driver
-        )
+        result = get_params_from_connection_uri(TestDBURI.connection_uri_pg_no_driver)
         assert result == TestDBURI.params_pg_driver_none
         result = get_params_from_connection_uri(TestDBURI.connection_uri_sq)
         assert result == TestDBURI.params_sq_driver_none
@@ -77,9 +71,7 @@ class TestDBURI:
     def test_get_connection_uri(self):
         result = get_connection_uri(**TestDBURI.params_pg)
         assert result == TestDBURI.connection_uri_pg
-        result = get_connection_uri(
-            **TestDBURI.params_pg, include_driver=False
-        )
+        result = get_connection_uri(**TestDBURI.params_pg, include_driver=False)
         assert result == TestDBURI.connection_uri_pg_no_driver
         result = get_connection_uri(**TestDBURI.params_pg_no_driver)
         assert result == TestDBURI.connection_uri_pg

hdx_python_database-1.3.5/tests/hdx/database/test_recreate.py → hdx_python_database-1.3.7/tests/hdx/database/test_drop_recreate.py

@@ -1,6 +1,13 @@
 from hdx.database import Database
 
 
+def test_drop_schema(mock_engine):
+    db_uri = "postgresql+psycopg://myuser:mypass@0.0.0.0:12345/mydatabase"
+    assert Database.drop_schema(mock_engine, db_uri) is True
+    db_uri = "Error"
+    assert Database.drop_schema(mock_engine, db_uri) is False
+
+
 def test_recreate_schema(mock_engine):
     db_uri = "postgresql+psycopg://myuser:mypass@0.0.0.0:12345/mydatabase"
     assert Database.recreate_schema(mock_engine, db_uri) is True

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/tests/hdx/database/test_reflect.py

@@ -27,7 +27,7 @@ class TestReflect:
             dburi = f"sqlite:///{dbpath}"
             assert str(dbengine.url) == dburi
             dbsession = dbdatabase.get_session()
-            assert isinstance(dbdatabase.base, DeclarativeAttributeIntercept)
+            assert isinstance(dbdatabase._base, DeclarativeAttributeIntercept)
             assert str(dbsession.bind.engine.url) == dburi
             Table1 = dbdatabase.get_reflected_classes().table1
             row = dbsession.execute(select(Table1)).scalar_one()

{hdx_python_database-1.3.5 → hdx_python_database-1.3.7}/tests/hdx/database/test_ssh.py

@@ -33,7 +33,7 @@ class TestSSH:
                 str(dbsession.bind.engine.url)
                 == "postgresql+psycopg://myuser@0.0.0.0:12345/mydatabase"
             )
-            assert dbdatabase.base == TZBase
+            assert dbdatabase._base == TZBase
 
         with Database(
             ssh_host="mysshhost", ssh_port=25, table_base=TZBase, **params
@@ -43,7 +43,7 @@ class TestSSH:
                 str(dbsession.bind.engine.url)
                 == "postgresql+psycopg://myuser@0.0.0.0:12345/mydatabase"
             )
-            assert dbdatabase.base == TZBase
+            assert dbdatabase._base == TZBase
 
         with Database(ssh_host="mysshhost", **self.params_pg) as dbdatabase:
             dbsession = dbdatabase.get_session()
@@ -51,14 +51,12 @@ class TestSSH:
                 str(dbsession.bind.engine.url)
                 == "postgresql+psycopg://myuser:***@0.0.0.0:12345/mydatabase"
             )
-            assert dbdatabase.base == NoTZBase
+            assert dbdatabase._base == NoTZBase
 
-        with Database(
-            ssh_host="mysshhost", ssh_port=25, **params
-        ) as dbdatabase:
+        with Database(ssh_host="mysshhost", ssh_port=25, **params) as dbdatabase:
             dbsession = dbdatabase.get_session()
             assert (
                 str(dbsession.bind.engine.url)
                 == "postgresql+psycopg://myuser@0.0.0.0:12345/mydatabase"
             )
-            assert dbdatabase.base == NoTZBase
+            assert dbdatabase._base == NoTZBase

hdx_python_database-1.3.5/documentation/.readthedocs.yaml

@@ -1,16 +0,0 @@
-version: 2
-
-mkdocs:
-  configuration: docs/mkdocs.yml
-  fail_on_warning: false
-
-build:
-  os: ubuntu-22.04
-  tools:
-    python: "3.11"
-  jobs:
-    pre_build:
-      - pip install --upgrade mkdocs
-      - pip install mkdocs-material
-      - pip install pydoc-markdown
-      - pydoc-markdown documentation/pydoc-markdown.yaml

hdx_python_database-1.3.5/documentation/pydoc-markdown.yaml

@@ -1,29 +0,0 @@
-loaders:
-  - type: python
-    search_path:
-      - ../src
-    packages:
-      - hdx.database
-renderer:
-  type: mkdocs
-  output_directory: docs
-  mkdocs_config:
-    site_name: HDX Python Database
-    theme: material
-    repo_url: "https://github.com/OCHA-DAP/hdx-python-database"
-  markdown:
-    source_linker:
-      type: github
-      repo: OCHA-DAP/hdx-python-database
-  pages:
-    - title: Home
-      name: index
-      source: main.md
-    - title: API Documentation
-      children:
-        - title: Database
-          contents:
-            - hdx.database.Database.*
-        - title: PostgreSQL specific
-          contents:
-            - hdx.database.postgresql.*

hdx_python_database-1.3.5/requirements.txt

@@ -1,62 +0,0 @@
-# This file was autogenerated by uv via the following command:
-#    uv pip compile pyproject.toml --resolver=backtracking --all-extras -o requirements.txt
-bcrypt==4.2.1
-    # via paramiko
-cffi==1.17.1
-    # via
-    #   cryptography
-    #   pynacl
-cfgv==3.4.0
-    # via pre-commit
-coverage==7.6.11
-    # via pytest-cov
-cryptography==44.0.0
-    # via paramiko
-distlib==0.3.9
-    # via virtualenv
-filelock==3.17.0
-    # via virtualenv
-greenlet==3.1.1
-    # via sqlalchemy
-identify==2.6.7
-    # via pre-commit
-iniconfig==2.0.0
-    # via pytest
-nodeenv==1.9.1
-    # via pre-commit
-packaging==24.2
-    # via pytest
-paramiko==3.5.1
-    # via sshtunnel
-platformdirs==4.3.6
-    # via virtualenv
-pluggy==1.5.0
-    # via pytest
-pre-commit==4.1.0
-    # via hdx-python-database (pyproject.toml)
-psycopg==3.2.4
-    # via hdx-python-database (pyproject.toml)
-psycopg-binary==3.2.4
-    # via psycopg
-pycparser==2.22
-    # via cffi
-pynacl==1.5.0
-    # via paramiko
-pytest==8.3.4
-    # via
-    #   hdx-python-database (pyproject.toml)
-    #   pytest-cov
-pytest-cov==6.0.0
-    # via hdx-python-database (pyproject.toml)
-pyyaml==6.0.2
-    # via pre-commit
-sqlalchemy==2.0.38
-    # via hdx-python-database (pyproject.toml)
-sshtunnel==0.4.0
-    # via hdx-python-database (pyproject.toml)
-typing-extensions==4.12.2
-    # via
-    #   psycopg
-    #   sqlalchemy
-virtualenv==20.29.1
-    # via pre-commit