diff --git a/lnbits/extensions/usermanager/crud.py b/lnbits/extensions/usermanager/crud.py index 4272d726..1ce66d4f 100644 --- a/lnbits/extensions/usermanager/crud.py +++ b/lnbits/extensions/usermanager/crud.py @@ -1,7 +1,12 @@ from typing import List, Optional -from lnbits.core.crud import (create_account, create_wallet, delete_wallet, - get_payments, get_user) +from lnbits.core.crud import ( + create_account, + create_wallet, + delete_wallet, + get_payments, + get_user, +) from lnbits.core.models import Payment from . import db diff --git a/lnbits/extensions/usermanager/models.py b/lnbits/extensions/usermanager/models.py index a0845233..1dc7b85a 100644 --- a/lnbits/extensions/usermanager/models.py +++ b/lnbits/extensions/usermanager/models.py @@ -11,6 +11,11 @@ class CreateUserData(BaseModel): email: str = Query("") password: str = Query("") +class CreateUserWallet(BaseModel): + user_id: str = Query(...) + wallet_name: str = Query(...) + admin_id: str = Query(...) + class Users(BaseModel): id: str diff --git a/lnbits/extensions/usermanager/views_api.py b/lnbits/extensions/usermanager/views_api.py index 7bdee8fc..d4808a6b 100644 --- a/lnbits/extensions/usermanager/views_api.py +++ b/lnbits/extensions/usermanager/views_api.py @@ -21,7 +21,7 @@ from .crud import ( get_usermanager_wallet_transactions, get_usermanager_wallets, ) -from .models import CreateUserData +from .models import CreateUserData, CreateUserWallet ### Users @@ -93,12 +93,10 @@ async def api_usermanager_activate_extension( @usermanager_ext.post("/api/v1/wallets") async def api_usermanager_wallets_create( - wallet: WalletTypeInfo = Depends(get_key_type), - user_id: str = Query(...), - wallet_name: str = Query(...), - admin_id: str = Query(...), + data: CreateUserWallet, + wallet: WalletTypeInfo = Depends(get_key_type) ): - user = await create_usermanager_wallet(user_id, wallet_name, admin_id) + user = await create_usermanager_wallet(user_id=data.user_id, wallet_name=data.wallet_name, admin_id=data.admin_id) return user.dict()