fix: migrate_roles (#16098)

This commit is contained in:
Beto Dealmeida 2021-08-06 10:05:19 -07:00 committed by GitHub
parent 772da8de63
commit 28c383af68
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions

View File

@ -38,6 +38,7 @@ from superset.migrations.shared.security_converge import (
revision = "f6196627326f"
down_revision = "143b6f2815da"
NEW_PVMS = {"Chart": ("can_read",)}
PVM_MAP = {
Pvm("Chart", "can_get_data"): (Pvm("Chart", "can_read"),),
Pvm("Chart", "can_post_data"): (Pvm("Chart", "can_read"),),
@ -49,6 +50,7 @@ def upgrade():
session = Session(bind=bind)
# Add the new permissions on the migration itself
add_pvms(session, NEW_PVMS)
migrate_roles(session, PVM_MAP)
try:
session.commit()