#!/usr/bin/env python3
#coding:utf-8

import requests
import json
import sys
import os
import datetime

#钉钉机器人的webhook的值
#token为钉钉后台上一步获取
webhook = "https://oapi.dingtalk.com/robot/send?access_token=xxxxxx"

#日志文件
log_path = "/tmp/dingding.log"

#接收数据
user=sys.argv[1]
title=sys.argv[2]
text=sys.argv[3]

#构建数据
data={
  "msgtype": "text",
  "text": {
    "content": text
  },
  "at": {
    "atMobiles": [
      user
    ],
    "isAtAll": False
  }
}

#提交
headers = {'Content-Type': 'application/json'}
x=requests.post(url=webhook,data=json.dumps(data),headers=headers)

#创建日志文件
if os.path.exists(log_path):
  f=open(log_path,"a+")
else:
  f=open(log_path,"w+")

#写入分割线
f.write("\n" + "--"*30)

#写日志
if x.json()["errcode"] == 0:
  f.write("\n" + str(datetime.datetime.now()) + "  " + str(user) + "  " + "发送成功" + "  " + str(title) + "  " + "\n" + str(text))
  f.close()
else:
  f.write("\n" + str(datetime.datetime.now()) + "  " + str(user) + "  " + "发送失败" + "  " + str(title) + "  " + "\n" + str(text))
  f.close()