forked from M-Labs/nac3
59 lines
1.9 KiB
Rust
59 lines
1.9 KiB
Rust
|
use rustpython_parser::ast::{Comparison, Operator, UnaryOperator};
|
||
|
|
|
||
|
|
pub fn binop_name(op: &Operator) -> &'static str {
|
||
|
|
match op {
|
||
|
|
Operator::Add => "__add__",
|
||
|
|
Operator::Sub => "__sub__",
|
||
|
|
Operator::Div => "__truediv__",
|
||
|
|
Operator::Mod => "__mod__",
|
||
|
|
Operator::Mult => "__mul__",
|
||
|
|
Operator::Pow => "__pow__",
|
||
|
|
Operator::BitOr => "__or__",
|
||
|
|
Operator::BitXor => "__xor__",
|
||
|
|
Operator::BitAnd => "__and__",
|
||
|
|
Operator::LShift => "__lshift__",
|
||
|
|
Operator::RShift => "__rshift__",
|
||
|
|
Operator::FloorDiv => "__floordiv__",
|
||
|
|
Operator::MatMult => "__matmul__",
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn binop_assign_name(op: &Operator) -> &'static str {
|
||
|
|
match op {
|
||
|
|
Operator::Add => "__iadd__",
|
||
|
|
Operator::Sub => "__isub__",
|
||
|
|
Operator::Div => "__itruediv__",
|
||
|
|
Operator::Mod => "__imod__",
|
||
|
|
Operator::Mult => "__imul__",
|
||
|
|
Operator::Pow => "__ipow__",
|
||
|
|
Operator::BitOr => "__ior__",
|
||
|
|
Operator::BitXor => "__ixor__",
|
||
|
|
Operator::BitAnd => "__iand__",
|
||
|
|
Operator::LShift => "__ilshift__",
|
||
|
|
Operator::RShift => "__irshift__",
|
||
|
|
Operator::FloorDiv => "__ifloordiv__",
|
||
|
|
Operator::MatMult => "__imatmul__",
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn unaryop_name(op: &UnaryOperator) -> &'static str {
|
||
|
|
match op {
|
||
|
|
UnaryOperator::Pos => "__pos__",
|
||
|
|
UnaryOperator::Neg => "__neg__",
|
||
|
|
UnaryOperator::Not => "__not__",
|
||
|
|
UnaryOperator::Inv => "__inv__",
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn comparison_name(op: &Comparison) -> Option<&'static str> {
|
||
|
|
match op {
|
||
|
|
Comparison::Less => Some("__lt__"),
|
||
|
|
Comparison::LessOrEqual => Some("__le__"),
|
||
|
|
Comparison::Greater => Some("__gt__"),
|
||
|
|
Comparison::GreaterOrEqual => Some("__ge__"),
|
||
|
|
Comparison::Equal => Some("__eq__"),
|
||
|
|
Comparison::NotEqual => Some("__ne__"),
|
||
|
|
_ => None,
|
||
|
|
}
|
||
|
|
}
|