Windowsのシャットダウンによる強制終了でも適切に終了処理を行う方法

Question

pythonスクリプトが強制終了された場合でも、終了処理を適切に実施する方法は無いでしょうか？
try ~ finally節はキーボードインタラプトなどは対応できますが、コンソールウィンドウの×ボタンを
押したり、Windows自体をシャットダウン／再起動した場合は対応できないようです。

try:
    while True:
        time.sleep(1)
finally:
    # 終了したらファイルにメッセージを出力する
    with open("hoge.txt", 'w') as f:
        f.write("終了しました¥n")

上記のようなコードを実行した場合Ctrl-Cに対しては意図した通りに動作しますが、
コンソールウィンドウの×ボタンによる終了や、Windowsをシャットダウンした場合は
何も出力されません。

Windowsをシャットダウンしても適切に終了処理をする良い方法はないでしょうか？

Answer 1

コンソールアプリケーションであれば、SetConsoleCtrlHandler win32apiを利用して以下のようなコードを書く事ができると思います。

# -*- coding: utf-8 -*-
import win32api
import win32con
import time

def ctrl_handler(evt):
  if(evt in (win32con.CTRL_C_EVENT, 
             win32con.CTRL_BREAK_EVENT)):
    with open("hoge.txt", 'w') as f:
    	f.write("C-c,C-break ¥n")
  if(evt in (win32con.CTRL_CLOSE_EVENT,
             win32con.CTRL_LOGOFF_EVENT,
             win32con.CTRL_SHUTDOWN_EVENT)):
    with open("hoge.txt", 'w') as f:
      f.write("Close | Log off | Shutdown ¥n")
  
  # return True
  return False

if __name__ == '__main__':
  win32api.SetConsoleCtrlHandler(ctrl_handler)
  try:
    while True:
      time.sleep(1)
  finally:
    with open("hoge.txt", 'w') as f:
      f.write("Exit. ¥n")

Comments

[fukatataku]

回答ありがとうございます。
win32apiを使う方法が最も良さそうですね。
ただ、試してみたところ×ボタン押下に対しては終了処理をしてくれましたが、
Windowsのシャットダウン／再起動では何故かうまくいきませんでした。
もう少し試行錯誤をしてみようかと思います。