-
-
Notifications
You must be signed in to change notification settings - Fork 4.2k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
3 changed files
with
95 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
92 changes: 92 additions & 0 deletions
92
src/sentry/migrations/0550_migrate_no_action_dupe_issue_alerts.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,92 @@ | ||
# Generated by Django 3.2.20 on 2023-09-12 18:15 | ||
|
||
from django.db import migrations | ||
|
||
from sentry.new_migrations.migrations import CheckedMigration | ||
from sentry.utils.query import RangeQuerySetWrapperWithProgressBar | ||
|
||
|
||
class ObjectStatus: | ||
ACTIVE = 0 | ||
HIDDEN = 1 | ||
PENDING_DELETION = 2 | ||
DELETION_IN_PROGRESS = 3 | ||
|
||
DISABLED = 1 | ||
|
||
|
||
def has_duplicate_rule(apps, rule_data, project, rule_id): | ||
Rule = apps.get_model("sentry", "Rule") | ||
|
||
matchers = {key for key in list(rule_data.keys()) if key not in ("name", "user_id")} | ||
extra_fields = ["actions", "environment"] | ||
matchers.update(extra_fields) | ||
existing_rules = Rule.objects.exclude(id=rule_id).filter( | ||
project=project, status=ObjectStatus.ACTIVE | ||
) | ||
for existing_rule in existing_rules: | ||
keys = 0 | ||
matches = 0 | ||
for matcher in matchers: | ||
if existing_rule.data.get(matcher) and rule_data.get(matcher): | ||
keys += 1 | ||
|
||
if existing_rule.data[matcher] == rule_data[matcher]: | ||
matches += 1 | ||
|
||
elif matcher in extra_fields: | ||
if not existing_rule.data.get(matcher) and not rule_data.get(matcher): | ||
# neither rule has the matcher | ||
continue | ||
|
||
elif matcher == "environment": | ||
if existing_rule.environment_id and rule_data.get(matcher): | ||
keys += 1 | ||
if existing_rule.environment_id == rule_data.get(matcher): | ||
matches += 1 | ||
else: | ||
keys += 1 | ||
else: | ||
# one rule has the matcher and the other one doesn't | ||
keys += 1 | ||
|
||
if keys == matches: | ||
return True | ||
return False | ||
|
||
|
||
def migrate_bad_rules(apps, schema_editor): | ||
Rule = apps.get_model("sentry", "Rule") | ||
|
||
for rule in RangeQuerySetWrapperWithProgressBar(Rule.objects.all()): | ||
if not rule.data.get("actions", []) or has_duplicate_rule( | ||
apps, rule.data, rule.project, rule.id | ||
): | ||
rule.status = ObjectStatus.DISABLED | ||
rule.save(update_fields=["status"]) | ||
|
||
|
||
class Migration(CheckedMigration): | ||
# This flag is used to mark that a migration shouldn't be automatically run in production. For | ||
# the most part, this should only be used for operations where it's safe to run the migration | ||
# after your code has deployed. So this should not be used for most operations that alter the | ||
# schema of a table. | ||
# Here are some things that make sense to mark as dangerous: | ||
# - Large data migrations. Typically we want these to be run manually by ops so that they can | ||
# be monitored and not block the deploy for a long period of time while they run. | ||
# - Adding indexes to large tables. Since this can take a long time, we'd generally prefer to | ||
# have ops run this and not block the deploy. Note that while adding an index is a schema | ||
# change, it's completely safe to run the operation after the code has deployed. | ||
is_dangerous = True | ||
|
||
dependencies = [ | ||
("sentry", "0549_re_add_groupsubscription_columns"), | ||
] | ||
|
||
operations = [ | ||
migrations.RunPython( | ||
migrate_bad_rules, | ||
migrations.RunPython.noop, | ||
hints={"tables": ["sentry_rule"]}, | ||
), | ||
] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters