diff --git a/lnbits/extensions/usermanager/crud.py b/lnbits/extensions/usermanager/crud.py index 649888a8..959914f2 100644 --- a/lnbits/extensions/usermanager/crud.py +++ b/lnbits/extensions/usermanager/crud.py @@ -10,12 +10,10 @@ from lnbits.core.crud import ( from lnbits.core.models import Payment from . import db -from .models import CreateUserData, Users, Wallets - -### Users +from .models import CreateUserData, User, Wallet -async def create_usermanager_user(data: CreateUserData) -> Users: +async def create_usermanager_user(data: CreateUserData) -> User: account = await create_account() user = await get_user(account.id) assert user, "Newly created user couldn't be retrieved" @@ -50,17 +48,17 @@ async def create_usermanager_user(data: CreateUserData) -> Users: return user_created -async def get_usermanager_user(user_id: str) -> Optional[Users]: +async def get_usermanager_user(user_id: str) -> Optional[User]: row = await db.fetchone("SELECT * FROM usermanager.users WHERE id = ?", (user_id,)) - return Users(**row) if row else None + return User(**row) if row else None -async def get_usermanager_users(user_id: str) -> List[Users]: +async def get_usermanager_users(user_id: str) -> List[User]: rows = await db.fetchall( "SELECT * FROM usermanager.users WHERE admin = ?", (user_id,) ) - return [Users(**row) for row in rows] + return [User(**row) for row in rows] async def delete_usermanager_user(user_id: str, delete_core: bool = True) -> None: @@ -73,12 +71,9 @@ async def delete_usermanager_user(user_id: str, delete_core: bool = True) -> Non await db.execute("""DELETE FROM usermanager.wallets WHERE "user" = ?""", (user_id,)) -### Wallets - - async def create_usermanager_wallet( user_id: str, wallet_name: str, admin_id: str -) -> Wallets: +) -> Wallet: wallet = await create_wallet(user_id=user_id, wallet_name=wallet_name) await db.execute( """ @@ -92,28 +87,28 @@ async def create_usermanager_wallet( return wallet_created -async def get_usermanager_wallet(wallet_id: str) -> Optional[Wallets]: +async def get_usermanager_wallet(wallet_id: str) -> Optional[Wallet]: row = await db.fetchone( "SELECT * FROM usermanager.wallets WHERE id = ?", (wallet_id,) ) - return Wallets(**row) if row else None + return Wallet(**row) if row else None -async def get_usermanager_wallets(admin_id: str) -> Optional[Wallets]: +async def get_usermanager_wallets(admin_id: str) -> List[Wallet]: rows = await db.fetchall( "SELECT * FROM usermanager.wallets WHERE admin = ?", (admin_id,) ) - return [Wallets(**row) for row in rows] + return [Wallet(**row) for row in rows] -async def get_usermanager_users_wallets(user_id: str) -> Optional[Wallets]: +async def get_usermanager_users_wallets(user_id: str) -> List[Wallet]: rows = await db.fetchall( """SELECT * FROM usermanager.wallets WHERE "user" = ?""", (user_id,) ) - return [Wallets(**row) for row in rows] + return [Wallet(**row) for row in rows] -async def get_usermanager_wallet_transactions(wallet_id: str) -> Optional[Payment]: +async def get_usermanager_wallet_transactions(wallet_id: str) -> List[Payment]: return await get_payments( wallet_id=wallet_id, complete=True, pending=False, outgoing=True, incoming=True ) diff --git a/lnbits/extensions/usermanager/models.py b/lnbits/extensions/usermanager/models.py index 15f50e28..05122cc8 100644 --- a/lnbits/extensions/usermanager/models.py +++ b/lnbits/extensions/usermanager/models.py @@ -19,7 +19,7 @@ class CreateUserWallet(BaseModel): admin_id: str = Query(...) -class Users(BaseModel): +class User(BaseModel): id: str name: str admin: str @@ -27,7 +27,7 @@ class Users(BaseModel): password: Optional[str] = None -class Wallets(BaseModel): +class Wallet(BaseModel): id: str admin: str name: str @@ -36,5 +36,5 @@ class Wallets(BaseModel): inkey: str @classmethod - def from_row(cls, row: Row) -> "Wallets": + def from_row(cls, row: Row) -> "Wallet": return cls(**dict(row)) diff --git a/lnbits/extensions/usermanager/views.py b/lnbits/extensions/usermanager/views.py index 420669b0..814659f7 100644 --- a/lnbits/extensions/usermanager/views.py +++ b/lnbits/extensions/usermanager/views.py @@ -9,7 +9,9 @@ from . import usermanager_ext, usermanager_renderer @usermanager_ext.get("/", response_class=HTMLResponse) -async def index(request: Request, user: User = Depends(check_user_exists)): +async def index( + request: Request, user: User = Depends(check_user_exists) # type: ignore +): return usermanager_renderer().TemplateResponse( "usermanager/index.html", {"request": request, "user": user.dict()} ) diff --git a/lnbits/extensions/usermanager/views_api.py b/lnbits/extensions/usermanager/views_api.py index b1bf8ef8..493a71bc 100644 --- a/lnbits/extensions/usermanager/views_api.py +++ b/lnbits/extensions/usermanager/views_api.py @@ -23,25 +23,31 @@ from .crud import ( ) 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)): +async def api_usermanager_users( + wallet: WalletTypeInfo = Depends(require_admin_key), # type: ignore +): 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)): +@usermanager_ext.get( + "/api/v1/users/{user_id}", + status_code=HTTPStatus.OK, + dependencies=[Depends(get_key_type)], +) +async def api_usermanager_user(user_id): user = await get_usermanager_user(user_id) - return user.dict() + return user.dict() if user else None -@usermanager_ext.post("/api/v1/users", status_code=HTTPStatus.CREATED) -async def api_usermanager_users_create( - data: CreateUserData, wallet: WalletTypeInfo = Depends(get_key_type) -): +@usermanager_ext.post( + "/api/v1/users", + status_code=HTTPStatus.CREATED, + dependencies=[Depends(get_key_type)], +) +async def api_usermanager_users_create(data: CreateUserData): user = await create_usermanager_user(data) full = user.dict() full["wallets"] = [ @@ -50,11 +56,12 @@ async def api_usermanager_users_create( return full -@usermanager_ext.delete("/api/v1/users/{user_id}") +@usermanager_ext.delete( + "/api/v1/users/{user_id}", dependencies=[Depends(require_admin_key)] +) async def api_usermanager_users_delete( user_id, delete_core: bool = Query(True), - wallet: WalletTypeInfo = Depends(require_admin_key), ): user = await get_usermanager_user(user_id) if not user: @@ -84,10 +91,8 @@ async def api_usermanager_activate_extension( # Wallets -@usermanager_ext.post("/api/v1/wallets") -async def api_usermanager_wallets_create( - data: CreateUserWallet, wallet: WalletTypeInfo = Depends(get_key_type) -): +@usermanager_ext.post("/api/v1/wallets", dependencies=[Depends(get_key_type)]) +async def api_usermanager_wallets_create(data: CreateUserWallet): user = await create_usermanager_wallet( user_id=data.user_id, wallet_name=data.wallet_name, admin_id=data.admin_id ) @@ -95,31 +100,33 @@ async def api_usermanager_wallets_create( @usermanager_ext.get("/api/v1/wallets") -async def api_usermanager_wallets(wallet: WalletTypeInfo = Depends(require_admin_key)): +async def api_usermanager_wallets( + wallet: WalletTypeInfo = Depends(require_admin_key), # type: ignore +): 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) -): +@usermanager_ext.get( + "/api/v1/transactions/{wallet_id}", dependencies=[Depends(get_key_type)] +) +async def api_usermanager_wallet_transactions(wallet_id): 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) -): +@usermanager_ext.get( + "/api/v1/wallets/{user_id}", dependencies=[Depends(require_admin_key)] +) +async def api_usermanager_users_wallets(user_id): 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) -): +@usermanager_ext.delete( + "/api/v1/wallets/{wallet_id}", dependencies=[Depends(require_admin_key)] +) +async def api_usermanager_wallets_delete(wallet_id): get_wallet = await get_usermanager_wallet(wallet_id) if not get_wallet: raise HTTPException( diff --git a/pyproject.toml b/pyproject.toml index fe56aa96..fa81bcff 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -111,7 +111,6 @@ exclude = """(?x)( | ^lnbits/extensions/streamalerts. | ^lnbits/extensions/tipjar. | ^lnbits/extensions/tpos. - | ^lnbits/extensions/usermanager. | ^lnbits/extensions/watchonly. | ^lnbits/extensions/withdraw. | ^lnbits/wallets/lnd_grpc_files.