1
1
""" Contains definition of configuration class for PyOData"""
2
2
3
3
from typing import Type , Dict
4
- from pyodata .policies import PolicyFatal , ParserError , ErrorPolicy
4
+ from pyodata .policies import PolicyFatal , ParserError , ErrorPolicyType
5
5
import pyodata .version
6
6
7
7
@@ -14,8 +14,8 @@ class Config:
14
14
15
15
def __init__ (self ,
16
16
odata_version : Type [pyodata .version .ODATAVersion ],
17
- custom_error_policies = None ,
18
- default_error_policy = None ,
17
+ custom_error_policies : Dict [ ParserError , ErrorPolicyType ] = None ,
18
+ default_error_policy : ErrorPolicyType = None ,
19
19
xml_namespaces = None
20
20
):
21
21
@@ -48,19 +48,19 @@ def __init__(self,
48
48
self ._annotation_namespaces = None
49
49
self ._aliases : Dict [str , str ] = dict ()
50
50
51
- def err_policy (self , error : ParserError ) -> ErrorPolicy :
51
+ def err_policy (self , error : ParserError ) -> ErrorPolicyType :
52
52
""" Returns error policy for given error. If custom error policy fo error is set, then returns that."""
53
53
if self ._custom_error_policy is None :
54
54
return self ._default_error_policy
55
55
56
56
return self ._custom_error_policy .get (error , self ._default_error_policy )
57
57
58
- def set_default_error_policy (self , policy : ErrorPolicy ):
58
+ def set_default_error_policy (self , policy : ErrorPolicyType ):
59
59
""" Sets default error policy as well as resets custom error policies"""
60
60
self ._custom_error_policy = None
61
61
self ._default_error_policy = policy
62
62
63
- def set_custom_error_policy (self , policies : Dict [ParserError , Type [ ErrorPolicy ] ]):
63
+ def set_custom_error_policy (self , policies : Dict [ParserError , ErrorPolicyType ]):
64
64
""" Sets custom error policy. It should be called only after setting default error policy, otherwise
65
65
it has no effect. See implementation of "set_default_error_policy" for more details.
66
66
"""
0 commit comments