fix: unlimited admin upload (#3679)
Co-authored-by: Arc <33088785+arcbtc@users.noreply.github.com>
This commit is contained in:
parent
08591f34c2
commit
281c3df826
2 changed files with 7 additions and 1 deletions
|
|
@ -16,7 +16,7 @@ async def create_user_asset(user_id: str, file: UploadFile, is_public: bool) ->
|
||||||
if file.content_type.lower() not in settings.lnbits_assets_allowed_mime_types:
|
if file.content_type.lower() not in settings.lnbits_assets_allowed_mime_types:
|
||||||
raise ValueError(f"File type '{file.content_type}' not allowed.")
|
raise ValueError(f"File type '{file.content_type}' not allowed.")
|
||||||
|
|
||||||
if user_id not in settings.lnbits_assets_no_limit_users:
|
if not settings.is_unlimited_assets_user(user_id):
|
||||||
user_assets_count = await get_user_assets_count(user_id)
|
user_assets_count = await get_user_assets_count(user_id)
|
||||||
if user_assets_count >= settings.lnbits_max_assets_per_user:
|
if user_assets_count >= settings.lnbits_max_assets_per_user:
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
|
|
|
||||||
|
|
@ -308,6 +308,12 @@ class AssetSettings(LNbitsSettings):
|
||||||
lnbits_max_assets_per_user: int = Field(default=1, ge=0)
|
lnbits_max_assets_per_user: int = Field(default=1, ge=0)
|
||||||
lnbits_assets_no_limit_users: list[str] = Field(default=[])
|
lnbits_assets_no_limit_users: list[str] = Field(default=[])
|
||||||
|
|
||||||
|
def is_unlimited_assets_user(self, user_id: str) -> bool:
|
||||||
|
return (
|
||||||
|
settings.is_admin_user(user_id)
|
||||||
|
or user_id in self.lnbits_assets_no_limit_users
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class FeeSettings(LNbitsSettings):
|
class FeeSettings(LNbitsSettings):
|
||||||
lnbits_reserve_fee_min: int = Field(default=2000, ge=0)
|
lnbits_reserve_fee_min: int = Field(default=2000, ge=0)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue