andrii strange-dv

yea

strange-dv / comp_multi.rs

Created January 12, 2025 17:52 — forked from kepler-5/comp_multi.rs

Python comprehension proc macro that handles multiple nested for-if-clauses, flattening nested structure

	// Python comprehension proc macro that handles multiple nested for-if-clauses, flattening nested structure.
	// Example:
	//
	// let vec_of_vecs = vec![vec![1, 2, 3], vec![4, 5, 6]];
	//
	// let result = comp![x for vec in vec_of_vecs for x in vec].collect::<Vec<_>>();
	// assert_eq!(result, [1, 2, 3, 4, 5, 6]);
	//

	use proc_macro2::TokenStream as TokenStream2;