Esempio n. 1
0
    def test_can_register_via_decorator_using_custom_name(self):
        registry = persisting_theory.Registry()

        @registry.register(name="custom_name")
        def something():
            pass

        self.assertEqual(registry.get('custom_name'), something)
Esempio n. 2
0
    def test_can_register_data_to_registry(self):

        data = "something"
        registry = persisting_theory.Registry()
        registry.register(data, name="key")

        self.assertEqual(len(registry), 1)
        self.assertEqual(registry.get("key"), data)
Esempio n. 3
0
    def test_can_register_class_and_function_via_decorator(self):
        registry = persisting_theory.Registry()

        @registry.register
        class ToRegister:
            pass

        self.assertEqual(registry.get('ToRegister'), ToRegister)

        @registry.register
        def something():
            pass

        self.assertEqual(registry.get('something'), something)
Esempio n. 4
0
    def test_can_infer_name_from_class_function_and_instance(self):
        registry = persisting_theory.Registry()

        def something():
            pass

        class MyClass:
            pass

        self.assertEqual(registry.get_object_name(something), "something")
        self.assertEqual(registry.get_object_name(MyClass), "MyClass")

        with self.assertRaises(ValueError):
            self.assertEqual(registry.get_object_name(MyClass()), "MyClass")
Esempio n. 5
0
    target = TargetSerializer()

    class Meta:
        model = models.UserFilter
        fields = ["uuid", "target", "creation_date"]
        read_only_fields = ["uuid", "creation_date"]

    def validate(self, data):
        target = data.pop("target")
        if target["type"] == "artist":
            data["target_artist"] = target["obj"]

        return data


state_serializers = persisting_theory.Registry()


class DescriptionStateMixin(object):
    def get_description(self, o):
        if o.description:
            return o.description.text


TAGS_FIELD = serializers.ListField(source="get_tags")


@state_serializers.register(name="music.Artist")
class ArtistStateSerializer(DescriptionStateMixin,
                            serializers.ModelSerializer):
    tags = TAGS_FIELD