コード例 #1
0
ファイル: provider.py プロジェクト: zuckerruebe/Pyckson
 def get(self, obj_type, parent_class, name_in_parent) -> Parser:
     if obj_type in BASIC_TYPES:
         return BasicParserWithCast(obj_type)
     if obj_type in EXTRA_TYPES:
         return BasicParser()
     if obj_type in DATE_TYPES:
         return DateParser(parent_class, obj_type)
     if obj_type is Decimal:
         return DecimalParser()
     if type(obj_type) is str or type(obj_type) is ForwardRef:
         return UnresolvedParser(TypeProvider(parent_class, obj_type),
                                 self.model_provider)
     if obj_type is list:
         type_info = get_cls_attr(parent_class, PYCKSON_TYPEINFO, dict())
         if name_in_parent in type_info:
             sub_type = type_info[name_in_parent]
             return ListParser(
                 self.get(sub_type, parent_class, name_in_parent))
         else:
             raise TypeError(
                 'list parameter {} in class {} has no subType'.format(
                     name_in_parent, parent_class.__name__))
     if obj_type is set:
         type_info = get_cls_attr(parent_class, PYCKSON_TYPEINFO, dict())
         if name_in_parent in type_info:
             sub_type = type_info[name_in_parent]
             return SetParser(
                 self.get(sub_type, parent_class, name_in_parent))
         else:
             raise TypeError(
                 'set parameter {} in class {} has no subType'.format(
                     name_in_parent, parent_class.__name__))
     if is_list_annotation(obj_type):
         return ListParser(
             self.get(obj_type.__args__[0], parent_class, name_in_parent))
     if is_set_annotation(obj_type):
         return SetParser(
             self.get(obj_type.__args__[0], parent_class, name_in_parent))
     if is_enum_annotation(obj_type):
         options = get_cls_attr(obj_type, PYCKSON_ENUM_OPTIONS, {})
         if options.get(ENUM_CASE_INSENSITIVE, False):
             return CaseInsensitiveEnumParser(obj_type)
         else:
             return DefaultEnumParser(obj_type)
     if is_basic_dict_annotation(obj_type):
         return BasicDictParser()
     if is_typing_dict_annotation(obj_type):
         if obj_type.__args__[0] != str:
             raise TypeError(
                 'typing.Dict key can only be str in class {}'.format(
                     parent_class))
         return TypingDictParser(
             self.get(obj_type.__args__[1], parent_class, name_in_parent))
     if has_cls_attr(obj_type, PYCKSON_PARSER):
         return CustomDeferredParser(obj_type)
     return ClassParser(obj_type, self.model_provider)
コード例 #2
0
ファイル: test_enum.py プロジェクト: zuckerruebe/Pyckson
class DefaultEnumParserTest(TestCase):
    def setUp(self):
        self.parser = DefaultEnumParser(MyEnum)

    def test_should_parse_value_in_enum(self):
        self.assertEqual(self.parser.parse('a'), MyEnum.a)
        self.assertEqual(self.parser.parse('A'), MyEnum.A)
        self.assertEqual(self.parser.parse('b'), MyEnum.b)

    def test_should_not_parse_uppercase_not_in_enum(self):
        with self.assertRaises(KeyError):
            self.parser.parse('B')

    def test_should_not_parse_value_not_in_enum(self):
        with self.assertRaises(KeyError):
            self.parser.parse('c')
コード例 #3
0
ファイル: test_enum.py プロジェクト: antidot/Pyckson
class DefaultEnumParserTest(TestCase):
    def setUp(self):
        self.parser = DefaultEnumParser(MyEnum)

    def test_should_parse_value_in_enum(self):
        self.assertEqual(self.parser.parse("a"), MyEnum.a)
        self.assertEqual(self.parser.parse("A"), MyEnum.A)
        self.assertEqual(self.parser.parse("b"), MyEnum.b)

    def test_should_not_parse_uppercase_not_in_enum(self):
        with self.assertRaises(KeyError):
            self.parser.parse("B")

    def test_should_not_parse_value_not_in_enum(self):
        with self.assertRaises(KeyError):
            self.parser.parse("c")
コード例 #4
0
ファイル: test_enum.py プロジェクト: zuckerruebe/Pyckson
 def setUp(self):
     self.parser = DefaultEnumParser(MyEnum)
コード例 #5
0
ファイル: test_enum.py プロジェクト: antidot/Pyckson
 def setUp(self):
     self.parser = DefaultEnumParser(MyEnum)