Skip to content

Commit

Permalink
Update main.py
Browse files Browse the repository at this point in the history
  • Loading branch information
@chase535
chase535 authored Sep 8, 2022
1 parent 1370a5a commit 8e0ada1
Showing 1 changed file with 39 additions and 37 deletions.
76 changes: 39 additions & 37 deletions WeChatPush/main.py
View file
Original file line number Diff line number Diff line change
@@ -1,43 +1,45 @@
# coding=utf-8

import sys
import os
import signal
from datetime import datetime

pid = str(os.getpid())

pid = os.getpid()
ppid = os.getppid()

def error(pid):
print(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + '程序运行出现错误，终止运行，错误信息已保存至程序目录下的error.log文件中')
with open(str((os.path.split(os.path.realpath(__file__))[0]).replace('\\', '/')) + '/error.log', 'a', encoding='utf-8') as f:
f.write(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + str(traceback.format_exc()) + '\n')
try:
os.killpg(os.getpgid(str(pid)), signal.SIGKILL)
except:
os.system('taskkill /F /T /PID ' + str(pid))

import sys
import signal
from datetime import datetime

if int(sys.version_info.major) < 3:
print(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + '程序仅支持Python3.x版本运行，程序强制停止运行')
try:
os.killpg(os.getpgid(os.getpid()), signal.SIGKILL)
except:
os.system('taskkill /F /T /PID ' + str(pid))
if float(str(sys.version_info.major) + '.' + str(sys.version_info.minor)) < 3.4:
print(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + '程序仅支持Python3.4及以上版本运行，程序强制停止运行')
os.kill(pid, signal.SIGTERM)

import requests
import importlib
import traceback
import platform
import time
from requests.packages import urllib3
from multiprocessing import Process, Manager


def error(pid, ppid):
system = str(platform.system())
print(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + '程序运行出现错误，终止运行，错误信息已保存至程序目录下的error.log文件中')
with open(str((os.path.split(os.path.realpath(__file__))[0]).replace('\\', '/')) + '/error.log', 'a', encoding='utf-8') as f:
f.write(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + str(traceback.format_exc()) + '\n')
if system == 'Linux':
os.killpg(os.getpgid(int(pid)), signal.SIGTERM)
elif system == 'Windows':
os.system('taskkill /F /T /PID ' + str(pid))
else:
os.kill(int(ppid), signal.SIGTERM)


try:
import config
except:
print(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + '配置获取异常,请检查配置文件是否存在/权限是否正确/语法是否有误')
error(str(pid))
error(pid, ppid)

if int(config.async_components):
import asyncio
Expand All @@ -52,15 +54,15 @@ def config_update(value):
importlib.reload(config)
except:
print(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + '配置获取异常,请检查配置文件是否存在/权限是否正确/语法是否有误')
error(str(value.get('pid')))
error(value.get('pid'), value.get('ppid'))
break
shield_mode_update = '0'
newcfg = {'pid': str(value.get('pid')), 'chat_push': str(config.chat_push), 'VoIP_push': str(config.VoIP_push),
'tdtt_alias': str(config.tdtt_alias), 'FarPush_regID': str(config.FarPush_regID),
'WirePusher_ID': str(config.WirePusher_ID), 'FarPush_Phone_Type': str(config.FarPush_Phone_Type),
'shield_mode': str(config.shield_mode), 'blacklist': list(config.blacklist),
'whitelist': list(config.whitelist), 'tdtt_interface': str(config.tdtt_interface),
newcfg = {'chat_push': str(config.chat_push), 'VoIP_push': str(config.VoIP_push), 'tdtt_alias': str(config.tdtt_alias),
'FarPush_regID': str(config.FarPush_regID), 'WirePusher_ID': str(config.WirePusher_ID),
'FarPush_Phone_Type': str(config.FarPush_Phone_Type), 'shield_mode': str(config.shield_mode),
'blacklist': list(config.blacklist), 'whitelist': list(config.whitelist), 'tdtt_interface': str(config.tdtt_interface),
'FarPush_interface': str(config.FarPush_interface), 'WirePusher_interface': str(config.WirePusher_interface)}
for a in value.keys():
for a in newcfg.keys():
if str(a) == 'shield_mode':
if newcfg.get('shield_mode') != value.get('shield_mode'):
if int(newcfg.get('shield_mode')):
Expand All @@ -81,7 +83,7 @@ def config_update(value):
except KeyboardInterrupt:
pass
except:
error(str(value.get('pid')))
error(value.get('pid'), value.get('ppid'))


def run(func):
Expand Down Expand Up @@ -150,7 +152,7 @@ def simple_reply(msg):
print(str(Name) + ': ' + str(typesymbol))
if str(msg.get('Type')) == str(itchat.content.VOIP):
if str(value.get('VoIP_push')) == '1' and str(value.get('tdtt_alias')) != '':
data_send(str(value.get('tdtt_interface')), title='微信 ' + str(Name), content=str(typesymbol), alias=str(value.get('tdtt_alias')))
data_send(str(value.get('tdtt_interface')), title='微信 ' + str(Name), content=str(typesymbol), alias=str(value.get('tdtt_alias')), action='weixin://', type='WeChat_VoIP')
elif str(value.get('VoIP_push')) == '2' and str(value.get('FarPush_regID')) != '':
data_send(str(value.get('FarPush_interface')), title='微信 ' + str(Name), content=str(typesymbol), regID=str(value.get('FarPush_regID')), phone=str(value.get('FarPush_Phone_Type')), through='0')
elif str(value.get('VoIP_push')) == '3' and str(value.get('WirePusher_ID')) != '':
Expand All @@ -159,7 +161,7 @@ def simple_reply(msg):
print(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + '配置有误，请更改配置')
else:
if str(value.get('chat_push')) == '1' and str(value.get('tdtt_alias')) != '':
data_send(str(value.get('tdtt_interface')), title='微信 ' + str(Name), content=str(typesymbol), alias=str(value.get('tdtt_alias')))
data_send(str(value.get('tdtt_interface')), title='微信 ' + str(Name), content=str(typesymbol), alias=str(value.get('tdtt_alias')), action='weixin://', type='WeChat_chat')
elif str(value.get('chat_push')) == '2' and str(value.get('FarPush_regID')) != '':
data_send(str(value.get('FarPush_interface')), title='微信 ' + str(Name), content=str(typesymbol), regID=str(value.get('FarPush_regID')), phone=str(value.get('FarPush_Phone_Type')), through='0')
elif str(value.get('chat_push')) == '3' and str(value.get('WirePusher_ID')) != '':
Expand All @@ -175,11 +177,11 @@ def simple_reply(msg):
run(itchat.check_login())
run(itchat.auto_login(hotReload=True, enableCmdQR=2))
value = Manager().dict()
value.update({'pid': str(pid), 'chat_push': str(config.chat_push), 'VoIP_push': str(config.VoIP_push),
'tdtt_alias': str(config.tdtt_alias), 'FarPush_regID': str(config.FarPush_regID),
'WirePusher_ID': str(config.WirePusher_ID), 'FarPush_Phone_Type': str(config.FarPush_Phone_Type),
'shield_mode': str(config.shield_mode), 'blacklist': list(config.blacklist),
'whitelist': list(config.whitelist), 'tdtt_interface': str(config.tdtt_interface),
value.update({'pid': str(pid), 'ppid': str(ppid), 'chat_push': str(config.chat_push),
'VoIP_push': str(config.VoIP_push), 'tdtt_alias': str(config.tdtt_alias),
'FarPush_regID': str(config.FarPush_regID), 'WirePusher_ID': str(config.WirePusher_ID),
'FarPush_Phone_Type': str(config.FarPush_Phone_Type), 'shield_mode': str(config.shield_mode),
'blacklist': list(config.blacklist), 'whitelist': list(config.whitelist), 'tdtt_interface': str(config.tdtt_interface),
'FarPush_interface': str(config.FarPush_interface), 'WirePusher_interface': str(config.WirePusher_interface)})
conf_update = Process(target=config_update, args=(value, ))
conf_update.daemon = True
Expand All @@ -192,4 +194,4 @@ def simple_reply(msg):
except KeyboardInterrupt:
print(str(datetime.now().strftime('[%Y.%m.%d %H:%M:%S] ')) + '由于键盘输入^C（ctrl+C），程序强制停止运行')
except:
error(str('pid'))
error(pid, ppid)

0 comments on commit 8e0ada1

Please sign in to comment.