Files
atc-agents/infra/historical/superset_repoint_s3.py
T

25 lines
801 B
Python

from superset.app import create_app
app = create_app()
with app.app_context():
from superset import db
from superset.connectors.sqla.models import SqlaTable
from superset.models.core import Database
t = db.session.query(SqlaTable).filter_by(table_name="historical_sales").first()
ice = (
db.session.query(Database)
.filter(Database.database_name.like("%Iceberg Lakehouse%"))
.first()
)
t.database_id = ice.id
t.schema = "s3lake"
db.session.commit()
try:
t.fetch_metadata()
db.session.commit()
except Exception as e:
print("fetch_metadata warn:", str(e)[:200])
print("repointed dataset", t.id, "-> db", ice.id, ice.database_name,
"schema", t.schema, "cols", [c.column_name for c in t.columns])