def get_rules()

in plugins/spark_upgrade/update_calendar_interval.py [0:0]

• 24 lines of code
• 1 McCabe index (conditional complexity)

    def get_rules(self) -> List[Rule]:
        update_CalendarInterval = Rule(
            name="update_CalendarInterval",
            query="cs CalendarInterval.:[x]",
            replace_node="*",
            replace="DateTimeConstants.@x",
            holes={"calendarInterval"},
        )
        add_import_DateTimeConstants = Rule(
            name="add_import_DateTimeConstants",
            query="(package_clause) @package_clause",
            replace_node="package_clause",
            replace="@package_clause\nimport org.apache.spark.sql.catalyst.util.DateTimeConstants",
            filters={
                Filter(
                    enclosing_node="(compilation_unit) @cu",
                    not_contains=[
                        "cs import org.apache.spark.sql.catalyst.util.DateTimeConstants"
                    ],
                )
            },
            is_seed_rule=False,
        )
        return [update_CalendarInterval, add_import_DateTimeConstants]