I jfinally managed to actually get to test the python script I mentioned in my question. It does work when asking to shut down as well.

I finally managed to actually test the Python script I mentioned as the second option in my question. It turns out that it does work when asking to shut down as well, not just on reboot.