Skip to content

Commit 50a31b6

Browse files
committed
fix drg revoke
1 parent e02baa9 commit 50a31b6

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

titan/lifecycle.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -431,12 +431,16 @@ def drop_database(urn: URN, data: dict, if_exists: bool) -> str:
431431

432432

433433
def drop_database_role_grant(urn: URN, data: dict, **kwargs):
434+
435+
from_type = "ROLE" if data["to_role"] else "DATABASE ROLE"
436+
from_name = data["to_role"] if data["to_role"] else data["to_database_role"]
437+
434438
return tidy_sql(
435439
"REVOKE DATABASE ROLE",
436440
ResourceName(data["database_role"]),
437441
"FROM",
438-
"ROLE",
439-
ResourceName(data["to_role"] if data["to_role"] else data["to_database_role"]),
442+
from_type,
443+
ResourceName(from_name),
440444
)
441445

442446

0 commit comments

Comments
 (0)