style and type-check

README.md

@@ -123,7 +123,6 @@ In case of database structure upgrade, developers must follow these steps:
 
 For details about the alembic migration tool, see the [Alembic tutorial](https://alembic.sqlalchemy.org/en/latest/tutorial.html).
 
-
 ## License
 
 ```

alembic/README

@@ -1 +1 @@
-Generic single-database configuration.
+Generic single-database configuration.

alembic/env.py

@@ -52,14 +52,14 @@ def run_migrations_online() -> None:
         alembic.context.configure(
             connection=connection,
             target_metadata=cacholote.database.Base.metadata,
-            version_table='alembic_version_cacholote'
+            version_table="alembic_version_cacholote",
         )
 
         with alembic.context.begin_transaction():
             alembic.context.run_migrations()
 
 
-if context.is_offline_mode():
+if alembic.context.is_offline_mode():
     run_migrations_offline()
 else:
     run_migrations_online()

alembic/versions/a38663d192e5_add_created_at_and_rename_timestamp_.py

@@ -1,20 +1,20 @@
 """add created_at and rename timestamp column.
 
 Revision ID: a38663d192e5
-Revises: 
+Revises:
 Create Date: 2024-07-24 15:53:43.989464
 
 """
-import datetime
+
 from typing import Sequence, Union
 
 import sqlalchemy as sa
 
-from alembic import op
 import cacholote
+from alembic import op
 
 # revision identifiers, used by Alembic.
-revision: str = 'a38663d192e5'
+revision: str = "a38663d192e5"
 down_revision: Union[str, None] = None
 branch_labels: Union[str, Sequence[str], None] = None
 depends_on: Union[str, Sequence[str], None] = None

cacholote/database.py

@@ -18,11 +18,16 @@
 import datetime
 import functools
 import json
+import os
 import warnings
 from typing import Any
 
 import sqlalchemy as sa
 import sqlalchemy.orm
+import sqlalchemy_utils
+
+import alembic.command
+import alembic.config
 
 from . import utils
 
@@ -116,7 +121,6 @@ def cached_sessionmaker(url: str, **kwargs: Any) -> sa.orm.sessionmaker[sa.orm.S
     return _cached_sessionmaker(url, **_encode_kwargs(**kwargs))
 
 
-
 def init_database(connection_string: str, force: bool = False) -> sa.engine.Engine:
     """
     Make sure the db located at URI `connection_string` exists updated and return the engine object.

ci/environment-ci.yml

@@ -33,3 +33,4 @@ dependencies:
 - zarr
 - pip:
   - pytest-structlog
+  - types-sqlalchemy-utils

environment.yml

@@ -13,4 +13,5 @@ dependencies:
 - pydantic
 - pydantic-settings
 - sqlalchemy>=2.0.9
+- sqlalchemy-utils
 - structlog

pyproject.toml

@@ -22,6 +22,7 @@ dependencies = [
   "pydantic",
   "pydantic-settings",
   "sqlalchemy>=2.0.9",
+  "sqlalchemy_utils",
   "structlog"
 ]
 description = "Efficiently cache calls to functions"