Skip to content

ref: fix types for sentry.api.base #89601

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Apr 15, 2025
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,6 @@ ignore_missing_imports = true
# - python3 -m tools.mypy_helpers.find_easiest_modules
[[tool.mypy.overrides]]
module = [
"sentry.api.base",
"sentry.api.endpoints.group_integration_details",
"sentry.api.endpoints.organization_events_meta",
"sentry.api.endpoints.organization_events_spans_performance",
Expand Down
2 changes: 2 additions & 0 deletions src/sentry/api/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -401,6 +401,7 @@ def dispatch(self, request: Request, *args, **kwargs) -> Response:
self.initial(request, *args, **kwargs)

# Get the appropriate handler method
assert request.method is not None
method = request.method.lower()
if method in self.http_method_names and hasattr(self, method):
handler = getattr(self, method)
Expand Down Expand Up @@ -445,6 +446,7 @@ def dispatch(self, request: Request, *args, **kwargs) -> Response:

# Only enforced in dev environment
if settings.ENFORCE_PAGINATION:
assert request.method is not None
if request.method.lower() == "get":
status = getattr(self.response, "status_code", 0)
# Response can either be Response or HttpResponse, check if
Expand Down
2 changes: 1 addition & 1 deletion src/sentry/models/apiapplication.py
Original file line number Diff line number Diff line change
Expand Up @@ -145,7 +145,7 @@ def is_valid_redirect_uri(self, value):
def get_default_redirect_uri(self):
return self.redirect_uris.split()[0]

def get_allowed_origins(self):
def get_allowed_origins(self) -> list[str]:
if not self.allowed_origins:
return []
return [origin for origin in self.allowed_origins.split()]
Expand Down
2 changes: 1 addition & 1 deletion src/sentry/models/apikey.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ def save(self, *args, **kwargs):
self.key = ApiKey.generate_api_key()
super().save(*args, **kwargs)

def get_allowed_origins(self):
def get_allowed_origins(self) -> list[str]:
if not self.allowed_origins:
return []
return list(filter(bool, self.allowed_origins.split("\n")))
Expand Down
4 changes: 2 additions & 2 deletions src/sentry/models/apitoken.py
Original file line number Diff line number Diff line change
Expand Up @@ -299,10 +299,10 @@ def is_expired(self):
def get_audit_log_data(self):
return {"scopes": self.get_scopes()}

def get_allowed_origins(self):
def get_allowed_origins(self) -> list[str]:
if self.application:
return self.application.get_allowed_origins()
return ()
return []

def refresh(self, expires_at=None):
if self.token_type == AuthTokenType.USER:
Expand Down
2 changes: 1 addition & 1 deletion src/sentry/models/orgauthtoken.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ def __str__(self):
def get_audit_log_data(self):
return {"name": self.name, "scopes": self.get_scopes()}

def get_allowed_origins(self):
def get_allowed_origins(self) -> list[str]:
return []

def get_scopes(self):
Expand Down
2 changes: 1 addition & 1 deletion src/sentry/models/projectkey.py
Original file line number Diff line number Diff line change
Expand Up @@ -310,7 +310,7 @@ def get_endpoint(self) -> str:

return endpoint

def get_allowed_origins(self):
def get_allowed_origins(self) -> frozenset[str]:
from sentry.utils.http import get_origins

return get_origins(self.project)
Expand Down
3 changes: 2 additions & 1 deletion src/sentry/utils/http.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from __future__ import annotations

from collections.abc import Collection
from typing import TYPE_CHECKING, NamedTuple, TypeGuard, overload
from urllib.parse import quote, urljoin, urlparse

Expand Down Expand Up @@ -103,7 +104,7 @@ def parse_uri_match(value: str) -> ParsedUriMatch:


def is_valid_origin(
origin: str | None, project: Project | None = None, allowed: frozenset[str] | None = None
origin: str | None, project: Project | None = None, allowed: Collection[str] | None = None
) -> bool:
"""
Given an ``origin`` which matches a base URI (e.g. http://example.com)
Expand Down
Loading