fix: Migration for single metric in Big Number with Time Comparison (#27351)

This commit is contained in:
Kamil Gabryjelski 2024-03-01 11:43:22 +01:00 committed by GitHub
parent d5814003dd
commit ad6327db95
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 93 additions and 1 deletions

View File

@ -143,7 +143,6 @@ export default function transformProps(chartProps: ChartProps) {
width,
height,
data,
metric,
metricName,
bigNumber,
prevNumber,

View File

@ -0,0 +1,93 @@
# 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.
"""big_number_kpi_single_metric
Revision ID: be1b217cd8cd
Revises: 17fcea065655
Create Date: 2024-03-01 10:47:42.373508
"""
# revision identifiers, used by Alembic.
revision = "be1b217cd8cd"
down_revision = "17fcea065655"
import json
from alembic import op
from sqlalchemy import Column, Integer, String, Text
from sqlalchemy.ext.declarative import declarative_base
from superset import db
from superset.migrations.shared.utils import paginated_update
Base = declarative_base()
class Slice(Base):
__tablename__ = "slices"
id = Column(Integer, primary_key=True)
params = Column(Text)
viz_type = Column(String(250))
def upgrade():
bind = op.get_bind()
session = db.Session(bind=bind)
for slc in paginated_update(
session.query(Slice).filter(Slice.viz_type == "pop_kpi")
):
try:
params = json.loads(slc.params)
if "metrics" in params:
if params["metrics"]:
params["metric"] = params["metrics"][0]
del params["metrics"]
slc.params = json.dumps(params, sort_keys=True)
except Exception:
pass
session.commit()
session.close()
def downgrade():
bind = op.get_bind()
session = db.Session(bind=bind)
for slc in paginated_update(
session.query(Slice).filter(Slice.viz_type == "pop_kpi")
):
try:
params = json.loads(slc.params)
if "metric" in params:
if params["metric"]:
params["metrics"] = [params["metric"]]
del params["metric"]
slc.params = json.dumps(params, sort_keys=True)
except Exception:
pass
session.commit()
session.close()