I use notmuch for my email workflow. It has Python APIs. This is trivial to do with it. The downside is that it's for PCs, not phones. I suppose one could hack a mail server around its tags so you could use it via the phone as well. Never thought about it...

https://notmuchmail.org/