#comprehension #python #macro #syntax

set_derive

Using Macros to Implement List comprehension Similar to Python Language

7 releases (2 stable)

1.1.0 Jan 9, 2024
1.0.0 Jan 9, 2024
0.1.4 Jun 19, 2023
0.1.3 Mar 26, 2023

#582 in Rust patterns

MIT license

20KB
472 lines

Using macro to implement the comprehensions of imitating python language in rust

  • You can initialize common collection types with relatively little code
  • Vec [v], HashMap {k=>v} and HashSet !{k} have been implemented

Vec Or HashSet Syntax

  • Ordinary element: [foo, 1, 2i32]
  • for each: [*(i; for i in 0..10)]
  • for each and filter: [*(i; for i in 0..10; if i != 5)]
  • for each and filter and sub for each: [*(i + j; for i in 0..10; if i != 5; for j in 1..i)]

HashMap Syntax

  • Ordinary element: {k => v, [k1]: v1}
  • String ordinary key: {%[key]: "jack"} (key: String)
  • Ident key: {name: "jack"} (name: &'static str)
  • String ident key: {%name: "jack"} (name: String)
  • Variable key: {*key: "jack"} (let key = "name";)
  • Literal key: {"name": "jack"}
  • Literal key tostring: {%"name": "jack"} (name: String)
  • for each: Just like in Vec, but slightly different. e.g {*(i => i+1; for i in 0..5)}

Other

  • for each reserve: [*[10](i; for i in 0..10)]
  • generic params: <i64>[1, 2, 3]
  • init with capacity: (20)[1, 2, 3] | <i64>(20)[1, 2, 3]
  • inspect block: [*(i; for i in 0..10; {dbg!(i);} if i & 1 == 0)]

crate: https://crates.io/crates/set_derive

Examples

# use set_derive::set_derive;
# use std::collections::HashMap;
let arr1 = vec![
    0, 1, 8, 9, 10, 15, 16, 17, 18, 19,
];
let arr2 = set_derive!([
    0, 1,
    *(i; for i in 8..=10),
    *(i; for i in 15..20),
]);
// python: [0, 1, *range(8, 10+1), *range(15, 20)]
assert_eq!(arr1, arr2);

let arr = set_derive!([*(
    i+j;
    for i in 0..6;
    if i%2!=0;
    for j in 0..i;
)]);
// python: [i+j for i in range(6) if i%2 for j in range(0, i)]
assert_eq!(arr, vec![1, 3, 4, 5, 5, 6, 7, 8, 9]);

let age_key = "age";
let jack = set_derive!({
    name: "jack",
    *age_key: "20",
});
assert_eq!(jack, HashMap::from([("name", "jack"), ("age", "20")]));

No runtime deps