I recently discovered zod for this kind of thing. It can be a little obtuse when you first pick it up, but it's incredibly cool and let's you do this kind of thing!