1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
# Copyright 2017-2020 Palantir Technologies, Inc.
# Copyright 2021- Python Language Server Contributors.
import traceback
class JsonRpcException(Exception):
def __init__(self, message=None, code=None, data=None):
super().__init__(message)
self.message = message or getattr(self.__class__, 'MESSAGE')
self.code = code or getattr(self.__class__, 'CODE')
self.data = data
def to_dict(self):
exception_dict = {
'code': self.code,
'message': self.message,
}
if self.data is not None:
exception_dict['data'] = self.data
return exception_dict
def __eq__(self, other):
return (
isinstance(other, self.__class__) and
self.code == other.code and
self.message == other.message
)
def __hash__(self):
return hash((self.code, self.message))
@staticmethod
def from_dict(error):
for exc_class in _EXCEPTIONS:
if exc_class.supports_code(error['code']):
return exc_class(**error)
return JsonRpcException(**error)
@classmethod
def supports_code(cls, code):
# Defaults to UnknownErrorCode
return getattr(cls, 'CODE', -32001) == code
class JsonRpcParseError(JsonRpcException):
CODE = -32700
MESSAGE = 'Parse Error'
class JsonRpcInvalidRequest(JsonRpcException):
CODE = -32600
MESSAGE = 'Invalid Request'
class JsonRpcMethodNotFound(JsonRpcException):
CODE = -32601
MESSAGE = 'Method Not Found'
@classmethod
def of(cls, method):
return cls(message=cls.MESSAGE + ': ' + method)
class JsonRpcInvalidParams(JsonRpcException):
CODE = -32602
MESSAGE = 'Invalid Params'
class JsonRpcInternalError(JsonRpcException):
CODE = -32602
MESSAGE = 'Internal Error'
@classmethod
def of(cls, exc_info):
exc_type, exc_value, exc_tb = exc_info
return cls(
message=''.join(traceback.format_exception_only(exc_type, exc_value)).strip(),
data={'traceback': traceback.format_tb(exc_tb)}
)
class JsonRpcRequestCancelled(JsonRpcException):
CODE = -32800
MESSAGE = 'Request Cancelled'
class JsonRpcServerError(JsonRpcException):
def __init__(self, message, code, data=None):
assert _is_server_error_code(code)
super().__init__(message=message, code=code, data=data)
@classmethod
def supports_code(cls, code):
return _is_server_error_code(code)
def _is_server_error_code(code):
return -32099 <= code <= -32000
_EXCEPTIONS = (
JsonRpcParseError,
JsonRpcInvalidRequest,
JsonRpcMethodNotFound,
JsonRpcInvalidParams,
JsonRpcInternalError,
JsonRpcRequestCancelled,
JsonRpcServerError,
)
|
