188
189
190
191
192
193
194
195
196

def init(
    self,
    content: typing.Any,
    statuscode: int = 200,
    headers: typing.Optional[typing.Dict[str, str]] = None,
    mediatype: typing.Optional[str] = None,
    background: typing.Optional[BackgroundTask] = None,
) -> None:
    super().__init(content, status_code, headers, media_type, background)

31
32
33

def render(self, content: Any) -> bytes:
    assert ujson is not None, “ujson must be installed to use UJSONResponse”
    return ujson.dumps(content, ensure_ascii=False).encode(“utf-8”)

65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

def init_headers(
    self, headers: typing.Optional[typing.Mapping[str, str]] = None
) -> None:
    if headers is None:
        raw_headers: typing.List[typing.Tuple[bytes, bytes]] = []
        populate_content_length = True
        populate_content_type = True
    else:
        raw_headers = [
            (k.lower().encode(“latin-1”), v.encode(“latin-1”))
            for k, v in headers.items()
        ]
        keys = [h[0] for h in raw_headers]
        populate_content_length = b“content-length” not in keys
        populate_content_type = b“content-type” not in keys
    body = getattr(self, “body”, None)
    if (
        body is not None
        and populate_content_length
        and not (self.status_code < 200 or self.status_code in (204, 304))
    ):
        content_length = str(len(body))
        raw_headers.append((b“content-length”, content_length.encode(“latin-1”)))
    content_type = self.media_type
    if content_type is not None and populate_content_type:
        if content_type.startswith(“text/“):
            content_type += “; charset=” + self.charset
        raw_headers.append((b“content-type”, content_type.encode(“latin-1”)))
    self.raw_headers = raw_headers

104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141

def set_cookie(
    self,
    key: str,
    value: str = “”,
    max_age: typing.Optional[int] = None,
    expires: typing.Optional[typing.Union[datetime, str, int]] = None,
    path: str = “/“,
    domain: typing.Optional[str] = None,
    secure: bool = False,
    httponly: bool = False,
    samesite: typing.Optional[Literal[“lax”, “strict”, “none”]] = “lax”,
) -> None:
    cookie: “http.cookies.BaseCookie[str]” = http.cookies.SimpleCookie()
    cookie[key] = value
    if max_age is not None:
        cookie[key][“max-age”] = max_age
    if expires is not None:
        if isinstance(expires, datetime):
            cookie[key][“expires”] = format_datetime(expires, usegmt=True)
        else:
            cookie[key][“expires”] = expires
    if path is not None:
        cookie[key][“path”] = path
    if domain is not None:
        cookie[key][“domain”] = domain
    if secure:
        cookie[key][“secure”] = True
    if httponly:
        cookie[key][“httponly”] = True
    if samesite is not None:
        assert samesite.lower() in [
            “strict”,
            “lax”,
            “none”,
        ], “samesite must be either ‘strict’, ‘lax’ or ‘none’”
        cookie[key][“samesite”] = samesite
    cookie_val = cookie.output(header=“”).strip()
    self.raw_headers.append((b“set-cookie”, cookie_val.encode(“latin-1”)))

143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161

def delete_cookie(
    self,
    key: str,
    path: str = “/“,
    domain: typing.Optional[str] = None,
    secure: bool = False,
    httponly: bool = False,
    samesite: typing.Optional[Literal[“lax”, “strict”, “none”]] = “lax”,
) -> None:
    self.set_cookie(
        key,
        max_age=0,
        expires=0,
        path=path,
        domain=domain,
        secure=secure,
        httponly=httponly,
        samesite=samesite,
    )

188
189
190
191
192
193
194
195
196

def init(
    self,
    content: typing.Any,
    statuscode: int = 200,
    headers: typing.Optional[typing.Dict[str, str]] = None,
    mediatype: typing.Optional[str] = None,
    background: typing.Optional[BackgroundTask] = None,
) -> None:
    super().__init(content, status_code, headers, media_type, background)

44
45
46
47
48

def render(self, content: Any) -> bytes:
    assert orjson is not None, “orjson must be installed to use ORJSONResponse”
    return orjson.dumps(
        content, option=orjson.OPT_NON_STR_KEYS | orjson.OPT_SERIALIZE_NUMPY
    )

65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

def init_headers(
    self, headers: typing.Optional[typing.Mapping[str, str]] = None
) -> None:
    if headers is None:
        raw_headers: typing.List[typing.Tuple[bytes, bytes]] = []
        populate_content_length = True
        populate_content_type = True
    else:
        raw_headers = [
            (k.lower().encode(“latin-1”), v.encode(“latin-1”))
            for k, v in headers.items()
        ]
        keys = [h[0] for h in raw_headers]
        populate_content_length = b“content-length” not in keys
        populate_content_type = b“content-type” not in keys
    body = getattr(self, “body”, None)
    if (
        body is not None
        and populate_content_length
        and not (self.status_code < 200 or self.status_code in (204, 304))
    ):
        content_length = str(len(body))
        raw_headers.append((b“content-length”, content_length.encode(“latin-1”)))
    content_type = self.media_type
    if content_type is not None and populate_content_type:
        if content_type.startswith(“text/“):
            content_type += “; charset=” + self.charset
        raw_headers.append((b“content-type”, content_type.encode(“latin-1”)))
    self.raw_headers = raw_headers

104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141

def set_cookie(
    self,
    key: str,
    value: str = “”,
    max_age: typing.Optional[int] = None,
    expires: typing.Optional[typing.Union[datetime, str, int]] = None,
    path: str = “/“,
    domain: typing.Optional[str] = None,
    secure: bool = False,
    httponly: bool = False,
    samesite: typing.Optional[Literal[“lax”, “strict”, “none”]] = “lax”,
) -> None:
    cookie: “http.cookies.BaseCookie[str]” = http.cookies.SimpleCookie()
    cookie[key] = value
    if max_age is not None:
        cookie[key][“max-age”] = max_age
    if expires is not None:
        if isinstance(expires, datetime):
            cookie[key][“expires”] = format_datetime(expires, usegmt=True)
        else:
            cookie[key][“expires”] = expires
    if path is not None:
        cookie[key][“path”] = path
    if domain is not None:
        cookie[key][“domain”] = domain
    if secure:
        cookie[key][“secure”] = True
    if httponly:
        cookie[key][“httponly”] = True
    if samesite is not None:
        assert samesite.lower() in [
            “strict”,
            “lax”,
            “none”,
        ], “samesite must be either ‘strict’, ‘lax’ or ‘none’”
        cookie[key][“samesite”] = samesite
    cookie_val = cookie.output(header=“”).strip()
    self.raw_headers.append((b“set-cookie”, cookie_val.encode(“latin-1”)))

143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161

def delete_cookie(
    self,
    key: str,
    path: str = “/“,
    domain: typing.Optional[str] = None,
    secure: bool = False,
    httponly: bool = False,
    samesite: typing.Optional[Literal[“lax”, “strict”, “none”]] = “lax”,
) -> None:
    self.set_cookie(
        key,
        max_age=0,
        expires=0,
        path=path,
        domain=domain,
        secure=secure,
        httponly=httponly,
        samesite=samesite,
    )

286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320

def init(
    self,
    path: typing.Union[str, “os.PathLike[str]”],
    status_code: int = 200,
    headers: typing.Optional[typing.Mapping[str, str]] = None,
    media_type: typing.Optional[str] = None,
    background: typing.Optional[BackgroundTask] = None,
    filename: typing.Optional[str] = None,
    stat_result: typing.Optional[os.stat_result] = None,
    method: typing.Optional[str] = None,
    content_disposition_type: str = “attachment”,
) -> None:
    self.path = path
    self.status_code = status_code
    self.filename = filename
    self.send_header_only = method is not None and method.upper() == “HEAD”
    if media_type is None:
        media_type = guess_type(filename or path)[0] or “text/plain”
    self.media_type = media_type
    self.background = background
    self.init_headers(headers)
    if self.filename is not None:
        content_disposition_filename = quote(self.filename)
        if content_disposition_filename != self.filename:
            content_disposition = “{}; filename*=utf-8’’{}“.format(
                content_disposition_type, content_disposition_filename
            )
        else:
            content_disposition = ‘{}; filename=”{}“‘.format(
                content_disposition_type, self.filename
            )
        self.headers.setdefault(“content-disposition”, content_disposition)
    self.stat_result = stat_result
    if stat_result is not None:
        self.set_stat_headers(stat_result)