highperfocused/lnbits
1
0
Fork 0
mirror of https://github.com/lnbits/lnbits.git synced 2025-03-28 10:41:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: add default extensions when a user is created (#3061)

Browse Source
This commit is contained in:
Vlad Stan 2025-03-23 01:54:15 +02:00 committed by GitHub
parent 30a8d88ada
commit b185d9585c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lnbits/core/services/users.py
View File

@ -46,7 +46,9 @@ async def create_user_account(
async def create_user_account_no_ckeck(
account: Optional[Account] = None, wallet_name: Optional[str] = None
account: Optional[Account] = None,
wallet_name: Optional[str] = None,
default_exts: Optional[list[str]] = None,
) -> User:
if account:
@ -69,9 +71,13 @@ async def create_user_account_no_ckeck(
wallet_name=wallet_name or settings.lnbits_default_wallet_name,
)
for ext_id in settings.lnbits_user_default_extensions:
user_ext = UserExtension(user=account.id, extension=ext_id, active=True)
await update_user_extension(user_ext)
user_extensions = (default_exts or []) + settings.lnbits_user_default_extensions
for ext_id in user_extensions:
try:
user_ext = UserExtension(user=account.id, extension=ext_id, active=True)
await create_user_extension(user_ext)
except Exception as e:
logger.error(f"Error enabeling default extension {ext_id}: {e}")
user = await get_user_from_account(account)
assert user, "Cannot find user for account."

2
lnbits/core/views/user_api.py
View File

@ -104,7 +104,7 @@ async def api_create_user(data: CreateUser) -> CreateUser:
)
account.validate_fields()
account.hash_password(data.password)
user = await create_user_account_no_ckeck(account)
user = await create_user_account_no_ckeck(account, default_exts=data.extensions)
data.id = user.id
return data