python調用釘釘機器人發送消息

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# example：python dingding.py 參數1 參數2 參數3
import requests
import json
import sys
import os
import time

headers = {'Content-Type': 'application/json'}
time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())

log_dir = '/usr/local/zabbix/logs/'
log_file = 'dingding.log'
api_url = '釘釘的Webhook地址'

def log(info):
    # 注意權限,否則寫不進去日誌
    file_path = log_dir + log_file
    if os.path.isdir(log_dir) == False:
        os.makedirs(log_dir)
    elif os.path.isfile(file_path) == False:
        f = open(file_path,'a+')
        f.write(info)
        f.close()
    else:
        f = open(file_path, 'ab+')
        f.write(info)
        f.close()

def msg(text, user):
    json_text= {"msgtype": "text","text": {"content": text},"at": {"atMobiles": [user],"isAtAll": False}}
    r=requests.post(api_url,data=json.dumps(json_text),headers=headers).json()
    code = r["errcode"]
    if code == 0:
        log(time + ":消息發送成功 返回碼:" + str(code) + "\n")
    else:
        log(time + ":消息發送失敗 返回碼:" + str(code) + "\n")
        exit(3)

if __name__ == '__main__':
    text = sys.argv[3]
    user = sys.argv[1]
    msg(text,user)

這個代碼有很多地方可以優化，請自行優化

python運維交流羣: 305357273