-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy patharchive_snap_pic.py
121 lines (114 loc) · 4.35 KB
/
archive_snap_pic.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
#!/usr/bin/python3
import xmlrpc.client
import os,sys,re,json,time
import socket,glob,shutil
#本程序功能为，将制定的snap目录下的snap开头的文件名，按照日期，分别归档到类似20200401这样的目录中去
#避免snap目录下文件数量太多，list时间太长
#client端的snap目录归档完毕后，自动把server端程序的snap目录下的snap文件，都delete掉
def load_cfg(cfg_file): #载入配置文件，设置相关的参数
#下面是默认参数配置
res={'hosts':['10.10.20.59:8888','10.10.20.59:7777'], #默认要监控的主机列表
'interval':30, #默认轮询间隔秒数
'timeout':30, #设置socket默认超时，从而影响xmlrpc.client的连接超时
'black_list':['360game.exe','qqlive.exe'], #进程黑名单，强制全部小写比对
'browsers':['chrome.exe', #谷歌chrome浏览器
'360se.exe', #360浏览器
'iexplore.exe', #老ie浏览器
'MicrosoftEdge.exe', #微软Edge
'MicrosoftEdgeCP.exe', #微软Edge
'ApplicationFrameHost.exe', #微软Edge
'firefox.exe', #火狐浏览器
'sogouexplorer.exe', #搜狗浏览器
'qqbrowser.exe', #qq浏览器
'wechatweb.exe' #微信内置浏览器
], #哪些浏览器被关注，强制小写比对
'browser_keywords':['游戏','game'], #浏览器敏感词识别
'do_kill':1, #满足black_list、condition_browser相关条件，是否kill
'seconds':10, #默认每次录音多少秒
'allow_periods':[{'p':'12:15-13:30','w':'1-5'}, #周一至周五，12:15-13:30可以玩，不kill
{'p':'12:15-14:30','w':'6'}, #周六，12:15-14:30可以玩，不kill
{'p':'17:45-19:30','w':'6,7'}, #周六日，17:45-19:30可以玩，不kill
{'p':'12:00-13:20','w':'7'}, #周日，12:00-13:20可以玩，不kill
], #哪些时间段可以
'snap_dir':'snap/', #抓取的截图和录音，放置的目录位置
'hide':0, #是否隐藏自己
}
#要是配置文件不存在，则新建一个配置文件，用默认参数值
if not os.path.isfile(cfg_file):
try:
f=open(cfg_file,'w')
f.write(json.dumps(res,indent=2))
f.close()
except Exception as e:
print(e)
return res
#要是配置文件存在，尝试载入配置文件，返回其设置的参数值
try:
f=open(cfg_file,'r')
res=json.loads(f.read())
f.close()
except Exception as e:
print("ERROR:"+str(e))
print('will use default config')
return res
def snap2arch(snapDir): #挂在给定snapDir目录下的图片，批量move到按日期分类的下面
files=glob.glob(os.path.join(snapDir,'snap*.*'))
for fname in files:
match=re.findall(r'snap[a-zA-Z]+(\d+)_(\d+)\.\S+',fname)
if match:
date=match[0][0]
date=os.path.join(snapDir,date)
if not os.path.isdir(date):
try:
os.mkdir(date)
except:
pass
try:
print("moving %s to %s" % (fname,date))
shutil.move(fname,date)
except Exception as e:
print(e)
pass
def portPing(host,port): #检测主机对应的端口是否活着
port=int(port)
s=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
print("checking host=%s,port=%s" % (host,port))
s.settimeout(1)
try:
s.connect((host,port))
except:
return False
return True
def selectHost(uriList): #给定一组host:port的列表，返回第一个能连通的
for uri in uriList:
(host,port)=uri.split(':')
if portPing(host,port):
return (host,port)
#调用RPC
def main():
global cfg
cfg_file='ChildControl.ini'
cfg=load_cfg(cfg_file)
print("cfg=%s" % json.dumps(cfg,indent=2))
socket.setdefaulttimeout(cfg['timeout']) #每次socket连接，默认最长多久超时
resUri=selectHost(cfg['hosts'])
if resUri:
uri='http://%s:%s' % (resUri[0],resUri[1])
mylog("try to %s" % uri)
s = xmlrpc.client.ServerProxy(uri)
else:
print("all in cfg['hosts'] can not to conenct!:%s" % str(cfg['hosts']))
sys.exit(1)
remote_app_path=s.get_path()['app_path']
print("=====Trying to list current png=====")
res=s.my_exec("dir %s\\snap*.*" % remote_app_path)
print("stdout=%s\nsterr=%s" % (res['stdout'],res['stderr']) )
print("=====Trying to delete snap =====")
res=s.my_exec("del %s\\snap*.*" % remote_app_path)
print("stdout=%s\nsterr=%s" % (res['stdout'],res['stderr']) )
print("=====check the delete result=====")
res=s.my_exec("dir %s\\snap*.*" % remote_app_path)
print("stdout=%s\nsterr=%s" % (res['stdout'],res['stderr']) )
snap2arch('/a/snap')
if __name__=='__main__':
main()