Esempio n. 1
0
    def test_get_class_metadata(self):
        class NoDoc:
            pass
        self.assertEqual(get_class_metadata(NoDoc), (None, None))

        class EmptyDoc1:
            ""
        self.assertEqual(get_class_metadata(EmptyDoc1), (None, None))

        class EmptyDoc2:
            """
            """
        self.assertEqual(get_class_metadata(EmptyDoc2), (None, None))

        class EmptyBoth:
            """

            """
        self.assertEqual(get_class_metadata(EmptyBoth), (None, None))

        class TitleOnly1:
            """title only"""
        self.assertEqual(get_class_metadata(TitleOnly1), ("title only", None))

        class TitleOnly2:
            """title only
            """
        self.assertEqual(get_class_metadata(TitleOnly2), ("title only", None))

        class TitleOnly3:
            """
            title only
            """
        self.assertEqual(get_class_metadata(TitleOnly3), ("title only", None))

        class AllOfThem1:
            """title

            description"""
        self.assertEqual(get_class_metadata(AllOfThem1), ("title", "description"))

        class AllOfThem2:
            """
            title

            description"""
        self.assertEqual(get_class_metadata(AllOfThem2), ("title", "description"))

        class AllOfThem3:
            """
            title

            description
            """
        self.assertEqual(get_class_metadata(AllOfThem3), ("title", "description"))
Esempio n. 2
0
    def _generate_json_schema(self):
        json_schema = {
            'type': 'object',
            'properties': {},
            'required': [],
            'additionalProperties': False
        }

        title, desc = get_class_metadata(self)

        if title:
            json_schema['title'] = title

        if desc:
            json_schema['description'] = desc

        for name, prop in self._untt_properties.items():
            json_schema['properties'][name] = prop.json_schema

            if prop.default is Ellipsis:
                json_schema['required'].append(name)

        if not json_schema['required']:
            del json_schema['required']

        return json_schema
Esempio n. 3
0
File: types.py Progetto: n9code/untt
    def _define_root_schema(self, ptype):
        class_title, class_desc = get_class_metadata(ptype)
        self._untt_root_def = {
            'type': 'string',
            'enum': [e.name for e in ptype],
        }

        if class_title:
            self._untt_root_def['title'] = class_title

        if class_desc:
            self._untt_root_def['description'] = class_desc

        add_root_definition(ptype.__name__, self._untt_root_def)