Skip to content

Commit 62075a9

Browse files
ref(py): Consistently use import * in models (#89444)
I know we're trying to just straight remove these, but let's at least be consistent for now so that mypy understands that these models are available from this module. Needed for this simplification #89440
1 parent f70e42b commit 62075a9

File tree

7 files changed

+20
-7
lines changed

7 files changed

+20
-7
lines changed

src/sentry/models/__init__.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
from .broadcast import * # NOQA
1717
from .commit import * # NOQA
1818
from .commitauthor import * # NOQA
19-
from .commitfilechange import CommitFileChange # noqa
19+
from .commitfilechange import * # noqa
2020
from .counter import * # NOQA
2121
from .dashboard import * # NOQA
2222
from .dashboard_permissions import * # NOQA
@@ -69,7 +69,7 @@
6969
from .organizationaccessrequest import * # NOQA
7070
from .organizationmapping import * # NOQA
7171
from .organizationmember import * # NOQA
72-
from .organizationmemberinvite import OrganizationMemberInvite # NOQA
72+
from .organizationmemberinvite import * # NOQA
7373
from .organizationmembermapping import * # NOQA
7474
from .organizationmemberteam import * # NOQA
7575
from .organizationmemberteamreplica import * # NOQA
@@ -81,12 +81,12 @@
8181
from .projectbookmark import * # NOQA
8282
from .projectcodeowners import * # NOQA
8383
from .projectkey import * # NOQA
84-
from .projectownership import ProjectOwnership # NOQA
84+
from .projectownership import * # NOQA
8585
from .projectplatform import * # NOQA
8686
from .projectredirect import * # NOQA
8787
from .projectsdk import * # NOQA
88-
from .projectteam import ProjectTeam # noqa
89-
from .projecttemplate import ProjectTemplate # noqa
88+
from .projectteam import * # NOQA
89+
from .projecttemplate import * # NOQA
9090
from .promptsactivity import * # NOQA
9191
from .pullrequest import * # NOQA
9292
from .recentsearch import * # NOQA
@@ -104,8 +104,8 @@
104104
from .rollbackorganization import * # NOQA
105105
from .rollbackuser import * # NOQA
106106
from .rule import * # NOQA
107-
from .rulefirehistory import RuleFireHistory # NOQA
108-
from .rulesnooze import RuleSnooze # NOQA
107+
from .rulefirehistory import * # NOQA
108+
from .rulesnooze import * # NOQA
109109
from .savedsearch import * # NOQA
110110
from .search_common import * # NOQA
111111
from .sentryshot import * # NOQA

src/sentry/models/commitfilechange.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,9 @@
1717
COMMIT_FILE_CHANGE_TYPES = frozenset(("A", "D", "M"))
1818

1919

20+
__all__ = ("CommitFileChange",)
21+
22+
2023
class CommitFileChangeManager(BaseManager["CommitFileChange"]):
2124
def get_count_for_commits(self, commits: Iterable[Any]) -> int:
2225
return int(self.filter(commit__in=commits).values("filename").distinct().count())

src/sentry/models/organizationmemberinvite.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@
1818

1919
INVITE_DAYS_VALID = 30
2020

21+
__all__ = ("OrganizationMemberInvite",)
22+
2123

2224
class InviteStatus(Enum):
2325
APPROVED = 0

src/sentry/models/projectownership.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@
2929
from sentry.models.team import Team
3030
from sentry.users.services.user import RpcUser
3131

32+
__all__ = ("ProjectOwnership",)
33+
3234
logger = logging.getLogger(__name__)
3335
READ_CACHE_DURATION = 3600
3436

src/sentry/models/projectteam.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@
1313
if TYPE_CHECKING:
1414
from sentry.models.team import Team
1515

16+
__all__ = ("ProjectTeam",)
17+
1618

1719
class ProjectTeamManager(BaseManager["ProjectTeam"]):
1820
def get_for_teams_with_org_cache(self, teams: Sequence["Team"]) -> QuerySet["ProjectTeam"]:

src/sentry/models/projecttemplate.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,8 @@
88
sane_repr,
99
)
1010

11+
__all__ = ("ProjectTemplate",)
12+
1113

1214
@region_silo_model
1315
class ProjectTemplate(DefaultFieldsModelExisting):

src/sentry/models/rulesnooze.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@
99
from sentry.db.models.fields.hybrid_cloud_foreign_key import HybridCloudForeignKey
1010
from sentry.db.models.manager.base import BaseManager
1111

12+
__all__ = ("RuleSnooze",)
13+
1214

1315
class RuleSnoozeManager(BaseManager["RuleSnooze"]):
1416
def is_snoozed_for_all(self, rule=None, alert_rule=None):

0 commit comments

Comments
 (0)