50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
from typing import Literal
|
|
|
|
from ninja import Field
|
|
|
|
from activities.models import PostInteraction
|
|
from activities.search import Searcher
|
|
from api import schemas
|
|
from api.decorators import identity_required
|
|
from api.views.base import api_router
|
|
|
|
|
|
@api_router.get("/v2/search", response=schemas.Search)
|
|
@identity_required
|
|
def search(
|
|
request,
|
|
q: str,
|
|
type: Literal["accounts", "hashtags", "statuses"] | None = None,
|
|
fetch_identities: bool = Field(False, alias="resolve"),
|
|
following: bool = False,
|
|
exclude_unreviewed: bool = False,
|
|
account_id: str | None = None,
|
|
max_id: str | None = None,
|
|
since_id: str | None = None,
|
|
min_id: str | None = None,
|
|
limit: int = 20,
|
|
offset: int = 0,
|
|
):
|
|
if limit > 40:
|
|
limit = 40
|
|
result: dict[str, list] = {"accounts": [], "statuses": [], "hashtags": []}
|
|
# We don't support pagination for searches yet
|
|
if max_id or since_id or min_id or offset:
|
|
return result
|
|
# Run search
|
|
searcher = Searcher(q, request.identity)
|
|
search_result = searcher.search_all()
|
|
if type is None or type == "accounts":
|
|
result["accounts"] = [i.to_mastodon_json() for i in search_result["identities"]]
|
|
if type is None or type == "hashtag":
|
|
result["hashtag"] = [h.to_mastodon_json() for h in search_result["hashtags"]]
|
|
if type is None or type == "statuses":
|
|
interactions = PostInteraction.get_post_interactions(
|
|
search_result["posts"], request.identity
|
|
)
|
|
result["statuses"] = [
|
|
p.to_mastodon_json(interactions=interactions)
|
|
for p in search_result["posts"]
|
|
]
|
|
return result
|