11""" Contains definition of configuration class for PyOData"""
22
33from typing import Type , Dict
4- from pyodata .policies import PolicyFatal , ParserError , ErrorPolicy
4+ from pyodata .policies import PolicyFatal , ParserError , ErrorPolicyType
55import pyodata .version
66
77
@@ -14,8 +14,8 @@ class Config:
1414
1515 def __init__ (self ,
1616 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 ,
1919 xml_namespaces = None
2020 ):
2121
@@ -48,19 +48,19 @@ def __init__(self,
4848 self ._annotation_namespaces = None
4949 self ._aliases : Dict [str , str ] = dict ()
5050
51- def err_policy (self , error : ParserError ) -> ErrorPolicy :
51+ def err_policy (self , error : ParserError ) -> ErrorPolicyType :
5252 """ Returns error policy for given error. If custom error policy fo error is set, then returns that."""
5353 if self ._custom_error_policy is None :
5454 return self ._default_error_policy
5555
5656 return self ._custom_error_policy .get (error , self ._default_error_policy )
5757
58- def set_default_error_policy (self , policy : ErrorPolicy ):
58+ def set_default_error_policy (self , policy : ErrorPolicyType ):
5959 """ Sets default error policy as well as resets custom error policies"""
6060 self ._custom_error_policy = None
6161 self ._default_error_policy = policy
6262
63- def set_custom_error_policy (self , policies : Dict [ParserError , Type [ ErrorPolicy ] ]):
63+ def set_custom_error_policy (self , policies : Dict [ParserError , ErrorPolicyType ]):
6464 """ Sets custom error policy. It should be called only after setting default error policy, otherwise
6565 it has no effect. See implementation of "set_default_error_policy" for more details.
6666 """
0 commit comments