Skip to content

Commit 1d5bc91

Browse files
committed
Fix system limit migration
The previous version failed to reuse the existing enum datatype. The change makes sure it is referenced instead.
1 parent 2a1c25c commit 1d5bc91

1 file changed

Lines changed: 2 additions & 5 deletions

File tree

src/models/migrations/versions/e8f3a2d51c9b_fix_system_limits.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
import sqlmodel
1212
import sqlmodel.sql
1313
from alembic import op
14+
from sqlalchemy.dialects import postgresql
1415
from ulid import ULID
1516

1617
# revision identifiers, used by Alembic.
@@ -22,11 +23,7 @@
2223

2324
def upgrade() -> None:
2425
"""Upgrade schema."""
25-
resource_type_enum = sa.Enum(
26-
"milli_vcpu", "ram", "iops", "database_size", "storage_size",
27-
name="resourcetype",
28-
create_type=False,
29-
)
26+
resource_type_enum = postgresql.ENUM(name="resourcetype", create_type=False)
3027

3128
# Create the organizationlimitdefault table
3229
op.create_table(

0 commit comments

Comments
 (0)