mirror of
https://github.com/zulip/zulip.git
synced 2026-07-03 21:10:12 +08:00
2080ff6c2a
response.content is binary data, but code usually assumes it to be text. Fix this by decoding response.content where required. Don't do this in tests yet.
58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
from __future__ import absolute_import
|
|
|
|
from django.http import HttpResponse, HttpResponseNotAllowed
|
|
import ujson
|
|
|
|
from typing import Optional, Any, Dict, List
|
|
from six import text_type
|
|
from zerver.lib.str_utils import force_bytes
|
|
|
|
|
|
class HttpResponseUnauthorized(HttpResponse):
|
|
status_code = 401
|
|
|
|
def __init__(self, realm, www_authenticate=None):
|
|
# type (text_type, Optional[text_type]) -> None
|
|
HttpResponse.__init__(self)
|
|
if www_authenticate is None:
|
|
self["WWW-Authenticate"] = 'Basic realm="%s"' % (realm,)
|
|
elif www_authenticate == "session":
|
|
self["WWW-Authenticate"] = 'Session realm="%s"' % (realm,)
|
|
else:
|
|
raise Exception("Invalid www_authenticate value!")
|
|
|
|
def json_unauthorized(message, www_authenticate=None):
|
|
# type: (text_type, Optional[text_type]) -> HttpResponse
|
|
resp = HttpResponseUnauthorized("zulip", www_authenticate=www_authenticate)
|
|
resp.content = force_bytes(ujson.dumps({"result": "error",
|
|
"msg": message}) + "\n")
|
|
return resp
|
|
|
|
def json_method_not_allowed(methods):
|
|
# type: (List[text_type]) -> text_type
|
|
resp = HttpResponseNotAllowed(methods)
|
|
resp.content = force_bytes(ujson.dumps({"result": "error",
|
|
"msg": "Method Not Allowed",
|
|
"allowed_methods": methods}))
|
|
return resp
|
|
|
|
def json_response(res_type="success", msg="", data=None, status=200):
|
|
# type: (text_type, text_type, Optional[Dict[str, Any]], int) -> HttpResponse
|
|
content = {"result": res_type, "msg": msg}
|
|
if data is not None:
|
|
content.update(data)
|
|
return HttpResponse(content=ujson.dumps(content) + "\n",
|
|
content_type='application/json', status=status)
|
|
|
|
def json_success(data=None):
|
|
# type: (Optional[Dict[str, Any]]) -> HttpResponse
|
|
return json_response(data=data)
|
|
|
|
def json_error(msg, data=None, status=400):
|
|
# type: (str, Optional[Dict[str, Any]], int) -> HttpResponse
|
|
return json_response(res_type="error", msg=msg, data=data, status=status)
|
|
|
|
def json_unhandled_exception():
|
|
# type: () -> HttpResponse
|
|
return json_response(res_type="error", msg="Internal server error", status=500)
|