def migrate_subscriptions(apps, schema_editor):
QuerySubscription = apps.get_model("sentry", "QuerySubscription")
AppSnubaQueryEventType = apps.get_model("sentry", "SnubaQueryEventType")
for subscription in RangeQuerySetWrapperWithProgressBar(
QuerySubscription.objects.select_related("snuba_query").all()):
if subscription.subscription_id is not None:
# The migration apps don't build this property, so manually set it.
raw_event_types = AppSnubaQueryEventType.objects.filter(
snuba_query=subscription.snuba_query).all()
event_types = [
SnubaQueryEventType.EventType(ev.type)
for ev in raw_event_types
]
setattr(subscription.snuba_query, "event_types", event_types)
subscription_id = None
try:
subscription_id = _create_in_snuba(subscription)
except Exception as e:
logging.exception(
f"failed to recreate {subscription.subscription_id}: {e}")
continue
try:
_delete_from_snuba(
QueryDatasets(subscription.snuba_query.dataset),
subscription.subscription_id,
)
except Exception as e:
try:
# Delete the subscription we just created to avoid orphans
_delete_from_snuba(
QueryDatasets(subscription.snuba_query.dataset),
subscription_id,
)
except Exception as oe:
logging.exception(
f"failed to delete orphan {subscription_id}: {oe}")
logging.exception(
f"failed to delete {subscription.subscription_id}: {e}")
continue
QuerySubscription.objects.filter(id=subscription.id).update(
subscription_id=subscription_id)
def get_attrs(self, item_list, user, **kwargs):
result = super().get_attrs(item_list, user, **kwargs)
alert_rules = {item.id: item for item in item_list}
for alert_rule_id, project_slug in AlertRuleExcludedProjects.objects.filter(
alert_rule__in=item_list
).values_list("alert_rule_id", "project__slug"):
exclusions = result[alert_rules[alert_rule_id]].setdefault("excluded_projects", [])
exclusions.append(project_slug)
query_to_alert_rule = {ar.snuba_query_id: ar for ar in item_list}
for event_type in SnubaQueryEventType.objects.filter(
snuba_query_id__in=[item.snuba_query_id for item in item_list]
):
event_types = result[query_to_alert_rule[event_type.snuba_query_id]].setdefault(
"event_types", []
)
event_types.append(SnubaQueryEventType.EventType(event_type.type).name.lower())
return result