Coverage for /pythoncovmergedfiles/medio/medio/src/aiohttp/aiohttp/http_exceptions.py: 92%

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

2 

3 

4from typing import Optional, Union 

5 

6from .typedefs import _CIMultiDict 

7 

8__all__ = ("HttpProcessingError",) 

9 

10 

11class HttpProcessingError(Exception): 

12 """HTTP error. 

13 

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

15 

16 code: HTTP Error code. 

17 message: (optional) Error message. 

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

19 """ 

20 

21 code = 0 

22 message = "" 

23 headers = None 

24 

25 def __init__( 

26 self, 

27 *, 

28 code: Optional[int] = None, 

29 message: str = "", 

30 headers: Optional[_CIMultiDict] = None, 

31 ) -> None: 

32 if code is not None: 

33 self.code = code 

34 self.headers = headers 

35 self.message = message 

36 

37 def __str__(self) -> str: 

38 return f"{self.code}, message={self.message!r}" 

39 

40 def __repr__(self) -> str: 

41 return f"<{self.__class__.__name__}: {self}>" 

42 

43 

44class BadHttpMessage(HttpProcessingError): 

45 code = 400 

46 message = "Bad Request" 

47 

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

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

50 self.args = (message,) 

51 

52 

53class HttpBadRequest(BadHttpMessage): 

54 code = 400 

55 message = "Bad Request" 

56 

57 

58class PayloadEncodingError(BadHttpMessage): 

59 """Base class for payload errors""" 

60 

61 

62class ContentEncodingError(PayloadEncodingError): 

63 """Content encoding error.""" 

64 

65 

66class TransferEncodingError(PayloadEncodingError): 

67 """transfer encoding error.""" 

68 

69 

70class ContentLengthError(PayloadEncodingError): 

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

72 

73 

74class LineTooLong(BadHttpMessage): 

75 def __init__( 

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

77 ) -> None: 

78 super().__init__( 

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

80 ) 

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

82 

83 

84class InvalidHeader(BadHttpMessage): 

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

86 if isinstance(hdr, bytes): 

87 hdr = hdr.decode("utf-8", "surrogateescape") 

88 super().__init__(f"Invalid HTTP Header: {hdr}") 

89 self.hdr = hdr 

90 self.args = (hdr,) 

91 

92 

93class BadStatusLine(BadHttpMessage): 

94 def __init__(self, line: str = "") -> None: 

95 if not isinstance(line, str): 

96 line = repr(line) 

97 super().__init__(f"Bad status line {line!r}") 

98 self.args = (line,) 

99 self.line = line 

100 

101 

102class InvalidURLError(BadHttpMessage): 

103 pass