Function collect_routes

plugins/ast/python/src/lib.rs:198–224

(body: &[Statement], routes: &mut Vec<Route>)

const ROUTE_VERBS: &[&str] = &["get", "post", "put", "patch", "delete", "options", "head"];

fn collect_routes(body: &[Statement], routes: &mut Vec<Route>) {
    for statement in body {
        match statement {
            Statement::ClassDef(class) => collect_routes(&class.body, routes),
            Statement::FunctionDef(function) => {
                for decorator in &function.decorator_list {
                    routes_from_decorator(&decorator.expression, routes);
                }
            }
            // Django URLConf: `urlpatterns = [path("p/", view), re_path(...), ...]`.
            Statement::Assign(Assignment { targets, value, .. }) => {
                let is_urlpatterns = targets.iter().any(|t| {
                    matches!(t, Expression::Name(NameExpression { id, .. }) if id.as_str() == "urlpatterns")
                });

                if is_urlpatterns {
                    if let Expression::List(list) = &**value {
                        for element in &list.elts {
                            routes_from_urlconf(element, routes);
                        }
                    }
                }
            }
            _ => {}
        }
    }
}

/// Extract a route from a Django URLConf entry: `path("p/", view)`, `re_path(r"^p$", view)`,
/// or legacy `url(...)`. Django doesn't bind a method at the URL layer, so we emit "ALL".