|
1 | | -"""initial migration |
2 | | -
|
3 | | -Revision ID: dd0839d991f2 |
4 | | -Revises: |
5 | | -Create Date: 2025-08-26 09:29:41.248512 |
6 | | -
|
7 | | -""" |
8 | | -from collections.abc import Sequence |
9 | | - |
10 | | -import sqlalchemy as sa |
11 | | -from alembic import op |
12 | | - |
13 | | -# revision identifiers, used by Alembic. |
14 | | -revision: str = "dd0839d991f2" |
15 | | -down_revision: str | None = None |
16 | | -branch_labels: str | Sequence[str] | None = None |
17 | | -depends_on: str | Sequence[str] | None = None |
18 | | - |
19 | | - |
20 | | -def upgrade() -> None: |
21 | | - # ### commands auto generated by Alembic - please adjust! ### |
22 | | - op.create_table("run_metadata", |
23 | | - sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
24 | | - sa.Column("cluster", sa.String(), nullable=False), |
25 | | - sa.Column("platform", sa.String(), nullable=False), |
26 | | - sa.Column("region", sa.String(), nullable=False), |
27 | | - sa.Column("namespace", sa.String(), nullable=False), |
28 | | - sa.Column("date", sa.String(), nullable=False), |
29 | | - sa.PrimaryKeyConstraint("id"), |
30 | | - sa.UniqueConstraint("cluster", "platform", "namespace", "date") |
31 | | - ) |
32 | | - op.create_table("container", |
33 | | - sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
34 | | - sa.Column("run_id", sa.Integer(), nullable=False), |
35 | | - sa.Column("name", sa.String(), nullable=False), |
36 | | - sa.Column("pod_id", sa.String(), nullable=True), |
37 | | - sa.Column("ready", sa.Integer(), nullable=True), |
38 | | - sa.Column("started", sa.Integer(), nullable=True), |
39 | | - sa.Column("scheduled", sa.Integer(), nullable=True), |
40 | | - sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
41 | | - sa.PrimaryKeyConstraint("id"), |
42 | | - sa.UniqueConstraint("pod_id", "name") |
43 | | - ) |
44 | | - op.create_table("node", |
45 | | - sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
46 | | - sa.Column("name", sa.String(), nullable=False), |
47 | | - sa.Column("run_id", sa.Integer(), nullable=False), |
48 | | - sa.Column("conditions", sa.String(), nullable=True), |
49 | | - sa.Column("cpu_capacity", sa.Float(), nullable=True), |
50 | | - sa.Column("cpu_allocatable", sa.Float(), nullable=True), |
51 | | - sa.Column("memory_capacity", sa.Float(), nullable=True), |
52 | | - sa.Column("memory_allocatable", sa.Float(), nullable=True), |
53 | | - sa.Column("cpu_reserved", sa.Float(), nullable=True), |
54 | | - sa.Column("memory_reserved", sa.Float(), nullable=True), |
55 | | - sa.Column("eviction_empty", sa.Integer(), nullable=True), |
56 | | - sa.Column("zone", sa.String(), nullable=True), |
57 | | - sa.Column("region", sa.String(), nullable=True), |
58 | | - sa.Column("node_size", sa.String(), nullable=True), |
59 | | - sa.Column("capacity_type", sa.String(), nullable=True), |
60 | | - sa.Column("created", sa.Integer(), nullable=True), |
61 | | - sa.Column("ready", sa.Integer(), nullable=True), |
62 | | - sa.Column("deleted", sa.Integer(), nullable=True), |
63 | | - sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
64 | | - sa.PrimaryKeyConstraint("id") |
65 | | - ) |
66 | | - op.create_table("pod", |
67 | | - sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
68 | | - sa.Column("name", sa.String(), nullable=False), |
69 | | - sa.Column("run_id", sa.Integer(), nullable=False), |
70 | | - sa.Column("request_id", sa.String(), nullable=True), |
71 | | - sa.Column("return_id", sa.String(), nullable=True), |
72 | | - sa.Column("template_id", sa.String(), nullable=True), |
73 | | - sa.Column("cpu_requested", sa.Float(), nullable=True), |
74 | | - sa.Column("cpu_limit", sa.Float(), nullable=True), |
75 | | - sa.Column("memory_requested", sa.Float(), nullable=True), |
76 | | - sa.Column("memory_limit", sa.Float(), nullable=True), |
77 | | - sa.Column("node_name", sa.String(), nullable=True), |
78 | | - sa.Column("node_id", sa.String(), nullable=True), |
79 | | - sa.Column("requested", sa.Integer(), nullable=True), |
80 | | - sa.Column("pending", sa.Integer(), nullable=True), |
81 | | - sa.Column("created", sa.Integer(), nullable=True), |
82 | | - sa.Column("scheduled", sa.Integer(), nullable=True), |
83 | | - sa.Column("running", sa.Integer(), nullable=True), |
84 | | - sa.Column("ready", sa.Integer(), nullable=True), |
85 | | - sa.Column("returned", sa.Integer(), nullable=True), |
86 | | - sa.Column("deleted", sa.Integer(), nullable=True), |
87 | | - sa.Column("failed", sa.Integer(), nullable=True), |
88 | | - sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
89 | | - sa.PrimaryKeyConstraint("id") |
90 | | - ) |
91 | | - op.create_table("request", |
92 | | - sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
93 | | - sa.Column("name", sa.String(), nullable=False), |
94 | | - sa.Column("run_id", sa.Integer(), nullable=False), |
95 | | - sa.Column("count", sa.Integer(), nullable=True), |
96 | | - sa.Column("template_id", sa.String(), nullable=True), |
97 | | - sa.Column("running", sa.Integer(), nullable=True), |
98 | | - sa.Column("complete", sa.Integer(), nullable=True), |
99 | | - sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
100 | | - sa.PrimaryKeyConstraint("id") |
101 | | - ) |
102 | | - op.create_table("return", |
103 | | - sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
104 | | - sa.Column("name", sa.String(), nullable=False), |
105 | | - sa.Column("run_id", sa.Integer(), nullable=False), |
106 | | - sa.Column("count", sa.Integer(), nullable=True), |
107 | | - sa.Column("running", sa.Integer(), nullable=True), |
108 | | - sa.Column("complete", sa.Integer(), nullable=True), |
109 | | - sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
110 | | - sa.PrimaryKeyConstraint("id") |
111 | | - ) |
112 | | - op.create_table("summary", |
113 | | - sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
114 | | - sa.Column("run_id", sa.Integer(), nullable=False), |
115 | | - sa.Column("start", sa.Integer(), nullable=True), |
116 | | - sa.Column("end", sa.Integer(), nullable=True), |
117 | | - sa.Column("cpu_minutes", sa.Float(), nullable=True), |
118 | | - sa.Column("percentage_node_usage", sa.Float(), nullable=True), |
119 | | - sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
120 | | - sa.PrimaryKeyConstraint("id") |
121 | | - ) |
122 | | - op.create_table("template", |
123 | | - sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
124 | | - sa.Column("name", sa.String(), nullable=False), |
125 | | - sa.Column("run_id", sa.Integer(), nullable=False), |
126 | | - sa.Column("output", sa.String(), nullable=True), |
127 | | - sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
128 | | - sa.PrimaryKeyConstraint("id") |
129 | | - ) |
130 | | - # ### end Alembic commands ### |
131 | | - |
132 | | - |
133 | | -def downgrade() -> None: |
134 | | - # ### commands auto generated by Alembic - please adjust! ### |
135 | | - op.drop_table("template") |
136 | | - op.drop_table("summary") |
137 | | - op.drop_table("return") |
138 | | - op.drop_table("request") |
139 | | - op.drop_table("pod") |
140 | | - op.drop_table("node") |
141 | | - op.drop_table("container") |
142 | | - op.drop_table("run_metadata") |
143 | | - # ### end Alembic commands ### |
| 1 | +"""initial migration |
| 2 | +
|
| 3 | +Revision ID: dd0839d991f2 |
| 4 | +Revises: |
| 5 | +Create Date: 2025-08-26 09:29:41.248512 |
| 6 | +
|
| 7 | +""" |
| 8 | +from collections.abc import Sequence |
| 9 | + |
| 10 | +import sqlalchemy as sa |
| 11 | +from alembic import op |
| 12 | + |
| 13 | +# revision identifiers, used by Alembic. |
| 14 | +revision: str = "dd0839d991f2" |
| 15 | +down_revision: str | None = None |
| 16 | +branch_labels: str | Sequence[str] | None = None |
| 17 | +depends_on: str | Sequence[str] | None = None |
| 18 | + |
| 19 | + |
| 20 | +def upgrade() -> None: |
| 21 | + # ### commands auto generated by Alembic - please adjust! ### |
| 22 | + op.create_table("run_metadata", |
| 23 | + sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
| 24 | + sa.Column("cluster", sa.String(), nullable=False), |
| 25 | + sa.Column("platform", sa.String(), nullable=False), |
| 26 | + sa.Column("region", sa.String(), nullable=False), |
| 27 | + sa.Column("namespace", sa.String(), nullable=False), |
| 28 | + sa.Column("date", sa.String(), nullable=False), |
| 29 | + sa.PrimaryKeyConstraint("id"), |
| 30 | + sa.UniqueConstraint("cluster", "platform", "namespace", "date") |
| 31 | + ) |
| 32 | + op.create_table("container", |
| 33 | + sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
| 34 | + sa.Column("run_id", sa.Integer(), nullable=False), |
| 35 | + sa.Column("name", sa.String(), nullable=False), |
| 36 | + sa.Column("pod_id", sa.String(), nullable=True), |
| 37 | + sa.Column("ready", sa.Integer(), nullable=True), |
| 38 | + sa.Column("started", sa.Integer(), nullable=True), |
| 39 | + sa.Column("scheduled", sa.Integer(), nullable=True), |
| 40 | + sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
| 41 | + sa.PrimaryKeyConstraint("id"), |
| 42 | + sa.UniqueConstraint("pod_id", "name") |
| 43 | + ) |
| 44 | + op.create_table("node", |
| 45 | + sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
| 46 | + sa.Column("name", sa.String(), nullable=False), |
| 47 | + sa.Column("run_id", sa.Integer(), nullable=False), |
| 48 | + sa.Column("conditions", sa.String(), nullable=True), |
| 49 | + sa.Column("cpu_capacity", sa.Float(), nullable=True), |
| 50 | + sa.Column("cpu_allocatable", sa.Float(), nullable=True), |
| 51 | + sa.Column("memory_capacity", sa.Float(), nullable=True), |
| 52 | + sa.Column("memory_allocatable", sa.Float(), nullable=True), |
| 53 | + sa.Column("cpu_reserved", sa.Float(), nullable=True), |
| 54 | + sa.Column("memory_reserved", sa.Float(), nullable=True), |
| 55 | + sa.Column("eviction_empty", sa.Integer(), nullable=True), |
| 56 | + sa.Column("zone", sa.String(), nullable=True), |
| 57 | + sa.Column("region", sa.String(), nullable=True), |
| 58 | + sa.Column("node_size", sa.String(), nullable=True), |
| 59 | + sa.Column("capacity_type", sa.String(), nullable=True), |
| 60 | + sa.Column("created", sa.Integer(), nullable=True), |
| 61 | + sa.Column("ready", sa.Integer(), nullable=True), |
| 62 | + sa.Column("deleted", sa.Integer(), nullable=True), |
| 63 | + sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
| 64 | + sa.PrimaryKeyConstraint("id") |
| 65 | + ) |
| 66 | + op.create_table("pod", |
| 67 | + sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
| 68 | + sa.Column("name", sa.String(), nullable=False), |
| 69 | + sa.Column("run_id", sa.Integer(), nullable=False), |
| 70 | + sa.Column("request_id", sa.String(), nullable=True), |
| 71 | + sa.Column("return_id", sa.String(), nullable=True), |
| 72 | + sa.Column("template_id", sa.String(), nullable=True), |
| 73 | + sa.Column("cpu_requested", sa.Float(), nullable=True), |
| 74 | + sa.Column("cpu_limit", sa.Float(), nullable=True), |
| 75 | + sa.Column("memory_requested", sa.Float(), nullable=True), |
| 76 | + sa.Column("memory_limit", sa.Float(), nullable=True), |
| 77 | + sa.Column("node_name", sa.String(), nullable=True), |
| 78 | + sa.Column("node_id", sa.String(), nullable=True), |
| 79 | + sa.Column("requested", sa.Integer(), nullable=True), |
| 80 | + sa.Column("pending", sa.Integer(), nullable=True), |
| 81 | + sa.Column("created", sa.Integer(), nullable=True), |
| 82 | + sa.Column("scheduled", sa.Integer(), nullable=True), |
| 83 | + sa.Column("running", sa.Integer(), nullable=True), |
| 84 | + sa.Column("ready", sa.Integer(), nullable=True), |
| 85 | + sa.Column("returned", sa.Integer(), nullable=True), |
| 86 | + sa.Column("deleted", sa.Integer(), nullable=True), |
| 87 | + sa.Column("failed", sa.Integer(), nullable=True), |
| 88 | + sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
| 89 | + sa.PrimaryKeyConstraint("id") |
| 90 | + ) |
| 91 | + op.create_table("request", |
| 92 | + sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
| 93 | + sa.Column("name", sa.String(), nullable=False), |
| 94 | + sa.Column("run_id", sa.Integer(), nullable=False), |
| 95 | + sa.Column("count", sa.Integer(), nullable=True), |
| 96 | + sa.Column("template_id", sa.String(), nullable=True), |
| 97 | + sa.Column("running", sa.Integer(), nullable=True), |
| 98 | + sa.Column("complete", sa.Integer(), nullable=True), |
| 99 | + sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
| 100 | + sa.PrimaryKeyConstraint("id") |
| 101 | + ) |
| 102 | + op.create_table("return", |
| 103 | + sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
| 104 | + sa.Column("name", sa.String(), nullable=False), |
| 105 | + sa.Column("run_id", sa.Integer(), nullable=False), |
| 106 | + sa.Column("count", sa.Integer(), nullable=True), |
| 107 | + sa.Column("running", sa.Integer(), nullable=True), |
| 108 | + sa.Column("complete", sa.Integer(), nullable=True), |
| 109 | + sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
| 110 | + sa.PrimaryKeyConstraint("id") |
| 111 | + ) |
| 112 | + op.create_table("summary", |
| 113 | + sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
| 114 | + sa.Column("run_id", sa.Integer(), nullable=False), |
| 115 | + sa.Column("start", sa.Integer(), nullable=True), |
| 116 | + sa.Column("end", sa.Integer(), nullable=True), |
| 117 | + sa.Column("cpu_minutes", sa.Float(), nullable=True), |
| 118 | + sa.Column("percentage_node_usage", sa.Float(), nullable=True), |
| 119 | + sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
| 120 | + sa.PrimaryKeyConstraint("id") |
| 121 | + ) |
| 122 | + op.create_table("template", |
| 123 | + sa.Column("id", sa.Integer(), autoincrement=True, nullable=False), |
| 124 | + sa.Column("name", sa.String(), nullable=False), |
| 125 | + sa.Column("run_id", sa.Integer(), nullable=False), |
| 126 | + sa.Column("output", sa.String(), nullable=True), |
| 127 | + sa.ForeignKeyConstraint(["run_id"], ["run_metadata.id"], ), |
| 128 | + sa.PrimaryKeyConstraint("id") |
| 129 | + ) |
| 130 | + # ### end Alembic commands ### |
| 131 | + |
| 132 | + |
| 133 | +def downgrade() -> None: |
| 134 | + # ### commands auto generated by Alembic - please adjust! ### |
| 135 | + op.drop_table("template") |
| 136 | + op.drop_table("summary") |
| 137 | + op.drop_table("return") |
| 138 | + op.drop_table("request") |
| 139 | + op.drop_table("pod") |
| 140 | + op.drop_table("node") |
| 141 | + op.drop_table("container") |
| 142 | + op.drop_table("run_metadata") |
| 143 | + # ### end Alembic commands ### |
0 commit comments