You can use the parted module for partitioning

docs.ansible.com/ansible/…/parted_module.html#ans…

Format with BTRFS or many others

docs.ansible.com/…/filesystem_module.html#ansible…

These might come in handy:

• docs.ansible.com/…/btrfs_info_module.html#ansible…
• docs.ansible.com/…/btrfs_subvolume_module.html#an…

For ZFS, I’m not familiar but I found

• docs.ansible.com/…/zpool_facts_module.html#ansibl…

• docs.ansible.com/ansible/…/zfs_module.html#ansibl…

• docs.ansible.com/ansible/…/zfs_facts_module.html#…

Here’s the index of all modules: docs.ansible.com/ansible/…/index_module.html