from sfgdpr import client
BBC = BumblebeeClient()
KPC = client.KeyLimePIIClient()
def set_table_temporary(namespace, table):
resource = f"{namespace}.{table}"
print(f"Marking {resource} temporary...")
perm_result = KPC.set_table_permanence(resource, is_permanent=False)
status = KPC.get_compliance_status(resource)
assert status['success'] == True
assert status['result'][0] == 'compliant'
def set_namespace_temporary(namespace):
ns = BBC.get_database(namespace)
tables = [t["table_name"] for t in ns["tables"]]
for table in tables:
set_table_temporary(namespace, table)
print("Done! 🎉")
def set_tables_temporary(namespace, tables):
for table in tables:
set_table_temporary(namespace, table)
set_namespace_temporary('mercury_dev_mzhao')