revert back to old migration
Taylor committed
Jul 12, 2023 at 16:55 UTC
76295bf725010a10ac9375274150be6dc8d442ed
1 file changed
+86
-109
backend/migrations/versions/0381c0088cbe_initial_migration.py
+86
-109
@@ -1,13 +1,10 @@
1
"""Initial migration.
2
-
2
Revision ID: 0381c0088cbe
3
Revises:
4
Create Date: 2023-07-11 13:24:04.087035
6
-
5
"""
6
import sqlalchemy as sa
7
from alembic import op
10
-from sqlalchemy.engine.reflection import Inspector
8
9
# revision identifiers, used by Alembic.
10
revision = "0381c0088cbe"
@@ -18,112 +15,92 @@ depends_on = None
15
16
def upgrade():
17
# ### commands auto generated by Alembic - please adjust! ###
21
- conn = op.get_bind()
22
- inspector = Inspector.from_engine(conn)
23
-
24
- if "agent_metadata" not in inspector.get_table_names():
25
- op.create_table(
26
- "agent_metadata",
27
- sa.Column("id", sa.Integer(), nullable=False),
28
- sa.Column("agent_id", sa.String(length=100), nullable=True),
29
- sa.Column("ip_address", sa.String(length=100), nullable=True),
30
- sa.Column("os", sa.String(length=100), nullable=True),
31
- sa.Column("hostname", sa.String(length=100), nullable=True),
32
- sa.Column("critical_asset", sa.Boolean(), nullable=True),
33
- sa.Column("last_seen", sa.DateTime(), nullable=True),
34
- sa.PrimaryKeyConstraint("id"),
35
- )
36
- if "artifact" not in inspector.get_table_names():
37
- op.create_table(
38
- "artifact",
39
- sa.Column("id", sa.Integer(), nullable=False),
40
- sa.Column("artifact_name", sa.String(length=100), nullable=True),
41
- sa.Column("artifact_results", sa.TEXT(), nullable=True),
42
- sa.Column("hostname", sa.String(length=100), nullable=True),
43
- sa.PrimaryKeyConstraint("id"),
44
- )
45
- if "case" not in inspector.get_table_names():
46
- op.create_table(
47
- "case",
48
- sa.Column("id", sa.Integer(), nullable=False),
49
- sa.Column("case_id", sa.Integer(), nullable=True),
50
- sa.Column("case_name", sa.String(length=100), nullable=True),
51
- sa.Column("agents", sa.String(length=1000), nullable=True),
52
- sa.PrimaryKeyConstraint("id"),
53
- )
54
- if "connectors" not in inspector.get_table_names():
55
- op.create_table(
56
- "connectors",
57
- sa.Column("id", sa.Integer(), nullable=False),
58
- sa.Column("connector_name", sa.String(length=100), nullable=True),
59
- sa.Column("connector_type", sa.String(length=100), nullable=True),
60
- sa.Column("connector_url", sa.String(length=100), nullable=True),
61
- sa.Column("connector_last_updated", sa.DateTime(), nullable=True),
62
- sa.Column("connector_username", sa.String(length=100), nullable=True),
63
- sa.Column("connector_password", sa.String(length=100), nullable=True),
64
- sa.Column("connector_api_key", sa.String(length=100), nullable=True),
65
- sa.PrimaryKeyConstraint("id"),
66
- sa.UniqueConstraint("connector_name"),
67
- )
68
- if "connectors_available" not in inspector.get_table_names():
69
- op.create_table(
70
- "connectors_available",
71
- sa.Column("id", sa.Integer(), nullable=False),
72
- sa.Column("connector_name", sa.String(length=100), nullable=True),
73
- sa.Column("connector_description", sa.String(length=100), nullable=True),
74
- sa.Column("connector_supports", sa.String(length=100), nullable=True),
75
- sa.Column("connector_configured", sa.Boolean(), nullable=True),
76
- sa.Column("connector_verified", sa.Boolean(), nullable=True),
77
- sa.PrimaryKeyConstraint("id"),
78
- sa.UniqueConstraint("connector_name"),
79
- )
80
- if "disabled_rules" not in inspector.get_table_names():
81
- op.create_table(
82
- "disabled_rules",
83
- sa.Column("id", sa.Integer(), nullable=False),
84
- sa.Column("rule_id", sa.String(length=100), nullable=True),
85
- sa.Column("previous_level", sa.String(length=1000), nullable=True),
86
- sa.Column("new_level", sa.String(length=1000), nullable=True),
87
- sa.Column("reason_for_disabling", sa.String(length=100), nullable=True),
88
- sa.Column("date_disabled", sa.DateTime(), nullable=True),
89
- sa.Column("length_of_time", sa.Integer(), nullable=True),
90
- sa.PrimaryKeyConstraint("id"),
91
- )
92
- if "graylog_metrics_allocation" not in inspector.get_table_names():
93
- op.create_table(
94
- "graylog_metrics_allocation",
95
- sa.Column("id", sa.Integer(), nullable=False),
96
- sa.Column("input_usage", sa.Float(), nullable=True),
97
- sa.Column("output_usage", sa.Float(), nullable=True),
98
- sa.Column("processor_usage", sa.Float(), nullable=True),
99
- sa.Column("input_1_sec_rate", sa.Float(), nullable=True),
100
- sa.Column("output_1_sec_rate", sa.Float(), nullable=True),
101
- sa.Column("total_input", sa.Float(), nullable=True),
102
- sa.Column("total_output", sa.Float(), nullable=True),
103
- sa.Column("timestamp", sa.DateTime(), nullable=True),
104
- sa.PrimaryKeyConstraint("id"),
105
- )
106
- if "wazuh_indexer_allocation" not in inspector.get_table_names():
107
- op.create_table(
108
- "wazuh_indexer_allocation",
109
- sa.Column("id", sa.Integer(), nullable=False),
110
- sa.Column("node", sa.String(length=100), nullable=True),
111
- sa.Column("disk_used", sa.Float(), nullable=True),
112
- sa.Column("disk_available", sa.Float(), nullable=True),
113
- sa.Column("disk_total", sa.Float(), nullable=True),
114
- sa.Column("disk_percent", sa.Float(), nullable=True),
115
- sa.Column("timestamp", sa.DateTime(), nullable=True),
116
- sa.PrimaryKeyConstraint("id"),
117
- )
118
- if "sublime_alerts" not in inspector.get_table_names():
119
- op.create_table(
120
- "sublime_alerts",
121
- sa.Column("id", sa.Integer(), nullable=False),
122
- sa.Column("message_id", sa.String(length=1000), nullable=True),
123
- sa.Column("timestamp", sa.DateTime(), nullable=True),
124
- sa.PrimaryKeyConstraint("id"),
125
- )
126
-
18
+ op.create_table(
19
+ "agent_metadata",
20
+ sa.Column("id", sa.Integer(), nullable=False),
21
+ sa.Column("agent_id", sa.String(length=100), nullable=True),
22
+ sa.Column("ip_address", sa.String(length=100), nullable=True),
23
+ sa.Column("os", sa.String(length=100), nullable=True),
24
+ sa.Column("hostname", sa.String(length=100), nullable=True),
25
+ sa.Column("critical_asset", sa.Boolean(), nullable=True),
26
+ sa.Column("last_seen", sa.DateTime(), nullable=True),
27
+ sa.PrimaryKeyConstraint("id"),
28
+ )
29
+ op.create_table(
30
+ "artifact",
31
+ sa.Column("id", sa.Integer(), nullable=False),
32
+ sa.Column("artifact_name", sa.String(length=100), nullable=True),
33
+ sa.Column("artifact_results", sa.TEXT(), nullable=True),
34
+ sa.Column("hostname", sa.String(length=100), nullable=True),
35
+ sa.PrimaryKeyConstraint("id"),
36
+ )
37
+ op.create_table(
38
+ "case",
39
+ sa.Column("id", sa.Integer(), nullable=False),
40
+ sa.Column("case_id", sa.Integer(), nullable=True),
41
+ sa.Column("case_name", sa.String(length=100), nullable=True),
42
+ sa.Column("agents", sa.String(length=1000), nullable=True),
43
+ sa.PrimaryKeyConstraint("id"),
44
+ )
45
+ op.create_table(
46
+ "connectors",
47
+ sa.Column("id", sa.Integer(), nullable=False),
48
+ sa.Column("connector_name", sa.String(length=100), nullable=True),
49
+ sa.Column("connector_type", sa.String(length=100), nullable=True),
50
+ sa.Column("connector_url", sa.String(length=100), nullable=True),
51
+ sa.Column("connector_last_updated", sa.DateTime(), nullable=True),
52
+ sa.Column("connector_username", sa.String(length=100), nullable=True),
53
+ sa.Column("connector_password", sa.String(length=100), nullable=True),
54
+ sa.Column("connector_api_key", sa.String(length=100), nullable=True),
55
+ sa.PrimaryKeyConstraint("id"),
56
+ sa.UniqueConstraint("connector_name"),
57
+ )
58
+ op.create_table(
59
+ "connectors_available",
60
+ sa.Column("id", sa.Integer(), nullable=False),
61
+ sa.Column("connector_name", sa.String(length=100), nullable=True),
62
+ sa.Column("connector_description", sa.String(length=100), nullable=True),
63
+ sa.Column("connector_supports", sa.String(length=100), nullable=True),
64
+ sa.Column("connector_configured", sa.Boolean(), nullable=True),
65
+ sa.Column("connector_verified", sa.Boolean(), nullable=True),
66
+ sa.PrimaryKeyConstraint("id"),
67
+ sa.UniqueConstraint("connector_name"),
68
+ )
69
+ op.create_table(
70
+ "disabled_rules",
71
+ sa.Column("id", sa.Integer(), nullable=False),
72
+ sa.Column("rule_id", sa.String(length=100), nullable=True),
73
+ sa.Column("previous_level", sa.String(length=1000), nullable=True),
74
+ sa.Column("new_level", sa.String(length=1000), nullable=True),
75
+ sa.Column("reason_for_disabling", sa.String(length=100), nullable=True),
76
+ sa.Column("date_disabled", sa.DateTime(), nullable=True),
77
+ sa.Column("length_of_time", sa.Integer(), nullable=True),
78
+ sa.PrimaryKeyConstraint("id"),
79
+ )
80
+ op.create_table(
81
+ "graylog_metrics_allocation",
82
+ sa.Column("id", sa.Integer(), nullable=False),
83
+ sa.Column("input_usage", sa.Float(), nullable=True),
84
+ sa.Column("output_usage", sa.Float(), nullable=True),
85
+ sa.Column("processor_usage", sa.Float(), nullable=True),
86
+ sa.Column("input_1_sec_rate", sa.Float(), nullable=True),
87
+ sa.Column("output_1_sec_rate", sa.Float(), nullable=True),
88
+ sa.Column("total_input", sa.Float(), nullable=True),
89
+ sa.Column("total_output", sa.Float(), nullable=True),
90
+ sa.Column("timestamp", sa.DateTime(), nullable=True),
91
+ sa.PrimaryKeyConstraint("id"),
92
+ )
93
+ op.create_table(
94
+ "wazuh_indexer_allocation",
95
+ sa.Column("id", sa.Integer(), nullable=False),
96
+ sa.Column("node", sa.String(length=100), nullable=True),
97
+ sa.Column("disk_used", sa.Float(), nullable=True),
98
+ sa.Column("disk_available", sa.Float(), nullable=True),
99
+ sa.Column("disk_total", sa.Float(), nullable=True),
100
+ sa.Column("disk_percent", sa.Float(), nullable=True),
101
+ sa.Column("timestamp", sa.DateTime(), nullable=True),
102
+ sa.PrimaryKeyConstraint("id"),
103
+ )
104
# ### end Alembic commands ###
105
106