Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/aiohttp/http_exceptions.py: 60%

1"""Low-level http related exceptions.""" 

2 

3 

4from textwrap import indent 

5from typing import Optional, Union 

6 

7from .typedefs import _CIMultiDict 

8 

9__all__ = ("HttpProcessingError",) 

10 

11 

12class HttpProcessingError(Exception): 

13 """HTTP error. 

14 

15 Shortcut for raising HTTP errors with custom code, message and headers. 

16 

17 code: HTTP Error code. 

18 message: (optional) Error message. 

19 headers: (optional) Headers to be sent in response, a list of pairs 

20 """ 

21 

22 code = 0 

23 message = "" 

24 headers = None 

25 

26 def __init__( 

27 self, 

28 *, 

29 code: Optional[int] = None, 

30 message: str = "", 

31 headers: Optional[_CIMultiDict] = None, 

32 ) -> None: 

33 if code is not None: 

34 self.code = code 

35 self.headers = headers 

36 self.message = message 

37 

38 def __str__(self) -> str: 

39 msg = indent(self.message, " ") 

40 return f"{self.code}, message:\n{msg}" 

41 

42 def __repr__(self) -> str: 

43 return f"<{self.__class__.__name__}: {self.code}, message={self.message!r}>" 

44 

45 

46class BadHttpMessage(HttpProcessingError): 

47 

48 code = 400 

49 message = "Bad Request" 

50 

51 def __init__(self, message: str, *, headers: Optional[_CIMultiDict] = None) -> None: 

52 super().__init__(message=message, headers=headers) 

53 self.args = (message,) 

54 

55 

56class HttpBadRequest(BadHttpMessage): 

57 

58 code = 400 

59 message = "Bad Request" 

60 

61 

62class PayloadEncodingError(BadHttpMessage): 

63 """Base class for payload errors""" 

64 

65 

66class ContentEncodingError(PayloadEncodingError): 

67 """Content encoding error.""" 

68 

69 

70class TransferEncodingError(PayloadEncodingError): 

71 """transfer encoding error.""" 

72 

73 

74class ContentLengthError(PayloadEncodingError): 

75 """Not enough data for satisfy content length header.""" 

76 

77 

78class LineTooLong(BadHttpMessage): 

79 def __init__( 

80 self, line: str, limit: str = "Unknown", actual_size: str = "Unknown" 

81 ) -> None: 

82 super().__init__( 

83 f"Got more than {limit} bytes ({actual_size}) when reading {line}." 

84 ) 

85 self.args = (line, limit, actual_size) 

86 

87 

88class InvalidHeader(BadHttpMessage): 

89 def __init__(self, hdr: Union[bytes, str]) -> None: 

90 hdr_s = hdr.decode(errors="backslashreplace") if isinstance(hdr, bytes) else hdr 

91 super().__init__(f"Invalid HTTP header: {hdr!r}") 

92 self.hdr = hdr_s 

93 self.args = (hdr,) 

94 

95 

96class BadStatusLine(BadHttpMessage): 

97 def __init__(self, line: str = "", error: Optional[str] = None) -> None: 

98 if not isinstance(line, str): 

99 line = repr(line) 

100 super().__init__(error or f"Bad status line {line!r}") 

101 self.args = (line,) 

102 self.line = line 

103 

104 

105class InvalidURLError(BadHttpMessage): 

106 pass