mirror of
https://github.com/apache/superset.git
synced 2026-05-11 10:55:43 +00:00
b59f6b1833
* added new tables for alerting refractor * reformatted inheritance structure * added workflow for updated framework * added suggested changes * cleaned up changes * added obervations to alert table to enable view * added comments * added requested changes * fix tests * added styling changes * mypy * added requested changes * updated operator logic * requested changes, 1 validator, styling changes * refactored tests * fix test alert workflow * fixed create_alert in test Co-authored-by: Jason Davis <@dropbox.com>
126 lines
5.3 KiB
Python
126 lines
5.3 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
"""refractor_alerting
|
|
|
|
Revision ID: 2e5a0ee25ed4
|
|
Revises: f80a3b88324b
|
|
Create Date: 2020-08-31 20:30:30.781478
|
|
|
|
"""
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "2e5a0ee25ed4"
|
|
down_revision = "f80a3b88324b"
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy.dialects import mysql
|
|
|
|
|
|
def upgrade():
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table(
|
|
"alert_validators",
|
|
sa.Column("created_on", sa.DateTime(), nullable=True),
|
|
sa.Column("changed_on", sa.DateTime(), nullable=True),
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("validator_type", sa.String(length=100), nullable=False),
|
|
sa.Column("config", sa.Text(), nullable=True),
|
|
sa.Column("created_by_fk", sa.Integer(), nullable=True),
|
|
sa.Column("changed_by_fk", sa.Integer(), nullable=True),
|
|
sa.Column("alert_id", sa.Integer(), nullable=False),
|
|
sa.ForeignKeyConstraint(["alert_id"], ["alerts.id"],),
|
|
sa.ForeignKeyConstraint(["changed_by_fk"], ["ab_user.id"],),
|
|
sa.ForeignKeyConstraint(["created_by_fk"], ["ab_user.id"],),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_table(
|
|
"sql_observers",
|
|
sa.Column("created_on", sa.DateTime(), nullable=True),
|
|
sa.Column("changed_on", sa.DateTime(), nullable=True),
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("sql", sa.Text(), nullable=False),
|
|
sa.Column("created_by_fk", sa.Integer(), nullable=True),
|
|
sa.Column("changed_by_fk", sa.Integer(), nullable=True),
|
|
sa.Column("alert_id", sa.Integer(), nullable=False),
|
|
sa.Column("database_id", sa.Integer(), nullable=False),
|
|
sa.ForeignKeyConstraint(["alert_id"], ["alerts.id"],),
|
|
sa.ForeignKeyConstraint(["changed_by_fk"], ["ab_user.id"],),
|
|
sa.ForeignKeyConstraint(["created_by_fk"], ["ab_user.id"],),
|
|
sa.ForeignKeyConstraint(["database_id"], ["dbs.id"],),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_table(
|
|
"sql_observations",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("dttm", sa.DateTime(), nullable=True),
|
|
sa.Column("observer_id", sa.Integer(), nullable=False),
|
|
sa.Column("alert_id", sa.Integer(), nullable=True),
|
|
sa.Column("value", sa.Float(), nullable=True),
|
|
sa.Column("error_msg", sa.String(length=500), nullable=True),
|
|
sa.ForeignKeyConstraint(["alert_id"], ["alerts.id"],),
|
|
sa.ForeignKeyConstraint(["observer_id"], ["sql_observers.id"],),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(
|
|
op.f("ix_sql_observations_dttm"), "sql_observations", ["dttm"], unique=False
|
|
)
|
|
|
|
with op.batch_alter_table("alerts") as batch_op:
|
|
batch_op.add_column(sa.Column("changed_by_fk", sa.Integer(), nullable=True))
|
|
batch_op.add_column(sa.Column("changed_on", sa.DateTime(), nullable=True))
|
|
batch_op.add_column(sa.Column("created_by_fk", sa.Integer(), nullable=True))
|
|
batch_op.add_column(sa.Column("created_on", sa.DateTime(), nullable=True))
|
|
batch_op.alter_column(
|
|
"crontab", existing_type=mysql.VARCHAR(length=50), nullable=False
|
|
)
|
|
batch_op.create_foreign_key(
|
|
"alerts_ibfk_3", "ab_user", ["changed_by_fk"], ["id"]
|
|
)
|
|
batch_op.create_foreign_key(
|
|
"alerts_ibfk_4", "ab_user", ["created_by_fk"], ["id"]
|
|
)
|
|
batch_op.drop_column("sql")
|
|
batch_op.drop_column("database_id")
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade():
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
with op.batch_alter_table("alerts") as batch_op:
|
|
batch_op.add_column(
|
|
sa.Column(
|
|
"database_id", mysql.INTEGER(), autoincrement=False, nullable=False
|
|
)
|
|
)
|
|
batch_op.add_column(sa.Column("sql", mysql.TEXT(), nullable=True))
|
|
batch_op.drop_constraint("alerts_ibfk_3", type_="foreignkey")
|
|
batch_op.drop_constraint("alerts_ibfk_4", type_="foreignkey")
|
|
batch_op.alter_column(
|
|
"crontab", existing_type=mysql.VARCHAR(length=50), nullable=True
|
|
)
|
|
batch_op.drop_column("created_on")
|
|
batch_op.drop_column("created_by_fk")
|
|
batch_op.drop_column("changed_on")
|
|
batch_op.drop_column("changed_by_fk")
|
|
|
|
op.drop_index(op.f("ix_sql_observations_dttm"), table_name="sql_observations")
|
|
op.drop_table("sql_observations")
|
|
op.drop_table("sql_observers")
|
|
op.drop_table("alert_validators")
|
|
# ### end Alembic commands ###
|