Avoid hard unwrap in PT checks (#1597)

This commit is contained in:
Charlie Marsh 2023-01-03 07:39:52 -05:00 committed by GitHub
parent da5a25b421
commit 8d01efb571
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -193,13 +193,14 @@ pub fn parametrize(checker: &mut Checker, decorators: &[Expr]) {
if let Some(decorator) = decorator { if let Some(decorator) = decorator {
if let ExprKind::Call { args, .. } = &decorator.node { if let ExprKind::Call { args, .. } = &decorator.node {
if checker.settings.enabled.contains(&CheckCode::PT006) { if checker.settings.enabled.contains(&CheckCode::PT006) {
let first = args.first().unwrap(); if let Some(arg) = args.get(0) {
check_names(checker, first); check_names(checker, arg);
}
} }
if checker.settings.enabled.contains(&CheckCode::PT007) { if checker.settings.enabled.contains(&CheckCode::PT007) {
let second = args.get(1).unwrap(); if let Some(arg) = args.get(1) {
check_values(checker, second); check_values(checker, arg);
}
} }
} }
} }