Skip to content

migrate_enum_annotation

Source code in src/library_analyzer/processing/migration/annotations/_migrate_enum_annotation.py 
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
def migrate_enum_annotation(origin_annotation: EnumAnnotation, mapping: Mapping) -> list[AbstractAnnotation]:
    annotated_apiv1_element = get_annotated_api_element(origin_annotation, mapping.get_apiv1_elements())
    if annotated_apiv1_element is None or not isinstance(annotated_apiv1_element, Parameter):
        return []

    migrated_annotations: list[AbstractAnnotation] = []
    for parameter in mapping.get_apiv2_elements():
        enum_annotation = deepcopy(origin_annotation)
        authors = enum_annotation.authors
        authors.append(migration_author)
        enum_annotation.authors = authors
        if isinstance(parameter, Attribute | Result):
            return []
        if isinstance(parameter, Parameter):
            if (
                parameter.type is not None
                and _contains_string(parameter.type)
                and _default_value_is_in_instance_values_or_is_empty(parameter.default_value, enum_annotation.pairs)
            ) or (parameter.type is None and annotated_apiv1_element.type is None):
                enum_annotation.target = parameter.id
                migrated_annotations.append(enum_annotation)
                continue
            if (
                isinstance(parameter.type, NamedType)
                and not _contains_string(parameter.type)
                and not (
                    isinstance(annotated_apiv1_element.type, NamedType)
                    and parameter.type.name == annotated_apiv1_element
                )
            ):
                # assuming api has been chanced to an enum type:
                # do not migrate annotation
                continue
            enum_annotation.reviewResult = EnumReviewResult.UNSURE
            enum_annotation.comment = get_migration_text(enum_annotation, mapping)
            enum_annotation.target = parameter.id
            migrated_annotations.append(enum_annotation)
            continue
        migrated_annotations.append(
            TodoAnnotation(
                parameter.id,
                authors,
                enum_annotation.reviewers,
                enum_annotation.comment,
                EnumReviewResult.NONE,
                get_migration_text(enum_annotation, mapping, for_todo_annotation=True),
            ),
        )
    return migrated_annotations