diff --git a/lnbits/db.py b/lnbits/db.py index 746735479..11548ac53 100644 --- a/lnbits/db.py +++ b/lnbits/db.py @@ -1,6 +1,6 @@ import os import trio -import psycopg2 +from typing import Optional from contextlib import asynccontextmanager from sqlalchemy import create_engine # type: ignore from sqlalchemy_aio import TRIO_STRATEGY # type: ignore @@ -13,8 +13,8 @@ SQLITE = "SQLITE" class Compat: - type = "" - schema = "" + type: Optional[str] = "" + schema: Optional[str] = "" def interval_seconds(self, seconds: int) -> str: if self.type == POSTGRES: @@ -84,6 +84,8 @@ class Database(Compat): database_uri = LNBITS_DATABASE_URL self.type = POSTGRES + import psycopg2 # type: ignore + DEC2FLOAT = psycopg2.extensions.new_type( psycopg2.extensions.DECIMAL.values, "DEC2FLOAT",