[REFACTOR] WalletType into enum (#1888)
* [REFACTOR] WalletType into enum - move wallettype models from decorators.py into models.py - use enum instead of int - use HTTPStatus for consistency * Update lnbits/core/models.py Co-authored-by: jackstar12 <62219658+jackstar12@users.noreply.github.com> * use dataclass --------- Co-authored-by: jackstar12 <62219658+jackstar12@users.noreply.github.com>
This commit is contained in:
@@ -38,6 +38,7 @@ from lnbits.core.models import (
|
||||
PaymentFilters,
|
||||
User,
|
||||
Wallet,
|
||||
WalletType,
|
||||
)
|
||||
from lnbits.db import Filters, Page
|
||||
from lnbits.decorators import (
|
||||
@@ -102,7 +103,7 @@ async def health():
|
||||
|
||||
@core_app.get("/api/v1/wallet")
|
||||
async def api_wallet(wallet: WalletTypeInfo = Depends(get_key_type)):
|
||||
if wallet.wallet_type == 0:
|
||||
if wallet.wallet_type == WalletType.admin:
|
||||
return {
|
||||
"id": wallet.wallet.id,
|
||||
"name": wallet.wallet.name,
|
||||
@@ -318,7 +319,7 @@ async def api_payments_create(
|
||||
wallet: WalletTypeInfo = Depends(require_invoice_key),
|
||||
invoiceData: CreateInvoice = Body(...),
|
||||
):
|
||||
if invoiceData.out is True and wallet.wallet_type == 0:
|
||||
if invoiceData.out is True and wallet.wallet_type == WalletType.admin:
|
||||
if not invoiceData.bolt11:
|
||||
raise HTTPException(
|
||||
status_code=HTTPStatus.BAD_REQUEST,
|
||||
|
||||
Reference in New Issue
Block a user