refactor(py): improve python logic by mirroring rust logic

[mibli]

Rust logic is pretty feature complete and stable. It could be used to improve python logic to behave properly. This would require a major refactoring of python code, but for feature parity it might be worth doing.