Michilis/lnbits
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

psycopg2 only imported when postgres.

Browse Source
This commit is contained in:
fiatjaf
2021-07-02 17:34:33 -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
lnbits/db.py
+5 -3
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",