From 42a9af986ab3620bdb61270961c6adbda4a66c95 Mon Sep 17 00:00:00 2001
From: Tiago vasconcelos <talvasconcelos@gmail.com>
Date: Tue, 22 Mar 2022 10:28:44 +0000
Subject: [PATCH] update admin settings in db

---
 lnbits/extensions/admin/models.py    | 29 +++++++++++++++++-----------
 lnbits/extensions/admin/views_api.py |  8 ++++----
 2 files changed, 22 insertions(+), 15 deletions(-)

diff --git a/lnbits/extensions/admin/models.py b/lnbits/extensions/admin/models.py
index f7c64de5..36d9b815 100644
--- a/lnbits/extensions/admin/models.py
+++ b/lnbits/extensions/admin/models.py
@@ -6,17 +6,24 @@ from pydantic import BaseModel, Field
 
 
 class UpdateAdminSettings(BaseModel):
-    site_title: Optional[str]
-    site_tagline: Optional[str]
-    site_description: Optional[str]
-    allowed_users: Optional[str]
-    admin_users: Optional[str]
-    default_wallet_name: Optional[str]
-    data_folder: Optional[str]
-    disabled_ext: Optional[str]
-    force_https: Optional[bool]
-    service_fee: Optional[float]
-    funding_source: Optional[str]
+    # users
+    admin_users: str = Query(None)
+    allowed_users: str = Query(None)
+    admin_ext: str = Query(None)
+    disabled_ext: str = Query(None)
+    funding_source: str = Query(None)
+    # ops
+    force_https: bool = Query(None)
+    service_fee: float = Query(None, ge=0)
+    hide_api: bool = Query(None)
+    # Change theme
+    site_title: str = Query(None)
+    site_tagline: str = Query(None)
+    site_description: str = Query(None)
+    default_wallet_name: str = Query(None)
+    denomination: str = Query(None)
+    theme: str = Query(None)
+    ad_space: str = Query(None)
 
 class Admin(BaseModel):
     # users
diff --git a/lnbits/extensions/admin/views_api.py b/lnbits/extensions/admin/views_api.py
index b2c65be2..cb526aa5 100644
--- a/lnbits/extensions/admin/views_api.py
+++ b/lnbits/extensions/admin/views_api.py
@@ -12,16 +12,16 @@ from .crud import get_admin, update_admin, update_wallet_balance
 
 
 @admin_ext.get("/api/v1/admin/{wallet_id}/{topup_amount}", status_code=HTTPStatus.OK)
-async def api_update_balance(wallet_id, topup_amount, g: WalletTypeInfo = Depends(require_admin_key)):
-    print(g.wallet)
+async def api_update_balance(wallet_id, topup_amount: int, g: WalletTypeInfo = Depends(require_admin_key)):
     try:
         wallet = await get_wallet(wallet_id)
     except:
         raise HTTPException(
                 status_code=HTTPStatus.FORBIDDEN, detail="Not allowed: not an admin"
             )
-    print(wallet)
-    print(topup_amount)
+            
+    await update_wallet_balance(wallet_id=wallet_id, amount=int(topup_amount))
+    
     return {"status": "Success"}