Skip to content

Commit b96263c

Browse files
committed
ref(py): Normalize deletions load_defaults imports
1 parent 480efb7 commit b96263c

File tree

2 files changed

+43
-44
lines changed

2 files changed

+43
-44
lines changed

Diff for: src/sentry/deletions/__init__.py

+33-44
Original file line numberDiff line numberDiff line change
@@ -91,47 +91,28 @@
9191

9292
def load_defaults(manager: DeletionTaskManager) -> None:
9393
from sentry import models
94-
from sentry.deletions.base import BulkModelDeletionTask
95-
from sentry.discover.models import DiscoverSavedQuery, DiscoverSavedQueryProject
96-
from sentry.incidents.models.alert_rule import (
97-
AlertRule,
98-
AlertRuleTrigger,
99-
AlertRuleTriggerAction,
100-
)
101-
from sentry.incidents.models.incident import Incident
102-
from sentry.integrations.models.organization_integration import OrganizationIntegration
103-
from sentry.integrations.models.repository_project_path_config import (
104-
RepositoryProjectPathConfig,
105-
)
106-
from sentry.models.commitfilechange import CommitFileChange
107-
from sentry.models.rulefirehistory import RuleFireHistory
94+
from sentry.discover import models as discover
95+
from sentry.incidents import models as incidents
96+
from sentry.integrations import models as integrations
10897
from sentry.monitors import models as monitor_models
109-
from sentry.sentry_apps.models.platformexternalissue import PlatformExternalIssue
110-
from sentry.sentry_apps.models.sentry_app import SentryApp
111-
from sentry.sentry_apps.models.sentry_app_installation import SentryAppInstallation
112-
from sentry.sentry_apps.models.sentry_app_installation_token import SentryAppInstallationToken
113-
from sentry.sentry_apps.models.servicehook import ServiceHook
114-
from sentry.snuba import models as snuba_models
115-
from sentry.workflow_engine.models import DataSource, Detector, Workflow
98+
from sentry.sentry_apps import models as sentry_apps
99+
from sentry.snuba import models as snuba
100+
from sentry.workflow_engine import models as workflow_engine
116101

117102
from . import defaults
103+
from .base import BulkModelDeletionTask
118104

105+
# fmt: off
119106
manager.register(models.Activity, BulkModelDeletionTask)
120-
manager.register(AlertRule, defaults.AlertRuleDeletionTask)
121-
manager.register(AlertRuleTrigger, defaults.AlertRuleTriggerDeletionTask)
122-
manager.register(AlertRuleTriggerAction, defaults.AlertRuleTriggerActionDeletionTask)
123107
manager.register(models.ApiApplication, defaults.ApiApplicationDeletionTask)
124108
manager.register(models.ApiGrant, BulkModelDeletionTask)
125109
manager.register(models.ApiKey, BulkModelDeletionTask)
126110
manager.register(models.ApiToken, BulkModelDeletionTask)
111+
manager.register(models.ArtifactBundle, defaults.ArtifactBundleDeletionTask)
127112
manager.register(models.Commit, defaults.CommitDeletionTask)
128113
manager.register(models.CommitAuthor, defaults.CommitAuthorDeletionTask)
129-
manager.register(CommitFileChange, BulkModelDeletionTask)
130-
manager.register(Detector, defaults.DetectorDeletionTask)
131-
manager.register(DataSource, defaults.DataSourceDeletionTask)
114+
manager.register(models.CommitFileChange, BulkModelDeletionTask)
132115
manager.register(models.Deploy, BulkModelDeletionTask)
133-
manager.register(DiscoverSavedQuery, defaults.DiscoverSavedQueryDeletionTask)
134-
manager.register(DiscoverSavedQueryProject, BulkModelDeletionTask)
135116
manager.register(models.Distribution, BulkModelDeletionTask)
136117
manager.register(models.EnvironmentProject, BulkModelDeletionTask)
137118
manager.register(models.Group, defaults.GroupDeletionTask)
@@ -153,38 +134,46 @@ def load_defaults(manager: DeletionTaskManager) -> None:
153134
manager.register(models.GroupShare, BulkModelDeletionTask)
154135
manager.register(models.GroupSnooze, BulkModelDeletionTask)
155136
manager.register(models.GroupSubscription, BulkModelDeletionTask)
156-
manager.register(Incident, defaults.IncidentDeletionTask)
157-
manager.register(monitor_models.Monitor, defaults.MonitorDeletionTask)
158-
manager.register(monitor_models.MonitorEnvironment, defaults.MonitorEnvironmentDeletionTask)
159137
manager.register(models.Organization, defaults.OrganizationDeletionTask)
160-
manager.register(OrganizationIntegration, defaults.OrganizationIntegrationDeletionTask)
161138
manager.register(models.OrganizationMember, defaults.OrganizationMemberDeletionTask)
162139
manager.register(models.OrganizationMemberTeam, BulkModelDeletionTask)
163-
manager.register(PlatformExternalIssue, defaults.PlatformExternalIssueDeletionTask)
164140
manager.register(models.Project, defaults.ProjectDeletionTask)
165141
manager.register(models.ProjectBookmark, BulkModelDeletionTask)
166142
manager.register(models.ProjectKey, BulkModelDeletionTask)
167143
manager.register(models.PullRequest, defaults.PullRequestDeletionTask)
168-
manager.register(snuba_models.QuerySubscription, defaults.QuerySubscriptionDeletionTask)
169144
manager.register(models.Release, defaults.ReleaseDeletionTask)
170145
manager.register(models.ReleaseCommit, BulkModelDeletionTask)
171146
manager.register(models.ReleaseEnvironment, BulkModelDeletionTask)
172147
manager.register(models.ReleaseHeadCommit, BulkModelDeletionTask)
173148
manager.register(models.ReleaseProject, BulkModelDeletionTask)
174149
manager.register(models.ReleaseProjectEnvironment, BulkModelDeletionTask)
175150
manager.register(models.Repository, defaults.RepositoryDeletionTask)
176-
manager.register(RepositoryProjectPathConfig, defaults.RepositoryProjectPathConfigDeletionTask)
177-
manager.register(SentryApp, defaults.SentryAppDeletionTask)
178-
manager.register(SentryAppInstallation, defaults.SentryAppInstallationDeletionTask)
179-
manager.register(SentryAppInstallationToken, defaults.SentryAppInstallationTokenDeletionTask)
180-
manager.register(ServiceHook, defaults.ServiceHookDeletionTask)
151+
manager.register(models.Rule, defaults.RuleDeletionTask)
152+
manager.register(models.RuleFireHistory, defaults.RuleFireHistoryDeletionTask)
181153
manager.register(models.SavedSearch, BulkModelDeletionTask)
182154
manager.register(models.Team, defaults.TeamDeletionTask)
183155
manager.register(models.UserReport, BulkModelDeletionTask)
184-
manager.register(models.ArtifactBundle, defaults.ArtifactBundleDeletionTask)
185-
manager.register(models.Rule, defaults.RuleDeletionTask)
186-
manager.register(RuleFireHistory, defaults.RuleFireHistoryDeletionTask)
187-
manager.register(Workflow, defaults.WorkflowDeletionTask)
156+
157+
manager.register(discover.DiscoverSavedQuery, defaults.DiscoverSavedQueryDeletionTask)
158+
manager.register(discover.DiscoverSavedQueryProject, BulkModelDeletionTask)
159+
manager.register(incidents.AlertRule, defaults.AlertRuleDeletionTask)
160+
manager.register(incidents.AlertRuleTrigger, defaults.AlertRuleTriggerDeletionTask)
161+
manager.register(incidents.AlertRuleTriggerAction, defaults.AlertRuleTriggerActionDeletionTask)
162+
manager.register(incidents.Incident, defaults.IncidentDeletionTask)
163+
manager.register(integrations.OrganizationIntegration, defaults.OrganizationIntegrationDeletionTask)
164+
manager.register(integrations.RepositoryProjectPathConfig, defaults.RepositoryProjectPathConfigDeletionTask)
165+
manager.register(monitor_models.Monitor, defaults.MonitorDeletionTask)
166+
manager.register(monitor_models.MonitorEnvironment, defaults.MonitorEnvironmentDeletionTask)
167+
manager.register(sentry_apps.PlatformExternalIssue, defaults.PlatformExternalIssueDeletionTask)
168+
manager.register(sentry_apps.SentryApp, defaults.SentryAppDeletionTask)
169+
manager.register(sentry_apps.SentryAppInstallation, defaults.SentryAppInstallationDeletionTask)
170+
manager.register(sentry_apps.SentryAppInstallationToken, defaults.SentryAppInstallationTokenDeletionTask)
171+
manager.register(sentry_apps.ServiceHook, defaults.ServiceHookDeletionTask)
172+
manager.register(snuba.QuerySubscription, defaults.QuerySubscriptionDeletionTask)
173+
manager.register(workflow_engine.DataSource, defaults.DataSourceDeletionTask)
174+
manager.register(workflow_engine.Detector, defaults.DetectorDeletionTask)
175+
manager.register(workflow_engine.Workflow, defaults.WorkflowDeletionTask)
176+
# fmt: on
188177

189178

190179
_default_manager = None

Diff for: src/sentry/incidents/models/__init__.py

+10
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
__all__ = (
2+
"AlertRule",
3+
"AlertRuleActivity",
4+
"AlertRuleTrigger",
5+
"AlertRuleTriggerAction",
6+
"Incident",
7+
)
8+
9+
from .alert_rule import AlertRule, AlertRuleActivity, AlertRuleTrigger, AlertRuleTriggerAction
10+
from .incident import Incident

0 commit comments

Comments
 (0)