[Python Study Notes]进程信息(丁丁软件监控进程,http-post)

'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
>>文件: 进程信息.py
>>作者: liu yang
>>邮箱: liuyang0001@outlook.com

'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys,os
import psutil

# http get/post
import requests
# 处理字符串
import json
# 获取系统时间
import time

# 丁丁url
url = 'https://oapi.dingtalk.com/robot/send?access_token=3d94c740a27d5f70da6acf2a1c9383b338ad5c562fbaece8d38595c405e7e6a9'
# 转换成当前的时间戳
time_now = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
# 要监控的进程
monitor_name = {'ufw'}
# 要监控的进程,与shell语句对应,先定义为空
monitor_map={}
#要监控的服务名集合,{'ufw': 'service ufw start'}
for m in monitor_name:
    monitor_map[m]='service '+m+' start'

print(monitor_map)

# print(time_now)

# 显示进程,分别得到进程字典{'pid':'name'}和去重元祖()
def process_message():
    # 进程字典,保存pid,进程名的键值对
    proc_dict={}
    # 进程列表,保存去重的进程名称
    proc_list=set()
    # psutil.Process(pid=4, name='System', started='2018-02-11 16:39:45')
    for p in psutil.process_iter(attrs=['pid','name']):
        proc_dict[p.info['pid']]=p.info['name']
        proc_list.add(p.info['name'])
    print(proc_list)
    print(proc_dict)

    proc_stop = monitor_name - proc_list
    #proc_stop:提取出来哪一个进程没有在当前操作系统下开启
    if proc_stop:
        #当前没有启动的这个进程集合不为空，有服务死掉了
        for p in proc_stop:
            print("正在重新启动%s"%p)
            #取出每一个没有启动的服务名
            p_status = '停止'
            p_name = p
            data = {
                "msgtype": "markdown",
                "markdown": {
                "title":"监控讯息",
                "text": "#### %s
" % time_now +
                 "> ##### 服务名: %s 

" % p_name +
                 "> ##### 状态: %s 
" % p_status +
                 "> ##### 正在尝试启动"
                 }
            }
            headers = {'Content-Type':'application/json ; charset=UTF-8'}
            send_data = json.dumps(data).encode('utf-8')
            requests.post(url=url,data=send_data,headers=headers)
            #post提交数据:提交的地址，提交的数据

            if sys.platform.lower()=='linux':
                os.system(monitor_map[p])
            #函数不会给你返回值

            proc_name = set()

            for p2 in psutil.process_iter(attrs=['pid','name']):
                proc_name.add(p2.info['name'])
                #操作系统下的所有进程名拿出来

            if p in proc_name:
                print("%s启动成功"%p)
                p_status = '启动'
                data = {
                    "msgtype": "markdown",
                    "markdown": {
                    "title":"监控讯息",
                    "text": "#### %s
" % time_now +
                     "> ##### 服务名: %s 

" % p_name +
                     "> ##### 状态: %s 
" % p_status +
                     "> ##### 重启成功"
                     }
                }
                headers = {'Content-Type':'application/json ; charset=UTF-8'}
                send_data = json.dumps(data).encode('utf-8')
                requests.post(url=url,data=send_data,headers=headers)
            else:
                print("%s启动失败"%p)
                p_status = '停止'
                data = {
                    "msgtype": "markdown",
                    "markdown": {
                    "title":"监控讯息",
                    "text": "#### %s
" % time_now +
                     "> ##### 服务名: %s 

" % p_name +
                     "> ##### 状态: %s 
" % p_status +
                     "> ##### 启动失败"
                     }
                }
                headers = {'Content-Type':'application/json ; charset=UTF-8'}
                send_data = json.dumps(data).encode('utf-8')
                requests.post(url=url,data=send_data,headers=headers)
    time.sleep(2)

if __name__ == '__main__':

    while True:
        process_message()