File tree Expand file tree Collapse file tree
src/models/migrations/versions Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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
You can’t perform that action at this time.
0 commit comments