mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-24 14:51:05 +01:00
8079325ec9
Requests to `POST /usermanager/api/v1/extensions` would silently fail due to a missing `await`.
127 lines
4 KiB
Python
127 lines
4 KiB
Python
from http import HTTPStatus
|
|
|
|
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, require_admin_key
|
|
|
|
from . import usermanager_ext
|
|
from .crud import (
|
|
create_usermanager_user,
|
|
create_usermanager_wallet,
|
|
delete_usermanager_user,
|
|
delete_usermanager_wallet,
|
|
get_usermanager_user,
|
|
get_usermanager_users,
|
|
get_usermanager_users_wallets,
|
|
get_usermanager_wallet,
|
|
get_usermanager_wallet_transactions,
|
|
get_usermanager_wallets,
|
|
)
|
|
from .models import CreateUserData, CreateUserWallet
|
|
|
|
# Users
|
|
|
|
|
|
@usermanager_ext.get("/api/v1/users", status_code=HTTPStatus.OK)
|
|
async def api_usermanager_users(wallet: WalletTypeInfo = Depends(require_admin_key)):
|
|
user_id = wallet.wallet.user
|
|
return [user.dict() for user in await get_usermanager_users(user_id)]
|
|
|
|
|
|
@usermanager_ext.get("/api/v1/users/{user_id}", status_code=HTTPStatus.OK)
|
|
async def api_usermanager_user(user_id, wallet: WalletTypeInfo = Depends(get_key_type)):
|
|
user = await get_usermanager_user(user_id)
|
|
return user.dict()
|
|
|
|
|
|
@usermanager_ext.post("/api/v1/users", status_code=HTTPStatus.CREATED)
|
|
async def api_usermanager_users_create(
|
|
data: CreateUserData, wallet: WalletTypeInfo = Depends(get_key_type)
|
|
):
|
|
user = await create_usermanager_user(data)
|
|
full = user.dict()
|
|
full["wallets"] = [
|
|
wallet.dict() for wallet in await get_usermanager_users_wallets(user.id)
|
|
]
|
|
return full
|
|
|
|
|
|
@usermanager_ext.delete("/api/v1/users/{user_id}")
|
|
async def api_usermanager_users_delete(
|
|
user_id, wallet: WalletTypeInfo = Depends(require_admin_key)
|
|
):
|
|
user = await get_usermanager_user(user_id)
|
|
if not user:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="User does not exist."
|
|
)
|
|
await delete_usermanager_user(user_id)
|
|
raise HTTPException(status_code=HTTPStatus.NO_CONTENT)
|
|
|
|
|
|
# Activate Extension
|
|
|
|
|
|
@usermanager_ext.post("/api/v1/extensions")
|
|
async def api_usermanager_activate_extension(
|
|
extension: str = Query(...), userid: str = Query(...), active: bool = Query(...)
|
|
):
|
|
user = await get_user(userid)
|
|
if not user:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="User does not exist."
|
|
)
|
|
await update_user_extension(user_id=userid, extension=extension, active=active)
|
|
return {"extension": "updated"}
|
|
|
|
|
|
# Wallets
|
|
|
|
|
|
@usermanager_ext.post("/api/v1/wallets")
|
|
async def api_usermanager_wallets_create(
|
|
data: CreateUserWallet, wallet: WalletTypeInfo = Depends(get_key_type)
|
|
):
|
|
user = await create_usermanager_wallet(
|
|
user_id=data.user_id, wallet_name=data.wallet_name, admin_id=data.admin_id
|
|
)
|
|
return user.dict()
|
|
|
|
|
|
@usermanager_ext.get("/api/v1/wallets")
|
|
async def api_usermanager_wallets(wallet: WalletTypeInfo = Depends(require_admin_key)):
|
|
admin_id = wallet.wallet.user
|
|
return [wallet.dict() for wallet in await get_usermanager_wallets(admin_id)]
|
|
|
|
|
|
@usermanager_ext.get("/api/v1/transactions/{wallet_id}")
|
|
async def api_usermanager_wallet_transactions(
|
|
wallet_id, wallet: WalletTypeInfo = Depends(get_key_type)
|
|
):
|
|
return await get_usermanager_wallet_transactions(wallet_id)
|
|
|
|
|
|
@usermanager_ext.get("/api/v1/wallets/{user_id}")
|
|
async def api_usermanager_users_wallets(
|
|
user_id, wallet: WalletTypeInfo = Depends(require_admin_key)
|
|
):
|
|
return [
|
|
s_wallet.dict() for s_wallet in await get_usermanager_users_wallets(user_id)
|
|
]
|
|
|
|
|
|
@usermanager_ext.delete("/api/v1/wallets/{wallet_id}")
|
|
async def api_usermanager_wallets_delete(
|
|
wallet_id, wallet: WalletTypeInfo = Depends(require_admin_key)
|
|
):
|
|
get_wallet = await get_usermanager_wallet(wallet_id)
|
|
if not get_wallet:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="Wallet does not exist."
|
|
)
|
|
await delete_usermanager_wallet(wallet_id, get_wallet.user)
|
|
raise HTTPException(status_code=HTTPStatus.NO_CONTENT)
|