Posted at

とりあえずLINEに通知が欲しいから、適当なスクリプトを作った

More than 1 year has passed since last update.


動機


  • 社内のGPUサーバーでAIの学習をしている際、学習が終了するまでとても時間がかかる


    • 気づいたら終わってる、みたいなことが多い


      • 急ぎの学習の場合などはすぐに後続の作業にうつりたい





ということで、学習が終わったら何らかの形で通知が欲しいなぁと思いました。

で、LINEに通知が来ればまぁどこにいても受信できそうなので、

簡単にLINEで通知が来るスクリプトをPythonで作成しました。


コード

とりあえずで作ったので超適当です



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"