kiba-core 0.5.3.dev20__tar.gz

kiba_core-0.5.3.dev20/.github/pull_request_template.md

@@ -0,0 +1,19 @@
+<!--- Title format: [Feature | Fix | Task#]: <summary of your changes> -->
+
+## Description
+
+<!--- Describe your changes -->
+<!--- Why is this change required? What problem does it solve? -->
+<!--- If it fixes an open issue, please link to the issue here. -->
+
+## Screenshots:
+
+<!--- If not relevant delete the sub-heading above -->
+
+## Checklist:
+
+<!--- Please put an `x` in all the following boxes that apply to these changes. -->
+
+- [ ] I have updated the CHANGELOG with a summary of my changes
+<!-- - [ ] I have updated the documentation accordingly -->
+<!-- - [ ] My changes have tests around them -->

kiba_core-0.5.3.dev20/.github/workflows/deploy.yml

@@ -0,0 +1,20 @@
+name: Deploy
+on:
+  push:
+    branches: [main]
+jobs:
+  deploy-package-next:
+    runs-on: ubuntu-latest
+    env:
+      DOCKER_IMAGE: package
+      PYPI_PUBLISH_TOKEN: ${{ secrets.PYPI_PUBLISH_TOKEN }}
+    steps:
+      - name: Check out the repo
+        uses: actions/checkout@v2
+        with:
+          # NOTE(krishan711): need full history to calculate difference
+          fetch-depth: 0
+      - name: Build docker image
+        run: docker build --tag $DOCKER_IMAGE .
+      - name: Publish next package to pypi
+        run: docker run -e UV_PUBLISH_TOKEN=$PYPI_PUBLISH_TOKEN $DOCKER_IMAGE /bin/bash -c " make publish-dev"

kiba_core-0.5.3.dev20/.github/workflows/pull-request.yml

@@ -0,0 +1,58 @@
+name: Build
+on:
+  pull_request:
+    branches: ["*"]
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}-pull-request
+  cancel-in-progress: true
+jobs:
+  core-build:
+    runs-on: ubuntu-latest
+    env:
+      DOCKER_IMAGE: core
+    steps:
+      - name: Check out the repo
+        uses: actions/checkout@v2
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v1
+      - name: Build docker image
+        uses: docker/build-push-action@v2
+        with:
+          context: .
+          load: true
+          cache-from: type=gha,scope=core
+          cache-to: type=gha,mode=max,scope=core
+          tags: ${{ env.DOCKER_IMAGE }}
+      - name: Run lint-check
+        run: |
+          docker run --name lint-check $DOCKER_IMAGE make lint-check-ci
+          docker cp lint-check:/app/lint-check-results.json ./lint-check-results.json
+      - name: Upload lint-check results
+        uses: kibalabs/github-action-create-annotations@main
+        with:
+          github-token: ${{ secrets.GITHUB_TOKEN }}
+          json-file-path: ./lint-check-results.json
+          check-name: core-lint-check
+          fail-on-error: false
+      - name: Run type-check
+        run: |
+          docker run --name type-check $DOCKER_IMAGE make type-check-ci
+          docker cp type-check:/app/type-check-results.json ./type-check-results.json
+      - name: Upload type-check results
+        uses: kibalabs/github-action-create-annotations@main
+        with:
+          github-token: ${{ secrets.GITHUB_TOKEN }}
+          json-file-path: ./type-check-results.json
+          check-name: core-type-check
+          fail-on-error: false
+      - name: Run tests
+        run: |
+          docker run --name tests $DOCKER_IMAGE make test-ci
+          # docker cp tests:/app/test-results.json ./test-results.json
+      # - name: Upload test results
+      #   uses: kibalabs/github-action-create-annotations@main
+      #   with:
+      #     github-token: ${{ secrets.GITHUB_TOKEN }}
+      #     json-file-path: ./test-results.json
+      #     check-name: core-tests
+      #     fail-on-error: true

kiba_core-0.5.3.dev20/.github/workflows/release.yml

@@ -0,0 +1,39 @@
+name: Release
+on:
+  push:
+    tags: ["v*"]
+jobs:
+  deploy-package:
+    runs-on: ubuntu-latest
+    env:
+      DOCKER_IMAGE: build
+      PYPI_PUBLISH_TOKEN: ${{ secrets.PYPI_PUBLISH_TOKEN }}
+    steps:
+      - name: Check out the repo
+        uses: actions/checkout@v2
+      - name: Check the package version is the same as the tag
+        run: |
+          version=$(python3 setup.py --version)
+          if [ "${{ github.ref }}" != "refs/tags/v${version}" ]; then
+            echo "version in setup.py (v${version}) is different to the given tag (${{ github.ref }})"
+            exit 1;
+          fi
+      - name: Build docker image
+        run: docker build --tag $DOCKER_IMAGE .
+      - name: Publish package to pypi
+        run: docker run -e UV_PUBLISH_TOKEN=$PYPI_PUBLISH_TOKEN $DOCKER_IMAGE /bin/bash -c " make publish"
+  create-release:
+    needs: deploy-package
+    runs-on: ubuntu-latest
+    env:
+      GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+    steps:
+      - name: Check out the repo
+        uses: actions/checkout@v2
+      - name: Create Release
+        uses: actions/create-release@v1
+        with:
+          tag_name: ${{ github.ref }}
+          release_name: Release ${{ github.ref }}
+          draft: false
+          prerelease: false

kiba_core-0.5.3.dev20/.gitignore

@@ -0,0 +1,67 @@
+# Mac
+*.DS_Store
+.Spotlight-V100
+.Trashes
+Thumbs.db
+Desktop.ini
+
+# IDEs
+.idea/
+.vscode/
+*.swp
+*.log
+
+# Protobufs
+*_pb2.*
+
+# Testing
+coverage
+.coverage
+tests.xml
+lint.xml
+.noseids
+nosetests.xml
+eslint.xml
+jest.xml
+local_testing*
+
+# Generated
+generated/
+dist/
+build/
+tmp/
+output/
+
+# Python
+__pycache__/
+eggs/
+dist/
+build/
+*.egg-info/
+.env*
+.mypy_cache
+
+# JS
+node_modules/
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+*.d.ts
+.next
+
+# Terraform
+**/.terraform/*
+*.tfstate
+*.tfstate.*
+crash.log
+*.tfvars
+*override.tf
+override.tf.json
+*_override.tf
+*_override.tf.json
+.terraformrc
+terraform.rc
+
+# Ethereum dev
+artifacts/
+cache/

kiba_core-0.5.3.dev20/CHANGELOG.md

@@ -0,0 +1,206 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/) with some additions:
+- For all changes include one of [PATCH | MINOR | MAJOR] with the scope of the change being made.
+
+## [Unreleased]
+
+### Added
+- [MAJOR] Update web3 to 7.8.0
+- [MINOR] Added `call_function_by_name`, `send_transaction_by_name` and `wait_for_transaction_receipt` to `RestEthClient`
+- [MINOR] Updated `RestEthClient` to use `maxPriorityFeePerGas` and `maxFeePerGas` instead of `gasPrice`
+- [MAJOR] Make all datetimes timezone aware
+- [MINOR] Added `Cache`, `FileCache` and `DictCache`
+- [MINOR] Added `url_util.encode_url`, `url_util.encode_url_part`, `url_util.decode_url` and `url_util.decode_url_part`
+- [MINOR] Added `http_util.CACHABLE_STATUS_CODES`
+- [PATCH] Moved `Requester` into a folder with same path as original
+- [MINOR] Added `date_util.datetime_from_date`
+- [MINOR] Add `json_util.dumps` and `json_util.loads` to use faster json serialization
+
+### Changed
+
+### Removed
+
+## [0.5.2] - 2024-08-22
+
+### Added
+- [MAJOR] Updated `MessageQueueProcessor` to use a list of `NotificationClients` to send server messages
+- [MAJOR] Updated `SlackClient`  to post `messageText` instead of text
+- [MINOR] Added `NotificationClient.py` abstract class to help send data to multiple platforms
+- [MINOR] Added `DiscordClient.py` to send data discord server
+- [MINOR] Added `calculate_diff_days` and `calculate_diff_years` to `date_util`
+- [MINOR] Added `datetime_to_utc` to `date_util`
+- [MINOR] Added `init_external_loggers` to `logging`
+- [MINOR] Added `end_of_day` to `date_util`
+- [MINOR] Added `aiosqlite engine` to `database`
+- [MINOR] Added core-api extra to server APIs with starlette only, no fastapi:
+    - Added `core.http.jwt.Jwt`
+    - Added `core.http.rest_method.RestMethod`
+    - Added `core.api.api_request.KibaApiRequest`
+    - Added `core.api.api_response.KibaJSONResponse`
+    - Added `core.api.authorizer.authorize_bearer_jwt`
+    - Added `core.api.json_route.json_route`
+
+### Changed
+- [MAJOR] Moved SlackClient to `core.notifications`
+- [MAJOR] Moved DiscordClient to `core.notifications`
+- [MAJOR] Moved NotificationClient to `core.notifications`
+- [MINOR] Updated `Requester` to send data correctly for PUT requests
+- [MINOR] Updated `Requester` to send data correctly for PATCH requests
+
+### Removed
+
+## [0.5.1] - 2023-02-16
+
+### Added
+- [MINOR] Added `call_contract_function` to `EthClientInterface`
+- [MINOR] Added `_insert_record`, `_update_records`, and `_delete_records` to `Saver`
+- [MINOR] Added `core.queues.aqs.AqsMessage` and `core.queues.aqs.AqsMessageQueue` to work with Azure Queues
+- [MINOR] Added `BooleanFieldFilter` to `Retriever`
+
+### Changed
+- [MAJOR] Moved `SqsMessage` and `SqsMessageQueue` to `core.queues.sqs`
+
+## [0.5.0] - 2022-11-08
+
+### Added
+- [MINOR] Added `requestId` to `Message`
+
+### Changed
+- [MINOR] Updated `Saver` to throw `SavingException` when saving fails
+- [MINOR] Added `MessageNeedsReprocessingException` to `MessageQueueProcessor` to support rescheduling a message
+- [MINOR] Use `www-authenticate` as message (if available) in 401 responses
+- [MAJOR] Upgraded to sqlalchemy2
+
+## [0.4.0] - 2022-07-19
+
+### Added
+- [MINOR] Added `head_file` to `S3Manager`
+- [MAJOR] Added exception lists to `exceptions`
+- [MINOR] Added `generate_clock_hour_intervals`, `generate_hourly_intervals`, `generate_datetime_intervals`, `generate_dates_in_range` to `date_util`
+
+### Changed
+- [MINOR] Updated `DatabaseConnectionMiddleware` to only create a transaction for non-GET requests
+- [MAJOR] Updated `Requester` to throw specific exception types
+- [MINOR] Updated `Requester.post_form` and `Requester.make_request` to accept `formFiles` parameter
+- [MINOR] Added `shouldAddCacheHeader` to relevant `RedirectException`s
+
+## [0.3.0] - 2022-04-23
+
+### Added
+- [MINOR] Added `send_messages` to `SqsMessageQueue` for sending multiple messages in one request
+- [MINOR] Added `Database` to make facilitate easy migration from databases package
+- [MINOR] Added `get_block_uncle_count` to `EthClient`
+- [MINOR] Added `shouldFollowRedirects` to `Requester`
+- [MINOR] Added `DatabaseConnectionMiddleware` to manage api database connections
+- [MAJOR] Added `logging` module for custom kiba log formatting
+- [MAJOR] Added `ExceptionHandlingMiddleware` to replace KibaRoute
+- [MAJOR] Added `ServerHeadersMiddleware` to replace KibaRoute
+- [MAJOR] Added `LoggingMiddleware` to replace KibaRoute
+- [MAJOR] Added `requestIdHolder` to `MessageQueueProcessor`
+
+### Changed
+- [MAJOR] Replaced use of databases package in `Saver` and `Retriever`
+- [MINOR] Update `Requester` to follow redirects by default
+- [MAJOR] Replaced use of `boto3` with `aiobotocore` for S3Manager
+- [MAJOR] Replaced use of `boto3` with `aiobotocore` for SqsMessageQueue
+
+### Removed
+- [MAJOR] Removed `KibaRouter` and `KibaRoute` - Use middlewares instead
+
+## [0.2.10] - 2022-01-24
+
+### Changed
+- [MINOR] Updated `RestEthClient` to throw `BadRequestException` for malformed responses
+
+## [0.2.9] - 2022-01-20
+
+### Added
+- [MINOR] Added `shouldHydrateTransactions` to `EthClient.get_block`
+
+## [0.2.8] - 2022-01-17
+
+### Added
+- [MINOR] Added `postDate` to `Message`
+
+### Changed
+- [MINOR] Set `Message.postDate` in `SqsMessageQueue` when posting to queue
+
+## [0.2.7] - 2022-01-15
+
+### Added
+- [MINOR] Added parallel processing to `MessageQueueProcessor.execute_batch`
+
+## [0.2.6] - 2022-01-06
+
+### Changed
+- [PATCH] Fixed `MessageQueueProcessor` to allow limitless processing
+
+## [0.2.5] - 2022-01-06
+
+### Added
+- [MINOR] Added batch processing to `MessageQueueProcessor`
+- [MINOR] Added one-off processing to `MessageQueueProcessor`
+
+## [0.2.4] - 2021-11-19
+
+### Added
+- [MINOR] Added `BURN_ADDRESS` to `chain_util`
+- [MINOR] Added `RedirectException` and subclasses
+
+### Changed
+- [MINOR] Handle converting `RedirectException` to correct response in `KibaRoute`
+
+## [0.2.3] - 2021-10-11
+
+### Added
+- [MINOR] Added `chain_util` with `normalize_address`
+
+### Changed
+- [MINOR] Updated to databases v0.5.1
+- [MINOR] Updated `MessageQueueProcessor` to make `slackClient` optional
+
+## [0.2.2] - 2021-08-21
+
+### Changed
+- [MINOR] Reverted `Message.COMMAND` change from 0.2.1
+- [MINOR] Added `Message.get_command()` to prevent private access lint error
+
+## [0.2.1] - 2021-08-18
+
+### Changed
+- [MINOR] Added `Message.COMMAND` and deprecation note on `Message._COMMAND` to suit linters
+
+## [0.2.0] - 2021-08-08
+
+### Added
+- [MINOR] Added `post_form` to `Requester`
+- [MINOR] Added `BasicAuthentication`
+
+### Changed
+- [MAJOR] Updated `Requester.post_json` to not accept data
+- [MAJOR] Updated `file_util.create_directory` to be async
+
+## [0.1.3] - 2021-07-14
+
+### Added
+- [MINOR] Added `create_directory` to `file_util`
+
+### Changed
+- [MINOR] Fix `Requester` to use dataDict correctly for GET params
+
+## [0.1.2] - 2021-06-09
+
+### Added
+- [MINOR] Added `EthClient`
+
+## [0.1.1] - 2021-05-20
+
+### Added
+- [MINOR] Added `IntegerFieldFilter` and `FloatFieldFilter` to `Saver`
+
+## [0.1.0] - 2021-05-18
+
+Initial Commit

kiba_core-0.5.3.dev20/Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.12.0-slim
+
+RUN apt-get update && apt-get install --yes --no-install-recommends make git
+
+WORKDIR /app
+COPY makefile .
+
+COPY pyproject.toml .
+COPY uv.lock .
+RUN make install
+
+COPY . .
+RUN make type-check

kiba_core-0.5.3.dev20/PKG-INFO

@@ -0,0 +1,40 @@
+Metadata-Version: 2.2
+Name: kiba-core
+Version: 0.5.3.dev20
+Summary: Kiba Labs' python utilities
+Project-URL: repository, https://github.com/kibalabs/core-py
+Requires-Python: ~=3.12
+Description-Content-Type: text/markdown
+Requires-Dist: aiofiles>=24.1.0
+Requires-Dist: pydantic>=2.10.6
+Requires-Dist: orjson>=3.10.15
+Requires-Dist: kiba-build
+Provides-Extra: api
+Requires-Dist: fastapi>=0.115.8; extra == "api"
+Requires-Dist: uvicorn[standard]>=0.34.0; extra == "api"
+Provides-Extra: core-api
+Requires-Dist: starlette>=0.45.3; extra == "core-api"
+Requires-Dist: uvicorn[standard]>=0.34.0; extra == "core-api"
+Provides-Extra: storage
+Requires-Dist: aiobotocore>=2.19.0; extra == "storage"
+Provides-Extra: queues
+Requires-Dist: aiobotocore>=2.19.0; extra == "queues"
+Provides-Extra: queue-sqs
+Requires-Dist: aiobotocore>=2.19.0; extra == "queue-sqs"
+Provides-Extra: queue-aqs
+Requires-Dist: azure-storage-queue[aio]>=12.12.0; extra == "queue-aqs"
+Provides-Extra: database-psql
+Requires-Dist: sqlalchemy[asyncio]>=2.0.38; extra == "database-psql"
+Requires-Dist: asyncpg>=0.30.0; extra == "database-psql"
+Provides-Extra: database-sqlite
+Requires-Dist: sqlalchemy[asyncio]>=2.0.38; extra == "database-sqlite"
+Requires-Dist: aiosqlite>=0.21.0; extra == "database-sqlite"
+Provides-Extra: requester
+Requires-Dist: httpx>=0.28.1; extra == "requester"
+Provides-Extra: web3
+Requires-Dist: web3>=7.8.0; extra == "web3"
+Provides-Extra: types
+Requires-Dist: types-aiobotocore[essential]>=2.19.0; extra == "types"
+Requires-Dist: types-aiofiles>=24.1.0.20241221; extra == "types"
+
+# Core-Py

kiba_core-0.5.3.dev20/README.md

@@ -0,0 +1 @@
+# Core-Py

kiba_core-0.5.3.dev20/core/api/api_request.py

@@ -0,0 +1,15 @@
+import typing
+
+from pydantic import BaseModel
+from starlette.requests import Request
+
+from core.http.basic_authentication import BasicAuthentication
+from core.http.jwt import Jwt
+
+ApiRequestDataType = typing.TypeVar('ApiRequestDataType', bound=BaseModel)
+
+
+class KibaApiRequest(Request, typing.Generic[ApiRequestDataType]):
+    data: ApiRequestDataType
+    authJwt: Jwt | None = None
+    authBasic: BasicAuthentication | None = None

kiba_core-0.5.3.dev20/core/api/api_respnse.py

@@ -0,0 +1,9 @@
+import typing
+
+import orjson
+from starlette.responses import JSONResponse
+
+
+class KibaJSONResponse(JSONResponse):
+    def render(self, content: typing.Any) -> bytes:  # type: ignore[explicit-any]
+        return orjson.dumps(content)

kiba_core-0.5.3.dev20/core/api/authorizer.py

@@ -0,0 +1,45 @@
+import functools
+import typing
+from typing import ParamSpec
+
+from mypy_extensions import Arg
+from pydantic import BaseModel
+
+from core.api.api_request import KibaApiRequest
+from core.exceptions import ForbiddenException
+from core.http.jwt import Jwt
+
+_P = ParamSpec('_P')
+
+
+class Authorizer:
+    async def validate_jwt(self, jwtString: str) -> Jwt:
+        raise NotImplementedError
+
+
+ApiRequest = typing.TypeVar('ApiRequest', bound=BaseModel)
+
+
+def authorize_bearer_jwt(  # type: ignore[explicit-any]
+    authorizer: Authorizer,
+) -> typing.Callable[[typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], typing.Awaitable[typing.Any]]], typing.Callable[_P, typing.Any]]:
+    def decorator(func: typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], typing.Awaitable[typing.Any]]) -> typing.Callable[_P, typing.Any]:  # type: ignore[explicit-any]
+        @functools.wraps(func)
+        async def async_wrapper(request: KibaApiRequest[ApiRequest]) -> typing.Any:  # type: ignore[explicit-any, misc]
+            authorization = request.headers.get('Authorization')
+            if not authorization:
+                raise ForbiddenException(message='AUTH_NOT_PROVIDED')
+            if not authorization.startswith('Bearer '):
+                raise ForbiddenException(message='AUTH_INVALID')
+            jwtString = authorization.replace('Bearer ', '')
+            try:
+                jwt = await authorizer.validate_jwt(jwtString=jwtString)
+            except BaseException:  # noqa: BLE001
+                raise ForbiddenException(message='AUTH_INVALID')
+            request.authJwt = jwt
+            return await func(request=request)
+
+        # TODO(krishan711): figure out correct typing here
+        return async_wrapper  # type: ignore[return-value]
+
+    return decorator

kiba_core-0.5.3.dev20/core/api/default_routes.py

@@ -0,0 +1,14 @@
+from starlette.requests import Request
+from starlette.responses import JSONResponse
+from starlette.responses import Response
+from starlette.routing import BaseRoute
+from starlette.routing import Route
+
+
+def create_default_routes(name: str, version: str, environment: str) -> list[BaseRoute]:
+    async def root(request: Request) -> Response:  # noqa: ARG001
+        return JSONResponse(content={'server': name, 'version': version, 'environment': environment})
+
+    return [
+        Route('/', methods=['GET'], endpoint=root),
+    ]

kiba_core-0.5.3.dev20/core/api/health.py

@@ -0,0 +1,11 @@
+from fastapi import APIRouter
+
+
+def create_api(name: str, version: str, environment: str) -> APIRouter:
+    router = APIRouter()
+
+    @router.get('/')
+    async def root() -> dict[str, str]:
+        return {'server': name, 'version': version, 'environment': environment}
+
+    return router

kiba_core-0.5.3.dev20/core/api/json_route.py

@@ -0,0 +1,57 @@
+import functools
+import typing
+from typing import ParamSpec
+
+from mypy_extensions import Arg
+from pydantic import BaseModel
+from pydantic import ValidationError
+
+from core.api.api_request import KibaApiRequest
+from core.api.api_respnse import KibaJSONResponse
+from core.exceptions import BadRequestException
+from core.exceptions import InternalServerErrorException
+from core.util import json_util
+from core.util.typing_util import JsonObject
+
+_P = ParamSpec('_P')
+
+ApiRequest = typing.TypeVar('ApiRequest', bound=BaseModel)
+ApiResponse = typing.TypeVar('ApiResponse', bound=BaseModel)
+
+
+def json_route(
+    requestType: typing.Type[ApiRequest],
+    responseType: typing.Type[ApiResponse],
+) -> typing.Callable[[typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], typing.Awaitable[ApiResponse]]], typing.Callable[_P, KibaJSONResponse]]:
+    def decorator(func: typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], typing.Awaitable[ApiResponse]]) -> typing.Callable[_P, KibaJSONResponse]:
+        @functools.wraps(func)
+        async def async_wrapper(*args: typing.Any) -> KibaJSONResponse:  # type: ignore[explicit-any, misc]
+            receivedRequest = args[0]
+            pathParams = receivedRequest.path_params
+            # TODO(krishan711): move these to get request only
+            queryParams = receivedRequest.query_params
+            bodyBytes = await args[0].body()
+            if len(bodyBytes) == 0:
+                body: JsonObject = {}
+            else:
+                try:
+                    body = typing.cast(JsonObject, json_util.loads(bodyBytes.decode()))
+                except json_util.JsonDecodeException as exception:
+                    raise BadRequestException(f'Invalid JSON body: {exception}')
+            allParams = {**pathParams, **body, **queryParams}
+            try:
+                requestParams = requestType(**allParams)
+            except ValidationError as exception:
+                validationErrorMessage = ', '.join([f'{".".join([str(value) for value in error["loc"]])}: {error["msg"]}' for error in exception.errors()])
+                raise BadRequestException(f'Invalid request: {validationErrorMessage}')
+            kibaRequest: KibaApiRequest[ApiRequest] = KibaApiRequest(scope=receivedRequest.scope, receive=receivedRequest._receive, send=receivedRequest._send)  # noqa: SLF001
+            kibaRequest.data = requestParams
+            receivedResponse = await func(request=kibaRequest)
+            if not isinstance(receivedResponse, responseType):
+                raise InternalServerErrorException(f'Expected response to be of type {responseType}, got {type(receivedResponse)}')
+            return KibaJSONResponse(content=receivedResponse.model_dump())
+
+        # TODO(krishan711): figure out correct typing here
+        return async_wrapper  # type: ignore[return-value]
+
+    return decorator

kiba_core-0.5.3.dev20/core/api/middleware/database_connection_middleware.py

@@ -0,0 +1,19 @@
+from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.middleware.base import RequestResponseEndpoint
+from starlette.requests import Request
+from starlette.responses import Response
+from starlette.types import ASGIApp
+
+from core.store.database import Database
+
+
+class DatabaseConnectionMiddleware(BaseHTTPMiddleware):
+    def __init__(self, app: ASGIApp, database: Database) -> None:
+        super().__init__(app=app)
+        self.database = database
+
+    async def dispatch(self, request: Request, call_next: RequestResponseEndpoint) -> Response:
+        # isReadonly = request.method in {'GET', 'OPTIONS', 'HEAD'}
+        async with self.database.create_context_connection():
+            response = await call_next(request)
+        return response