mirror of https://github.com/astral-sh/ruff
31 lines
747 B
Rust
31 lines
747 B
Rust
use rustpython_ast::{Expr, ExprKind};
|
|
|
|
use crate::ast::types::Range;
|
|
use crate::checkers::ast::Checker;
|
|
use crate::registry::Diagnostic;
|
|
use crate::violations;
|
|
|
|
/// B003
|
|
pub fn assignment_to_os_environ(checker: &mut Checker, targets: &[Expr]) {
|
|
if targets.len() != 1 {
|
|
return;
|
|
}
|
|
let target = &targets[0];
|
|
let ExprKind::Attribute { value, attr, .. } = &target.node else {
|
|
return;
|
|
};
|
|
if attr != "environ" {
|
|
return;
|
|
}
|
|
let ExprKind::Name { id, .. } = &value.node else {
|
|
return;
|
|
};
|
|
if id != "os" {
|
|
return;
|
|
}
|
|
checker.diagnostics.push(Diagnostic::new(
|
|
violations::AssignmentToOsEnviron,
|
|
Range::from_located(target),
|
|
));
|
|
}
|