def delete_property(
db: PartitionedDatabase,
model_id: str,
property_name: str,
modify_records: bool = False,
) -> None:
x_bf_trace_id = AuditLogger.trace_id_header()
max_record_count = current_app.config[
"config"].max_record_count_for_property_deletion
with db.transaction() as tx:
model = db.get_model_tx(tx, model_id)
if modify_records:
record_count = db.model_property_record_count_tx(
tx, model_id, property_name)
if record_count > 0:
if record_count > max_record_count:
raise BadRequest(
f"Cannot delete properties that are used on > {max_record_count} records. This property is used on {record_count}"
)
model_properties = [
p for p in db.get_properties_tx(tx, model_id)
if p.name == property_name
]
if not model_properties:
raise NotFound(f"no such property {property_name} exists")
updated_records = db.delete_property_from_all_records_tx(
tx, model_id, model_properties[0])
if updated_records != record_count:
raise ServerError(
"the property was not removed from all records")
deleted = db.delete_property_tx(tx, model_id, property_name)
if deleted is None:
raise NotFound(
f"Could not delete property [{model_id}.{property_name}]")
PennsieveJobsClient.get().send_changelog_event(
organization_id=db.organization_id,
dataset_id=db.dataset_id,
user_id=db.user_id,
event=DeleteModelProperty(
property_name=deleted.name,
model_id=UUID(model.id),
model_name=model.name,
),
trace_id=TraceId(x_bf_trace_id),
)
def delete_property(db: PartitionedDatabase, concept_id_or_name: str,
property_id: str) -> None:
x_bf_trace_id = AuditLogger.trace_id_header()
with db.transaction() as tx:
model = db.get_model_tx(tx, concept_id_or_name)
deleted = db.delete_property_tx(tx, model, property_id)
if deleted is None:
raise NotFound(
f"Could not find property {property_id} of model {concept_id_or_name}"
)
# Emit "UpdateModel" event:
PennsieveJobsClient.get().send_changelog_event(
organization_id=db.organization_id,
dataset_id=db.dataset_id,
user_id=db.user_id,
event=DeleteModelProperty(
property_name=deleted.name,
model_id=UUID(model.id),
model_name=model.name,
),
trace_id=TraceId(x_bf_trace_id),
)