動機
- 社内のGPUサーバーでAIの学習をしている際、学習が終了するまでとても時間がかかる
- 気づいたら終わってる、みたいなことが多い
- 急ぎの学習の場合などはすぐに後続の作業にうつりたい
- 気づいたら終わってる、みたいなことが多い
ということで、学習が終わったら何らかの形で通知が欲しいなぁと思いました。
で、LINEに通知が来ればまぁどこにいても受信できそうなので、
簡単にLINEで通知が来るスクリプトをPythonで作成しました。
コード
- 前提
- LINE Notifyのページのマイページから、アクセストークンが発行できていること
とりあえずで作ったので超適当です
Python2.7環境で動作確認しました。
line_notification.py
import subprocess
import argparse
if __name__ == '__main__':
parser = argparse.ArgumentParser(
prog='line_notification.py',
usage='Exec on terminal.',
add_help=True,
)
parser.add_argument('-t', '--token', help='LINE ACCESS TOKEN', required=True, type=str)
parser.add_argument('-m', '--message', help='message data', required=True, type=str)
args = parser.parse_args()
# LINE notification
command = 'curl -X POST -H \'Authorization: Bearer {TOKEN}\' -F \"message={MESSAGE}\" https://notify-api.line.me/api/notify'.format(
TOKEN=args.token,
MESSAGE=args.message
)
subprocess.call(command, shell=True)
使い方
python line_notification.py -t "ACCESS_TOKEN" -m "MESSAGE"