mirror of
https://github.com/lnbits/lnbits.git
synced 2025-08-03 15:32:22 +02:00
fix: add default extensions when a user is created (#3061)
This commit is contained in:
@@ -46,7 +46,9 @@ async def create_user_account(
|
|||||||
|
|
||||||
|
|
||||||
async def create_user_account_no_ckeck(
|
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:
|
) -> User:
|
||||||
|
|
||||||
if account:
|
if account:
|
||||||
@@ -69,9 +71,13 @@ async def create_user_account_no_ckeck(
|
|||||||
wallet_name=wallet_name or settings.lnbits_default_wallet_name,
|
wallet_name=wallet_name or settings.lnbits_default_wallet_name,
|
||||||
)
|
)
|
||||||
|
|
||||||
for ext_id in settings.lnbits_user_default_extensions:
|
user_extensions = (default_exts or []) + settings.lnbits_user_default_extensions
|
||||||
user_ext = UserExtension(user=account.id, extension=ext_id, active=True)
|
for ext_id in user_extensions:
|
||||||
await update_user_extension(user_ext)
|
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)
|
user = await get_user_from_account(account)
|
||||||
assert user, "Cannot find user for account."
|
assert user, "Cannot find user for account."
|
||||||
|
@@ -104,7 +104,7 @@ async def api_create_user(data: CreateUser) -> CreateUser:
|
|||||||
)
|
)
|
||||||
account.validate_fields()
|
account.validate_fields()
|
||||||
account.hash_password(data.password)
|
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
|
data.id = user.id
|
||||||
return data
|
return data
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user