From 70885ce160164b0d9c5d2db3f5d687691e14eb35 Mon Sep 17 00:00:00 2001 From: Tiago vasconcelos Date: Mon, 18 Oct 2021 10:58:09 +0100 Subject: [PATCH] clean up (black, sort) --- lnbits/extensions/lndhub/__init__.py | 13 +++++------ lnbits/extensions/lndhub/decorators.py | 3 --- lnbits/extensions/lndhub/views.py | 2 +- lnbits/extensions/lndhub/views_api.py | 7 ------ lnbits/extensions/tpos/__init__.py | 4 ---- lnbits/extensions/tpos/crud.py | 2 +- lnbits/extensions/tpos/models.py | 2 +- lnbits/extensions/tpos/views.py | 14 +++++------- lnbits/extensions/tpos/views_api.py | 10 +++------ lnbits/extensions/usermanager/__init__.py | 5 +---- lnbits/extensions/usermanager/crud.py | 14 ++++-------- lnbits/extensions/usermanager/models.py | 5 +++-- lnbits/extensions/usermanager/views.py | 3 +-- lnbits/extensions/usermanager/views_api.py | 19 ++++++++-------- lnbits/extensions/watchonly/__init__.py | 4 +--- lnbits/extensions/watchonly/crud.py | 9 ++++---- lnbits/extensions/watchonly/models.py | 1 + lnbits/extensions/watchonly/views.py | 8 +++---- lnbits/extensions/watchonly/views_api.py | 26 ++++++---------------- lnbits/extensions/withdraw/__init__.py | 5 ++--- lnbits/extensions/withdraw/crud.py | 3 ++- lnbits/extensions/withdraw/lnurl.py | 12 +++++----- lnbits/extensions/withdraw/models.py | 10 +++++---- lnbits/extensions/withdraw/views.py | 19 ++++++++-------- lnbits/extensions/withdraw/views_api.py | 23 +++++++------------ 25 files changed, 85 insertions(+), 138 deletions(-) diff --git a/lnbits/extensions/lndhub/__init__.py b/lnbits/extensions/lndhub/__init__.py index cdd363fd..5980ab0d 100644 --- a/lnbits/extensions/lndhub/__init__.py +++ b/lnbits/extensions/lndhub/__init__.py @@ -1,10 +1,7 @@ -import asyncio -from fastapi import APIRouter, FastAPI -from fastapi.staticfiles import StaticFiles -from starlette.routing import Mount +from fastapi import APIRouter + from lnbits.db import Database from lnbits.helpers import template_renderer -from lnbits.tasks import catch_everything_and_restart db = Database("ext_lndhub") @@ -15,7 +12,7 @@ def lndhub_renderer(): return template_renderer(["lnbits/extensions/lndhub/templates"]) -from .views_api import * # noqa -from .views import * # noqa -from .utils import * # noqa from .decorators import * # noqa +from .utils import * # noqa +from .views import * # noqa +from .views_api import * # noqa diff --git a/lnbits/extensions/lndhub/decorators.py b/lnbits/extensions/lndhub/decorators.py index 8cb765ff..14931164 100644 --- a/lnbits/extensions/lndhub/decorators.py +++ b/lnbits/extensions/lndhub/decorators.py @@ -1,13 +1,10 @@ from base64 import b64decode -from functools import wraps from fastapi.param_functions import Security from fastapi.security.api_key import APIKeyHeader -from lnbits.core.crud import get_wallet_for_key from fastapi import Request, status from starlette.exceptions import HTTPException -from starlette.responses import HTMLResponse, JSONResponse from lnbits.decorators import WalletTypeInfo, get_key_type # type: ignore diff --git a/lnbits/extensions/lndhub/views.py b/lnbits/extensions/lndhub/views.py index 275c692e..4b015c09 100644 --- a/lnbits/extensions/lndhub/views.py +++ b/lnbits/extensions/lndhub/views.py @@ -1,6 +1,6 @@ from lnbits.decorators import check_user_exists from . import lndhub_ext, lndhub_renderer -from fastapi import FastAPI, Request +from fastapi import Request from fastapi.params import Depends from lnbits.core.models import User diff --git a/lnbits/extensions/lndhub/views_api.py b/lnbits/extensions/lndhub/views_api.py index 7079703c..2df4141d 100644 --- a/lnbits/extensions/lndhub/views_api.py +++ b/lnbits/extensions/lndhub/views_api.py @@ -13,12 +13,8 @@ from .decorators import check_wallet, require_admin_key from .utils import to_buffer, decoded_as_lndhub from http import HTTPStatus from starlette.exceptions import HTTPException -from starlette.responses import HTMLResponse, JSONResponse # type: ignore -from starlette.requests import Request -from fastapi import Body from fastapi.params import Depends from fastapi.param_functions import Query -from fastapi.security import OAuth2PasswordBearer @lndhub_ext.get("/ext/getinfo") @@ -111,13 +107,11 @@ async def lndhub_payinvoice( @lndhub_ext.get("/ext/balance") -# @check_wallet() async def lndhub_balance(wallet: WalletTypeInfo = Depends(check_wallet),): return {"BTC": {"AvailableBalance": wallet.wallet.balance}} @lndhub_ext.get("/ext/gettxs") -# @check_wallet() async def lndhub_gettxs( wallet: WalletTypeInfo = Depends(check_wallet), limit: int = Query(0, ge=0, lt=200) ): @@ -208,7 +202,6 @@ async def lndhub_getbtc(wallet: WalletTypeInfo = Depends(check_wallet)): @lndhub_ext.get("/ext/getpending") -# @check_wallet() async def lndhub_getpending(wallet: WalletTypeInfo = Depends(check_wallet)): "pending onchain transactions" return [] diff --git a/lnbits/extensions/tpos/__init__.py b/lnbits/extensions/tpos/__init__.py index eb5affe0..6caf88fa 100644 --- a/lnbits/extensions/tpos/__init__.py +++ b/lnbits/extensions/tpos/__init__.py @@ -1,17 +1,13 @@ -import asyncio - from fastapi import APIRouter from lnbits.db import Database from lnbits.helpers import template_renderer -from lnbits.tasks import catch_everything_and_restart db = Database("ext_tpos") tpos_ext: APIRouter = APIRouter( prefix="/tpos", tags=["TPoS"] - # "tpos", __name__, static_folder="static", template_folder="templates" ) diff --git a/lnbits/extensions/tpos/crud.py b/lnbits/extensions/tpos/crud.py index 25a65904..1a198769 100644 --- a/lnbits/extensions/tpos/crud.py +++ b/lnbits/extensions/tpos/crud.py @@ -3,7 +3,7 @@ from typing import List, Optional, Union from lnbits.helpers import urlsafe_short_hash from . import db -from .models import TPoS, CreateTposData +from .models import CreateTposData, TPoS async def create_tpos(wallet_id: str, data: CreateTposData) -> TPoS: diff --git a/lnbits/extensions/tpos/models.py b/lnbits/extensions/tpos/models.py index 7d5f9964..653a055c 100644 --- a/lnbits/extensions/tpos/models.py +++ b/lnbits/extensions/tpos/models.py @@ -1,5 +1,5 @@ from sqlite3 import Row -from fastapi.param_functions import Query + from pydantic import BaseModel diff --git a/lnbits/extensions/tpos/views.py b/lnbits/extensions/tpos/views.py index 80c53684..2d78ecce 100644 --- a/lnbits/extensions/tpos/views.py +++ b/lnbits/extensions/tpos/views.py @@ -1,22 +1,21 @@ +from http import HTTPStatus + +from fastapi import Request +from fastapi.params import Depends +from fastapi.templating import Jinja2Templates from starlette.exceptions import HTTPException from starlette.responses import HTMLResponse + from lnbits.core.models import User -from lnbits.core.crud import get_wallet from lnbits.decorators import check_user_exists -from http import HTTPStatus from . import tpos_ext, tpos_renderer from .crud import get_tpos -from fastapi import FastAPI, Request -from fastapi.params import Depends -from fastapi.templating import Jinja2Templates templates = Jinja2Templates(directory="templates") @tpos_ext.get("/", response_class=HTMLResponse) -# @validate_uuids(["usr"], required=True) -# @check_user_exists() async def index(request: Request, user: User = Depends(check_user_exists)): return tpos_renderer().TemplateResponse( "tpos/index.html", {"request": request, "user": user.dict()} @@ -30,7 +29,6 @@ async def tpos(request: Request, tpos_id): raise HTTPException( status_code=HTTPStatus.NOT_FOUND, detail="TPoS does not exist." ) - # abort(HTTPStatus.NOT_FOUND, "TPoS does not exist.") return tpos_renderer().TemplateResponse( "tpos/tpos.html", {"request": request, "tpos": tpos} diff --git a/lnbits/extensions/tpos/views_api.py b/lnbits/extensions/tpos/views_api.py index 197fe168..8d640a8a 100644 --- a/lnbits/extensions/tpos/views_api.py +++ b/lnbits/extensions/tpos/views_api.py @@ -2,19 +2,15 @@ from http import HTTPStatus from fastapi import Query from fastapi.params import Depends - -from pydantic import BaseModel from starlette.exceptions import HTTPException -from starlette.requests import Request -from starlette.responses import HTMLResponse, JSONResponse # type: ignore from lnbits.core.crud import get_user, get_wallet -from lnbits.core.services import create_invoice, check_invoice_status +from lnbits.core.services import check_invoice_status, create_invoice from lnbits.decorators import WalletTypeInfo, get_key_type from . import tpos_ext -from .crud import create_tpos, get_tpos, get_tposs, delete_tpos -from .models import TPoS, CreateTposData +from .crud import create_tpos, delete_tpos, get_tpos, get_tposs +from .models import CreateTposData @tpos_ext.get("/api/v1/tposs", status_code=HTTPStatus.OK) diff --git a/lnbits/extensions/usermanager/__init__.py b/lnbits/extensions/usermanager/__init__.py index 2e3d6d5a..915948a2 100644 --- a/lnbits/extensions/usermanager/__init__.py +++ b/lnbits/extensions/usermanager/__init__.py @@ -1,5 +1,3 @@ -import asyncio - from fastapi import APIRouter from lnbits.db import Database @@ -10,7 +8,6 @@ db = Database("ext_usermanager") usermanager_ext: APIRouter = APIRouter( prefix="/usermanager", tags=["usermanager"] - # "usermanager", __name__, static_folder="static", template_folder="templates" ) @@ -18,5 +15,5 @@ def usermanager_renderer(): return template_renderer(["lnbits/extensions/usermanager/templates"]) -from .views_api import * # noqa from .views import * # noqa +from .views_api import * # noqa diff --git a/lnbits/extensions/usermanager/crud.py b/lnbits/extensions/usermanager/crud.py index f27acd68..4272d726 100644 --- a/lnbits/extensions/usermanager/crud.py +++ b/lnbits/extensions/usermanager/crud.py @@ -1,17 +1,11 @@ -from typing import Optional, List +from typing import List, Optional +from lnbits.core.crud import (create_account, create_wallet, delete_wallet, + get_payments, get_user) from lnbits.core.models import Payment -from lnbits.core.crud import ( - create_account, - get_user, - get_payments, - create_wallet, - delete_wallet, -) from . import db -from .models import Users, Wallets, CreateUserData - +from .models import CreateUserData, Users, Wallets ### Users diff --git a/lnbits/extensions/usermanager/models.py b/lnbits/extensions/usermanager/models.py index 4dbe4389..a0845233 100644 --- a/lnbits/extensions/usermanager/models.py +++ b/lnbits/extensions/usermanager/models.py @@ -1,7 +1,8 @@ -from pydantic import BaseModel -from fastapi.param_functions import Query from sqlite3 import Row +from fastapi.param_functions import Query +from pydantic import BaseModel + class CreateUserData(BaseModel): user_name: str = Query(...) diff --git a/lnbits/extensions/usermanager/views.py b/lnbits/extensions/usermanager/views.py index 5faec4db..420669b0 100644 --- a/lnbits/extensions/usermanager/views.py +++ b/lnbits/extensions/usermanager/views.py @@ -1,6 +1,5 @@ -from fastapi import FastAPI, Request +from fastapi import Request from fastapi.params import Depends -from fastapi.templating import Jinja2Templates from starlette.responses import HTMLResponse from lnbits.core.models import User diff --git a/lnbits/extensions/usermanager/views_api.py b/lnbits/extensions/usermanager/views_api.py index 08d0e4f5..7bdee8fc 100644 --- a/lnbits/extensions/usermanager/views_api.py +++ b/lnbits/extensions/usermanager/views_api.py @@ -1,28 +1,27 @@ from http import HTTPStatus -from starlette.exceptions import HTTPException from fastapi import Query from fastapi.params import Depends +from starlette.exceptions import HTTPException +from lnbits.core import update_user_extension from lnbits.core.crud import get_user from lnbits.decorators import WalletTypeInfo, get_key_type from . import usermanager_ext -from .models import CreateUserData from .crud import ( create_usermanager_user, + create_usermanager_wallet, + delete_usermanager_user, + delete_usermanager_wallet, get_usermanager_user, get_usermanager_users, - get_usermanager_wallet_transactions, - delete_usermanager_user, - create_usermanager_wallet, - get_usermanager_wallet, - get_usermanager_wallets, get_usermanager_users_wallets, - delete_usermanager_wallet, + get_usermanager_wallet, + get_usermanager_wallet_transactions, + get_usermanager_wallets, ) -from lnbits.core import update_user_extension - +from .models import CreateUserData ### Users diff --git a/lnbits/extensions/watchonly/__init__.py b/lnbits/extensions/watchonly/__init__.py index 9cd4d69a..0c525980 100644 --- a/lnbits/extensions/watchonly/__init__.py +++ b/lnbits/extensions/watchonly/__init__.py @@ -1,5 +1,3 @@ -import asyncio - from fastapi import APIRouter from lnbits.db import Database @@ -15,5 +13,5 @@ def watchonly_renderer(): return template_renderer(["lnbits/extensions/watchonly/templates"]) -from .views_api import * # noqa from .views import * # noqa +from .views_api import * # noqa diff --git a/lnbits/extensions/watchonly/crud.py b/lnbits/extensions/watchonly/crud.py index f9b92c2f..68b59da6 100644 --- a/lnbits/extensions/watchonly/crud.py +++ b/lnbits/extensions/watchonly/crud.py @@ -1,14 +1,13 @@ from typing import List, Optional -from . import db -from .models import Wallets, Addresses, Mempool - -from lnbits.helpers import urlsafe_short_hash - from embit.descriptor import Descriptor, Key # type: ignore from embit.descriptor.arguments import AllowedDerivation # type: ignore from embit.networks import NETWORKS # type: ignore +from lnbits.helpers import urlsafe_short_hash + +from . import db +from .models import Addresses, Mempool, Wallets ##########################WALLETS#################### diff --git a/lnbits/extensions/watchonly/models.py b/lnbits/extensions/watchonly/models.py index aed4ca61..d0894097 100644 --- a/lnbits/extensions/watchonly/models.py +++ b/lnbits/extensions/watchonly/models.py @@ -1,4 +1,5 @@ from sqlite3 import Row + from fastapi.param_functions import Query from pydantic import BaseModel diff --git a/lnbits/extensions/watchonly/views.py b/lnbits/extensions/watchonly/views.py index 37f8b089..e0646651 100644 --- a/lnbits/extensions/watchonly/views.py +++ b/lnbits/extensions/watchonly/views.py @@ -1,8 +1,7 @@ -from http import HTTPStatus -from starlette.exceptions import HTTPException -from starlette.responses import HTMLResponse -from starlette.requests import Request from fastapi.params import Depends +from fastapi.templating import Jinja2Templates +from starlette.requests import Request +from starlette.responses import HTMLResponse from lnbits.core.models import User from lnbits.decorators import check_user_exists @@ -11,7 +10,6 @@ from . import watchonly_ext, watchonly_renderer # from .crud import get_payment -from fastapi.templating import Jinja2Templates templates = Jinja2Templates(directory="templates") diff --git a/lnbits/extensions/watchonly/views_api.py b/lnbits/extensions/watchonly/views_api.py index c457f06f..61978b74 100644 --- a/lnbits/extensions/watchonly/views_api.py +++ b/lnbits/extensions/watchonly/views_api.py @@ -1,29 +1,17 @@ -import hashlib from http import HTTPStatus -import httpx -import json - -from lnbits.core.crud import get_user -from lnbits.decorators import WalletTypeInfo, get_key_type from fastapi import Query from fastapi.params import Depends from starlette.exceptions import HTTPException -from .models import CreateWallet +from lnbits.core.crud import get_user +from lnbits.decorators import WalletTypeInfo, get_key_type from lnbits.extensions.watchonly import watchonly_ext -from .crud import ( - create_watch_wallet, - get_watch_wallet, - get_watch_wallets, - update_watch_wallet, - delete_watch_wallet, - get_fresh_address, - get_addresses, - create_mempool, - update_mempool, - get_mempool, -) + +from .crud import (create_mempool, create_watch_wallet, delete_watch_wallet, + get_addresses, get_fresh_address, get_mempool, + get_watch_wallet, get_watch_wallets, update_mempool) +from .models import CreateWallet ###################WALLETS############################# diff --git a/lnbits/extensions/withdraw/__init__.py b/lnbits/extensions/withdraw/__init__.py index 98efd0bf..b756aeea 100644 --- a/lnbits/extensions/withdraw/__init__.py +++ b/lnbits/extensions/withdraw/__init__.py @@ -26,10 +26,9 @@ def withdraw_renderer(): return template_renderer(["lnbits/extensions/withdraw/templates"]) -from .views_api import * # noqa -from .views import * # noqa from .lnurl import * # noqa - +from .views import * # noqa +from .views_api import * # noqa # @withdraw_ext.on_event("startup") # def _do_it(): diff --git a/lnbits/extensions/withdraw/crud.py b/lnbits/extensions/withdraw/crud.py index a060c722..01a841cf 100644 --- a/lnbits/extensions/withdraw/crud.py +++ b/lnbits/extensions/withdraw/crud.py @@ -1,9 +1,10 @@ from datetime import datetime from typing import List, Optional, Union + from lnbits.helpers import urlsafe_short_hash from . import db -from .models import WithdrawLink, HashCheck, CreateWithdrawData +from .models import CreateWithdrawData, HashCheck, WithdrawLink async def create_withdraw_link( diff --git a/lnbits/extensions/withdraw/lnurl.py b/lnbits/extensions/withdraw/lnurl.py index 1429d0d2..af3ecff4 100644 --- a/lnbits/extensions/withdraw/lnurl.py +++ b/lnbits/extensions/withdraw/lnurl.py @@ -1,16 +1,16 @@ -from fastapi.param_functions import Query -from fastapi import HTTPException -import shortuuid # type: ignore -from http import HTTPStatus from datetime import datetime +from http import HTTPStatus + +import shortuuid # type: ignore +from fastapi import HTTPException +from fastapi.param_functions import Query +from starlette.requests import Request from lnbits.core.services import pay_invoice -from starlette.requests import Request from . import withdraw_ext from .crud import get_withdraw_link_by_hash, update_withdraw_link - # FOR LNURLs WHICH ARE NOT UNIQUE diff --git a/lnbits/extensions/withdraw/models.py b/lnbits/extensions/withdraw/models.py index e2b057fd..8de38e38 100644 --- a/lnbits/extensions/withdraw/models.py +++ b/lnbits/extensions/withdraw/models.py @@ -1,9 +1,11 @@ -from starlette.requests import Request -from fastapi.param_functions import Query -from lnurl import Lnurl, LnurlWithdrawResponse, encode as lnurl_encode # type: ignore from sqlite3 import Row -from pydantic import BaseModel + import shortuuid # type: ignore +from fastapi.param_functions import Query +from lnurl import Lnurl, LnurlWithdrawResponse +from lnurl import encode as lnurl_encode # type: ignore +from pydantic import BaseModel +from starlette.requests import Request class CreateWithdrawData(BaseModel): diff --git a/lnbits/extensions/withdraw/views.py b/lnbits/extensions/withdraw/views.py index 82189c53..6329ce64 100644 --- a/lnbits/extensions/withdraw/views.py +++ b/lnbits/extensions/withdraw/views.py @@ -1,17 +1,18 @@ from http import HTTPStatus -import pyqrcode from io import BytesIO + +import pyqrcode +from fastapi import Request +from fastapi.params import Depends +from fastapi.templating import Jinja2Templates +from starlette.exceptions import HTTPException +from starlette.responses import HTMLResponse, StreamingResponse + +from lnbits.core.models import User from lnbits.decorators import check_user_exists from . import withdraw_ext, withdraw_renderer -from .crud import get_withdraw_link, chunks -from fastapi import FastAPI, Request -from fastapi.params import Depends -from fastapi.templating import Jinja2Templates - -from starlette.exceptions import HTTPException -from starlette.responses import HTMLResponse, StreamingResponse -from lnbits.core.models import User +from .crud import chunks, get_withdraw_link templates = Jinja2Templates(directory="templates") diff --git a/lnbits/extensions/withdraw/views_api.py b/lnbits/extensions/withdraw/views_api.py index 21c01973..678346f9 100644 --- a/lnbits/extensions/withdraw/views_api.py +++ b/lnbits/extensions/withdraw/views_api.py @@ -1,29 +1,22 @@ -from fastapi.params import Depends -from fastapi.param_functions import Query -from pydantic.main import BaseModel - from http import HTTPStatus + +from fastapi.param_functions import Query +from fastapi.params import Depends from lnurl.exceptions import InvalidUrl as LnurlInvalidUrl # type: ignore from starlette.exceptions import HTTPException from starlette.requests import Request -from starlette.responses import HTMLResponse, JSONResponse # type: ignore from lnbits.core.crud import get_user from lnbits.decorators import WalletTypeInfo, get_key_type + +from . import withdraw_ext +from .crud import (create_withdraw_link, + delete_withdraw_link, get_hash_check, get_withdraw_link, + get_withdraw_links, update_withdraw_link) from .models import CreateWithdrawData # from fastapi import FastAPI, Query, Response -from . import withdraw_ext -from .crud import ( - create_withdraw_link, - get_withdraw_link, - get_withdraw_links, - update_withdraw_link, - delete_withdraw_link, - create_hash_check, - get_hash_check, -) @withdraw_ext.get("/api/v1/links", status_code=HTTPStatus.OK)