Skip to content

Commit 992b60f

Browse files
committed
Update default limit values
1 parent 13f711a commit 992b60f

1 file changed

Lines changed: 49 additions & 0 deletions

File tree

Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,49 @@
1+
"""update-resource-defaults
2+
3+
Revision ID: 6726a34eb0ec
4+
Revises: ad471311850e
5+
Create Date: 2026-03-25 14:27:12.714577
6+
7+
"""
8+
from typing import Sequence, Union
9+
10+
from alembic import op
11+
import sqlalchemy as sa
12+
import sqlmodel
13+
import sqlmodel.sql
14+
15+
16+
# revision identifiers, used by Alembic.
17+
revision: str = '6726a34eb0ec'
18+
down_revision: Union[str, Sequence[str], None] = 'ad471311850e'
19+
branch_labels: Union[str, Sequence[str], None] = None
20+
depends_on: Union[str, Sequence[str], None] = None
21+
22+
23+
_GIB = 1024 ** 3
24+
_GB = 1000 ** 3
25+
26+
27+
def upgrade() -> None:
28+
"""Upgrade schema."""
29+
conn = op.get_bind()
30+
for resource, value in {
31+
"milli_vcpu": 10_000,
32+
"ram": 20 * _GIB,
33+
"iops": 100_000,
34+
"database_size": 100 * _GB,
35+
"storage_size": 10 * _GB,
36+
}.items():
37+
conn.execute(
38+
sa.text(
39+
"UPDATE organizationlimitdefault "
40+
"SET max_total = :val, max_per_branch = :val "
41+
"WHERE resource = :resource"
42+
),
43+
{"val": value, "resource": resource},
44+
)
45+
46+
47+
def downgrade() -> None:
48+
"""Downgrade schema."""
49+
pass # Previous values not stored; cannot trivially reverse

0 commit comments

Comments
 (0)