aiolabs/lnbits
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

psycopg2 only imported when postgres.

Browse source
This commit is contained in:
fiatjaf 2021-07-01 13:09:02 -03:00
parent e2d43e786f
commit 46eab3b35e
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
lnbits/db.py
View file

@ -1,6 +1,6 @@
import os import os
import trio import trio
import psycopg2 from typing import Optional
from contextlib import asynccontextmanager from contextlib import asynccontextmanager
from sqlalchemy import create_engine # type: ignore from sqlalchemy import create_engine # type: ignore
from sqlalchemy_aio import TRIO_STRATEGY # type: ignore from sqlalchemy_aio import TRIO_STRATEGY # type: ignore
@ -13,8 +13,8 @@ SQLITE = "SQLITE"
class Compat: class Compat:
type = "<inherited>" type: Optional[str] = "<inherited>"
schema = "<inherited>" schema: Optional[str] = "<inherited>"
def interval_seconds(self, seconds: int) -> str: def interval_seconds(self, seconds: int) -> str:
if self.type == POSTGRES: if self.type == POSTGRES:
@ -84,6 +84,8 @@ class Database(Compat):
database_uri = LNBITS_DATABASE_URL database_uri = LNBITS_DATABASE_URL
self.type = POSTGRES self.type = POSTGRES
import psycopg2 # type: ignore
DEC2FLOAT = psycopg2.extensions.new_type( DEC2FLOAT = psycopg2.extensions.new_type(
psycopg2.extensions.DECIMAL.values, psycopg2.extensions.DECIMAL.values,
"DEC2FLOAT", "DEC2FLOAT",