laser_pc/baseFile/MonitorPid.py

import subprocess

import psutil,time
import pymysql
import redis
import logging
import win32serviceutil
import win32service
import win32event
import inspect
import os
import signal
import json
import win32timezone


host = '127.0.0.1' 
rport =16379  # redis服务端口
mport =3307   # mysql服务端口
class checkProces :
    def handle(r,conn,buildStatus,sceneStatus):
        flagid=r.exists('build:scene:status:id')
        logging.info("** flagid %s " ,flagid)
        if flagid :
            id =r.get('build:scene:status:id')
            id = int(id.replace('L', ''))
            logging.info("**build %s " ,id)
            r.delete('build:scene:status')
            r.delete('build:scene:status:id')
            conn.ping(reconnect=True)
            cur = conn.cursor()
            type =r.get('build:scene:type')
            logging.info("build_type: %s ",type)

            if type == '"buildObj"':
                sql1 = "UPDATE t_scene_file_build  set use_build_time=TIMESTAMPDIFF(SECOND,start_build_time,NOW())*1000 ,build_status={0} WHERE id= {1}".format(buildStatus,id)
                sql2 = "UPDATE t_scene set build_obj_status={0} WHERE scene_code = (SELECT scene_num FROM t_scene_file_build WHERE id = {1} )".format(buildStatus,id)
              
            else :
                export_mesh_obj = checkProces.get_export_mesh_obj_by_id(id)
                if export_mesh_obj is not None and export_mesh_obj==1 :
                    sql1 = "UPDATE t_scene_file_build  set use_build_time=TIMESTAMPDIFF(SECOND,start_build_time,NOW())*1000 ,build_status={0} WHERE id= {1}".format(buildStatus,id)
                    sql2 = "UPDATE t_scene set status={0} and build_obj_status={1} WHERE scene_code = (SELECT scene_num FROM t_scene_file_build WHERE id = {2} )".format(sceneStatus,buildStatus,id)
                else:
                    sql1 = "UPDATE t_scene_file_build  set use_build_time=TIMESTAMPDIFF(SECOND,start_build_time,NOW())*1000 ,build_status={0} WHERE id= {1}".format(buildStatus,id)
                    sql2 = "UPDATE t_scene set status={0} WHERE scene_code = (SELECT scene_num FROM t_scene_file_build WHERE id = {1} )".format(sceneStatus,id)
    
            res1 = cur.execute(sql1)
            res2 = cur.execute(sql2)
            logging.info(sql1)
            logging.info(sql2)

            logging.info(res1)
            logging.info(res2)
            conn.commit()
            conn.close()
            cur.close()
            logging.info('\n** checkLaserProcess stop  **')

            win32serviceutil.StopService("checkLaserProcess")


    def execute() :
        seetingjson = os.environ.get("FDMGEA_HOME") + "\\bin\\resources\\static\\setting.json"
        buildCallPath = json.load(open(seetingjson, 'r', encoding="utf-8"))['buildCallPath']
        logging.basicConfig(filename=buildCallPath + '\\monitor.log',
                            format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
                            level=logging.INFO,
                            filemode='a',
                            datefmt='%Y-%m-%d%I:%M:%S %p')
        r = redis.StrictRedis(host=host, port=rport, db=0, decode_responses=True)
        conn = pymysql.connect(host=host, port=mport, db="laser", user="root", password="laser", charset='utf8')

        print("\n")
        logging.info('\n** Search Running Process Using List Comprehension **')

        #  Pass the Process that needs to be checked & Check, If process run under this user.
        MainLoader = 'MainLoader.exe'
        laser = 'laser.exe'

        TIME = 2  # 程序状态检测间隔（单位：秒钟）

        file = open(buildCallPath + '\\monitor.log', 'w+').close()

        try:
            while 1:
                time.sleep(TIME)
                flag = r.exists('build:scene:status')
                stepflag = r.exists('build:scene:step')
                sceneCodeFlag=r.exists("build:scene:code")
                if flag and stepflag and sceneCodeFlag:
                    pid_counter = 0
                    MainLoaderFlag = 0
                    laserFlag = 0
                    JavaPortFlag = 0
                    V4JavaPortFlag = 0
                    sceneCode=r.get('build:scene:code')
                    pauseFlag=r.exists("build:scene:pause:code:"+sceneCode)
                    if pauseFlag:
                        logging.info("** pause continue...")
                        continue
                    step = r.get('build:scene:step')
                    step = int(step)
                    for procObj in psutil.process_iter(['pid', 'ppid', 'name', 'username']):
                        if MainLoader.lower() in procObj.name().lower():
                            MainLoaderFlag = procObj.pid
                        if laser.lower() in procObj.name().lower():
                            laserFlag = procObj.ppid()
                        if procObj.info['name'] == 'java.exe'  :
                            if "fdkk-v4" in str(procObj.cmdline()) :
                                V4JavaPortFlag= procObj.info['pid']
                            if "fdkk-laser" in str(procObj.cmdline()) :
                                JavaPortFlag= procObj.info['pid']

                    logging.info("** MainLoaderFlag = %s ,laserFlag= %s,JavaPortFlag=%s,v4JavaPort=%s", MainLoaderFlag, laserFlag, JavaPortFlag,V4JavaPortFlag)
                    if JavaPortFlag == 0:
                        logging.info("** only Process java  to Run...")
                        checkProces.closePid(MainLoaderFlag)
                        logging.info("** update...")
                        checkProces.handle(r, conn, -1, 1)
                    elif MainLoaderFlag == 0 and laserFlag == 0:
                        logging.info("** all Process MainLoaderFlag laser  to Run...")
                        checkProces.closePid(MainLoaderFlag)
                        if JavaPortFlag > 0:
                            checkProces.closePid(JavaPortFlag)
                            checkProces.closePid(V4JavaPortFlag)
                            logging.info("** update...")
                            checkProces.handle(r, conn, -1, 1)
                    # elif laserFlag > 1 and MainLoaderFlag == 0 and step ==1 :
                    #     logging.info("** only Process laser to Run...")
                    #     checkProces.handle(r,conn,-1,1)
                    elif laserFlag == 0 and MainLoaderFlag > 1:
                        logging.info("** only Process MainLoaderFlag  to Run...")
                        checkProces.closePid(MainLoaderFlag)
                        checkProces.closePid(JavaPortFlag)
                        checkProces.closePid(V4JavaPortFlag)
                        logging.info("** update...")
                        checkProces.handle(r, conn, -1, 1)
                else:
                    win32serviceutil.StopService("checkLaserProcess")
                    logging.info("**not build %s",flag)

        except KeyboardInterrupt as e:
            print("检测到CTRL+C，准备退出程序!")

    def closePid(PID):
        try:
            os.kill(PID, 9)
            logging.info("进程已关闭")
        except OSError:
            logging.info("进程不存在")
            pass

    def get_export_mesh_obj_by_id(id,cur):
      
    
        # Execute the query to select the export_mesh_obj column from t_scene based on the ID
        sql="select export_mesh_obj from t_scene_file_build where id={0}".format(id)
        cur.execute(sql)
        # Fetch the row from the result set
        row = cur.fetchone()

        if row is not None:
            export_mesh_obj = row[0]
            return export_mesh_obj
        else:
            return None

     

class PySer(win32serviceutil.ServiceFramework):
    _svc_name_ = "checkLaserProcess"
    _svc_display_name_ = "checkLaserProcess"#服务显示的名称，可以自己修改
    _svc_description_ = "service checkLaserProcess" #服务显示的描述
    def __init__(self, args):
        win32serviceutil.ServiceFramework.__init__(self, args)
        self.hWaitStop = win32event.CreateEvent(None, 0, 0, None)
        self.run = True
    def SvcDoRun(self):

        checkProces.execute()
    def SvcStop(self):
        self.logger.info("service is stop....")
        self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
        win32event.SetEvent(self.hWaitStop)
        self.ReportServiceStatus(win32service.SERVICE_STOPPED)
        self.run = False
if __name__=='__main__':
    # win32serviceutil.HandleCommandLine(PythonService)
    import sys
    import servicemanager
    if len(sys.argv) == 1:
        try:
            evtsrc_dll = os.path.abspath(servicemanager.__file__)
            servicemanager.PrepareToHostSingle(PySer) #如果修改过名字，名字要统一
            servicemanager.Initialize('checkLaserProcess',evtsrc_dll) #如果修改过名字，名字要统一
            servicemanager.StartServiceCtrlDispatcher()
        except win32service.error as details:
            import winerror
            if details == winerror.ERROR_FAILED_SERVICE_CONTROLLER_CONNECT:
                win32serviceutil.usage()
    else:
        win32serviceutil.HandleCommandLine(PySer) #如果修改过名字，名字要统一```