add _directory_quota, _subdomain_quota column to User
This commit is contained in:
parent
bccfcee780
commit
85c6e791bc
|
@ -30,6 +30,8 @@ from app.config import (
|
||||||
DISABLE_ONBOARDING,
|
DISABLE_ONBOARDING,
|
||||||
UNSUBSCRIBER,
|
UNSUBSCRIBER,
|
||||||
ALIAS_RANDOM_SUFFIX_LENGTH,
|
ALIAS_RANDOM_SUFFIX_LENGTH,
|
||||||
|
MAX_NB_SUBDOMAIN,
|
||||||
|
MAX_NB_DIRECTORY,
|
||||||
)
|
)
|
||||||
from app.db import Session
|
from app.db import Session
|
||||||
from app.errors import AliasInTrashError, DirectoryInTrashError, SubdomainInTrashError
|
from app.errors import AliasInTrashError, DirectoryInTrashError, SubdomainInTrashError
|
||||||
|
@ -397,6 +399,29 @@ class User(Base, ModelMixin, UserMixin, PasswordOracle):
|
||||||
server_default="0",
|
server_default="0",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
_directory_quota = sa.Column(
|
||||||
|
"directory_quota", sa.Integer, default=50, nullable=False, server_default="50"
|
||||||
|
)
|
||||||
|
|
||||||
|
_subdomain_quota = sa.Column(
|
||||||
|
"subdomain_quota", sa.Integer, default=5, nullable=False, server_default="5"
|
||||||
|
)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def directory_quota(self):
|
||||||
|
return min(
|
||||||
|
self._directory_quota,
|
||||||
|
MAX_NB_DIRECTORY - Directory.filter_by(user_id=self.id).count(),
|
||||||
|
)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def subdomain_quota(self):
|
||||||
|
return min(
|
||||||
|
self._subdomain_quota,
|
||||||
|
MAX_NB_SUBDOMAIN
|
||||||
|
- CustomDomain.filter_by(user_id=self.id, is_sl_subdomain=True).count(),
|
||||||
|
)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def subdomain_is_available():
|
def subdomain_is_available():
|
||||||
return SLDomain.filter_by(can_use_subdomain=True).count() > 0
|
return SLDomain.filter_by(can_use_subdomain=True).count() > 0
|
||||||
|
|
31
migrations/versions/2021_111810_b8fd175c084a_.py
Normal file
31
migrations/versions/2021_111810_b8fd175c084a_.py
Normal file
|
@ -0,0 +1,31 @@
|
||||||
|
"""empty message
|
||||||
|
|
||||||
|
Revision ID: b8fd175c084a
|
||||||
|
Revises: 9031c9e28510
|
||||||
|
Create Date: 2021-11-18 10:29:13.512071
|
||||||
|
|
||||||
|
"""
|
||||||
|
import sqlalchemy_utils
|
||||||
|
from alembic import op
|
||||||
|
import sqlalchemy as sa
|
||||||
|
|
||||||
|
|
||||||
|
# revision identifiers, used by Alembic.
|
||||||
|
revision = 'b8fd175c084a'
|
||||||
|
down_revision = '9031c9e28510'
|
||||||
|
branch_labels = None
|
||||||
|
depends_on = None
|
||||||
|
|
||||||
|
|
||||||
|
def upgrade():
|
||||||
|
# ### commands auto generated by Alembic - please adjust! ###
|
||||||
|
op.add_column('users', sa.Column('directory_quota', sa.Integer(), server_default='50', nullable=False))
|
||||||
|
op.add_column('users', sa.Column('subdomain_quota', sa.Integer(), server_default='5', nullable=False))
|
||||||
|
# ### end Alembic commands ###
|
||||||
|
|
||||||
|
|
||||||
|
def downgrade():
|
||||||
|
# ### commands auto generated by Alembic - please adjust! ###
|
||||||
|
op.drop_column('users', 'subdomain_quota')
|
||||||
|
op.drop_column('users', 'directory_quota')
|
||||||
|
# ### end Alembic commands ###
|
Loading…
Reference in a new issue