This repository has been archived by the owner on Oct 25, 2020. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
/
process.py
1075 lines (1003 loc) · 51.8 KB
/
process.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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
#coding=utf-8
from mirai import Mirai, Plain, MessageChain, Friend, Image, Group, protocol, Member, At, Face, JsonMessage,LightApp
from variable import *
from function import *
import datetime
import re
import requests
import json
from VGG16 import predictImage
from dateutil.relativedelta import relativedelta
import shutil
adminConfig=["repeat","setu","bizhi","real","speakMode","search"]
adminCheck=["group","speakMode","countLimit","setu","bizhi","real","r18","search"]
hostConfig=["countLimit","r18","switch"]
settingCode={"Disable":0,"Enable":1,"on":1,"off":0,"Local":1,"Net":0,"normal":"normal","zuanLow":"zuanLow","zuanHigh":"zuanHigh","rainbow":"rainbow","chat":"chat","online":"online","offline":"offline","wyy":"wyy","qq":"qq","off":"off"}
sleepMuteCallText=["精致睡眠","晚安","晚安，精致睡眠"]
muteAllCallText=["万籁俱寂"]
unmuteAllCallText=["春回大地","万物复苏"]
blackList=getBlacklist()
# setting语句处理
def settingProcess(groupId,sender,config,change):
if sender in getAdmin(groupId):
if configChangeJudge(config,change):
if config in adminConfig:
updateSetting(groupId,config,settingCode[change])
record("setting:%s set to %s"%(config,change),"none",sender,groupId,True,"function")
else:
if not sender==HostQQ:
record("setting:Insufficient permissions","none",sender,groupId,False,"function")
return [
At(target=sender),
Plain(text="Insufficient permissions!")
]
else:
if change.isnumeric():
updateSetting(groupId,config,change)
else:
updateSetting(groupId,config,settingCode[change])
record("setting:%s set to %s"%(config,change),"none",sender,groupId,True,"function")
if (config=="real" or config=="setu") and change=="Enable" and (getSetting(groupId,"setu") and getSetting(groupId,"real")):
updateSetting(groupId,"forbiddenCount",0)
else:
record("setting:command error","none",sender,groupId,False,"function")
return [
At(target=sender),
Plain("Command error!")
]
return [
Plain(text="group:%d %s set to %s"%(groupId,config,change))
]
else:
return [
At(target=sender),
Plain(text="爬爬爬，你没有管理权限！离人家远一点啦！死变态！")
]
# info语句处理
def infoProcess(groupId,sender,check):
if sender in getAdmin(groupId):
if infoCheckJudge(check):
if check in adminCheck:
return showSetting(groupId,sender,check)
else:
if not sender==HostQQ:
record("setting:Insufficient permissions","none",sender,groupId,False,"function")
return [
At(target=sender),
Plain(text="Insufficient permissions!")
]
else:
return showSetting(groupId,sender,check)
else:
record("setting:command error","none",sender,groupId,False,"function")
return [
At(target=sender),
Plain("Command error!")
]
else:
return [
At(target=sender),
Plain(text="爬爬爬，你没有管理权限！离人家远一点啦！死变态！")
]
# Wiki语句处理
def wikiProcess(groupId,sender,messageText):
firstLevelDirectory=["function","management"]
secondFLevelDirectory=["img","weather","yxh","blhx","ask","translate","speakMode","mute"]
secondMLevelDirectory=["setting","info"]
finalFLevelDirectory={"setu":wikiSetu,"real":wikiReal,"bizhi":wikiBizhi,"search":wikiSearch,"predict":wikiPredict,"weather":wikiWeather,"yxh":wikiYxh,"blhx":wikiBlhx,"ask":wikiAsk,"translate":wikiTranslate,"speakMode":wikiSpeakMode,"mute":wikiMute,"linux":wikiLinux,"quotes":wikiQuotes,"music":wikiMusic,"epidemic":wikiEpidemic}
finalSLevelDirectory={"setuSetting":setuSetting,"r18Setting":r18Setting,"realSetting":realSetting,"bizhiSetting":bizhiSetting,"searchSetting":searchSetting,"countLimitSetting":countLimitSetting,"limitSetting":limitSetting,"blacklistSetting":blacklistSetting,"repeatSetting":repeatSetting,"speakModeSetting":speakModeSetting}
finalILevelDirectory={"repeatInfo":repeatInfo,"setuLocalInfo":setuLocalInfo,"bizhiLocalInfo":bizhiLocalInfo,"countLimitInfo":countLimitInfo,"setuInfo":setuInfo,"bizhiInfo":bizhiInfo,"realInfo":realInfo,"r18Info":r18Info,"speakModeInfo":speakModeInfo,"switchInfo":switchInfo,"allInfo":allInfo,"sysInfo":sysInfo,"groupInfo":groupInfo}
print(messageText)
if messageText.replace("[At::target=%i] "%BotQQ,"")=="wiki" or messageText.replace("[At::target=%i] "%BotQQ,"")=="wiki:wiki":
return [
At(target=sender),
Plain(text="Wiki 下属目录：\n"),
Plain(text="1.function\n"),
Plain(text="2.management\n"),
Plain(text="3.acknowledgement\n"),
Plain(text="4.requirements\n"),
Plain(text="使用方法：@bot wiki:name\n"),
Plain(text="如：@bot wiki:function\n")
]
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"")=="function":
nextMenu=messageText.replace("[At::target=%i] wiki:"%BotQQ,"")
return [
At(target=sender),
Plain(text="function 下属目录：\n"),
Plain(text="1.img(图片功能)\n"),
Plain(text="2.development(开发相关功能)\n"),
Plain(text="3.weather(天气功能)\n"),
Plain(text="4.yxh(营销号生成器功能)\n"),
Plain(text="5.blhx(blhxWiki查询功能)\n"),
Plain(text="6.ask(问我问题给出网址解答)\n"),
Plain(text="7.translate(翻译)\n"),
Plain(text="8.speakMode(不同回复模式)\n"),
Plain(text="9.mute(有关禁言的功能)\n"),
Plain(text="10.quotes(有关群语录)\n"),
Plain(text="11.music(有关点歌功能)\n"),
Plain(text="12.epidemic(疫情查询功能)\n"),
Plain(text="使用方法：@bot wiki:name(不用括号里的)\n"),
Plain(text="如：@bot wiki:img\n")
]
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"")=="img":
return [
At(target=sender),
Plain(text="img 下属目录：\n"),
Plain(text="1.setu(涩图功能)\n"),
Plain(text="2.real(三次元涩图功能)\n"),
Plain(text="3.bizhi(壁纸功能)\n"),
Plain(text="4.search(搜图查询功能)\n"),
Plain(text="5.predict(搜图查询功能)\n"),
Plain(text="使用方法：@bot wiki:name(不用括号里的)\n"),
Plain(text="如：@bot wiki:img\n")
]
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"")=="development":
return [
At(target=sender),
Plain(text="development 下属目录：\n"),
Plain(text="1.linux(linux命令查询功能)\n"),
Plain(text="使用方法：@bot wiki:name(不用括号里的)\n"),
Plain(text="如：@bot wiki:linux\n")
]
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"")=="management":
return [
At(target=sender),
Plain(text="management 下属目录：\n"),
Plain(text="1.setting(设置)\n"),
Plain(text="2.info(查询)\n"),
Plain(text="3.wiki(使用方法(开始套娃))\n"),
Plain(text="使用方法：@bot wiki:name(不用括号里的)\n"),
Plain(text="如：@bot wiki:setting\n")
]
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"")=="setting":
return [
At(target=sender),
Plain(text="setting 下属目录：\n"),
Plain(text="1.imgSetting(图片功能设置)\n"),
Plain(text="2.blacklist(添加黑名单)\n"),
Plain(text="3.repeatSetting(复读设置)\n"),
Plain(text="4.speakModeSetting(不同回复模式设置)\n"),
Plain(text="使用方法：@bot wiki:name(不用括号里的)\n"),
Plain(text="如：@bot wiki:imgSetting\n")
]
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"")=="imgSetting":
return [
At(target=sender),
Plain(text="imgSetting 下属目录：\n"),
Plain(text="1.setuSetting(涩图功能设置)\n"),
Plain(text="2.realSetting(三次元涩图功能设置)\n"),
Plain(text="3.bizhiSetting(壁纸功能设置)\n"),
Plain(text="4.r18Setting(R18设置)\n"),
Plain(text="5.searchSetting(搜图功能设置)\n"),
Plain(text="6.countLimitSetting(限制要图次数(pis/m)功能开关设置)\n"),
Plain(text="7.limitSetting(限制要图次数(pis/m)功能设置)\n"),
Plain(text="使用方法：@bot wiki:name(不用括号里的)\n"),
Plain(text="如：@bot wiki:setuSetting\n")
]
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"")=="info":
return [
At(target=sender),
Plain(text="info 下属目录：\n"),
Plain(text="1.allInfo(全部信息)\n"),
Plain(text="2.groupInfo(群组设置等信息)\n"),
Plain(text="3.sysInfo(系统信息)\n"),
Plain(text="4.imgInfo(图片功能设置信息)\n"),
Plain(text="5.repeatInfo(复读功能设置信息)\n"),
Plain(text="6.speakModeInfo(聊天模式设置信息)\n"),
Plain(text="7.switchInfo(机器人开关信息)\n"),
Plain(text="使用方法：@bot wiki:name(不用括号里的)\n"),
Plain(text="如：@bot wiki:all\n")
]
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"")=="imgInfo":
return [
At(target=sender),
Plain(text="imgInfo 下属目录：\n"),
Plain(text="1.setuInfo(setu设置信息)\n"),
Plain(text="2.realInfo(real设置信息)\n"),
Plain(text="3.r18Info(r18设置信息)\n"),
Plain(text="4.bizhiInfo(bizhi设置信息)\n"),
Plain(text="5.searchInfo(搜图功能设置信息)\n"),
Plain(text="6.predictInfo(预测图片功能设置信息)\n"),
Plain(text="7.setuLocalInfo(setu库位置设置信息)\n"),
Plain(text="8.bizhiLocalInfo(bizhi库位置设置信息)\n"),
Plain(text="9.countLimitInfo(每分钟要图限制设置信息)\n"),
Plain(text="10.limitInfo(每分钟要图数量限制设置信息)\n"),
Plain(text="使用方法：@bot wiki:name(不用括号里的)\n"),
Plain(text="如：@bot wiki:setuInfo\n")
]
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"")=="acknowledgement":
nextMenu=messageText.replace("[At::target=%i] wiki:"%BotQQ,"")
return [
At(target=sender),
Plain(text="致谢名单:\n"),
Plain(text="1.Mirai,一个高效率机器人库\n"),
Plain(text="2.mirai-api-http,提供 http 接口进行接入\n"),
Plain(text="3.python-mirai,Mirai的Python接口\n")
]
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"")=="requirements":
nextMenu=messageText.replace("[At::target=%i] wiki:"%BotQQ,"")
answer=requirements
answer.insert(0,At(target=sender))
return answer
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"") in finalFLevelDirectory:
finalKey=messageText.replace("[At::target=%i] wiki:"%BotQQ,"")
answer=finalFLevelDirectory[finalKey]
answer.insert(0,At(target=sender))
return answer
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"") in finalSLevelDirectory:
finalKey=messageText.replace("[At::target=%i] wiki:"%BotQQ,"")
answer=finalSLevelDirectory[finalKey]
answer.insert(0,At(target=sender))
return answer
elif messageText.replace("[At::target=%i] wiki:"%BotQQ,"") in finalILevelDirectory:
finalKey=messageText.replace("[At::target=%i] wiki:"%BotQQ,"")
answer=finalILevelDirectory[finalKey]
answer.insert(0,At(target=sender))
return answer
else:
return [
At(target=sender),
Plain(text="仔细看看是不是输入错误了呐~")
]
# func语句处理
def funcProcess(groupId,sender,func,content,target,quoteFormat):
if sender in getAdmin(groupId):
if func=="addQuote":
try:
target=re.findall(r'\[At::target=(.*?)\]',target,re.S)[0]
except:
pass
return addCelebrityQuotes(groupId,int(target),content,quoteFormat)
elif func=="addNickname":
try:
target=re.findall(r'\[At::target=(.*?)\]',target,re.S)[0]
except:
pass
return addNickname(groupId,int(target),content)
else:
return [
At(target=sender),
Plain(text="只有主人和管理员才能添加群语录哦~")
]
# 语句处理
async def Process(message,groupId,sender,memberList):
responseCalled=getData("responseCalled")
responseCalled+=1 #responseCalled计数
updateData(responseCalled,"response")
if sender in blackList: #发送人在黑名单中
return "noneReply"
# message预处理
messageText=message.toString()
# setu功能
if messageText in setuCallText:
if sender==80000000:
return [Plain(text="匿名请求已被屏蔽！做涩批就要光明正大！匿名算什么好汉！")]
updateDragon(groupId,sender,"normal")
updateUserCalled(groupId,sender,"setu",1)
setuCalled=getData("setuCalled")
setuCalled+=1 #setuCalled计数
updateData(setuCalled,"setu")
if not getSetting(groupId,"setu"): #本群禁止要setu
forbiddenCount=getSetting(groupId,"forbiddenCount")
forbiddenCount+=1
updateSetting(groupId,"forbiddenCount",forbiddenCount)
record("setu","none",sender,groupId,False,"img")
if forbiddenCount<=3:
return [Plain(text="我们是正规群呐，不搞那一套哦，想看去辣种群看哟~")]
elif forbiddenCount<=6:
return [Plain(text="Kora!都说了是正规群啦！怎么老要这种东西呀，真是够讨厌的呢！再问我就生气啦！")]
elif forbiddenCount<=9:
return [Plain(text="爬爬爬，天天脑子里都是些什么啊，滚呐！爷生气啦！打你哦！")]
else:
return [Image.fromFileSystem(angryDist)]
else:
# if sender in blackList: #发送人在黑名单中
# record("setu","none",sender,groupId,False,"img")
# return [
# At(target=sender),
# Plain(text="要你🐎？大胆妖孽！我一眼就看出来你不是人！大威天龙！世尊地藏！般若诸佛！般若巴麻空！")
# ]
if getSetting(groupId,"tribute") and not getTributeInfo(sender,"VIP"): #是否已完成上贡
return [
At(target=sender),
Plain(text="啊嘞嘞~你还不是VIP哦~不能使用setu功能呐~\n"),
Plain(text="快来完成每月指标吧~\n"),
Plain(text="群内发送'开始上贡'即可开始发送图片哦~\n"),
Plain(text="群内发送'停止上贡'即可停止发送图片哦~")
]
if getSetting(groupId,"countLimit"): #如果有每分钟调用次数限制
if not getMemberPicStatus(groupId,sender):
record("setu","none",sender,groupId,False,"img")
return [Plain(text="你已达到限制，每分钟最多只能要%d张setu/real哦~\n歇会儿再来吧！"%getSetting(groupId,"limit"))]
if getSetting(groupId,"tribute") and not getTributeInfo(sender,"VIP"): #是否已完成上贡
return [
At(target=sender),
Plain(text="啊嘞嘞~你还不是VIP哦~不能使用setu功能呐~\n"),
Plain(text="快来完成每月指标吧~\n"),
Plain(text="群内发送'开始上贡'即可开始发送图片哦~\n"),
Plain(text="群内发送'停止上贡'即可停止发送图片哦~")
]
if getSetting(groupId,"setuLocal"): #是否为本地库
if getSetting(groupId,"imgLightning") and randomJudge():
record("setu","lightning",sender,groupId,False,"img")
return "lightningPic"
if getSetting(groupId,"r18"):
dist=randomPic(setu18Dist)
record("setu18",dist,sender,groupId,True,"img")
else:
dist=randomPic(setuDist)
record("setu",dist,sender,groupId,True,"img")
print("本地setu图片地址：",dist)
return [Image.fromFileSystem(dist)]
else:
pass #因api变动不稳定，暂时不进行编写
# real功能
elif messageText=="real":
if sender==80000000:
return [Plain(text="匿名请求已被屏蔽！做涩批就要光明正大！匿名算什么好汉！")]
updateDragon(groupId,sender,"normal")
updateUserCalled(groupId,sender,"real",1)
realCalled=getData("realCalled")
realCalled+=1 #realCalled计数
updateData(realCalled,"real")
if not getSetting(groupId,"real"): #本群禁止要real
forbiddenCount=getSetting(groupId,"forbiddenCount")
forbiddenCount+=1
updateSetting(groupId,"forbiddenCount",forbiddenCount)
record("real","none",sender,groupId,False,"img")
if forbiddenCount<=3:
return [Plain(text="我们是正规群呐，不搞那一套哦，想看去辣种群看哟~")]
elif forbiddenCount<=6:
return [Plain(text="Kora!都说了是正规群啦！怎么老要这种东西呀，真是够讨厌的呢！再问我就生气啦！")]
elif forbiddenCount<=9:
return [Plain(text="爬爬爬，天天脑子里都是些什么啊，滚呐！爷生气啦！打你哦！")]
else:
return [Image.fromFileSystem(angryDist)]
else:
# if sender in blackList: #发送人在黑名单中
# record("real",dist,sender,groupId,False,"img")
# return [Plain(text="要要要你🐎？大胆妖孽！我一眼就看出来你不是人！大威天龙！世尊地藏！般若诸佛！般若巴麻空！")]
if getSetting(groupId,"tribute") and not getTributeInfo(sender,"VIP"): #是否已完成上贡
return [
At(target=sender),
Plain(text="啊嘞嘞~你还不是VIP哦~不能使用real功能呐~\n"),
Plain(text="快来完成每月指标吧~\n"),
Plain(text="群内发送'开始上贡'即可开始发送图片哦~\n"),
Plain(text="群内发送'停止上贡'即可停止发送图片哦~")
]
if getSetting(groupId,"countLimit"): #如果有每分钟调用次数限制
if not getMemberPicStatus(groupId,sender):
record("real","none",sender,groupId,False,"img")
return [
At(target=sender),
Plain(text="你已达到限制，每分钟最多只能要%d张setu/real哦~\n歇会儿再来吧！"%getSetting(groupId,"limit"))
]
if getSetting(groupId,"imgLightning") and randomJudge():
record("real","lightning",sender,groupId,False,"img")
return "lightningPic"
dist=randomPic(realDist)
record("real",dist,sender,groupId,True,"img")
print("本地real图片地址：",dist)
return [Image.fromFileSystem(dist)]
# bizhi功能
elif messageText=="bizhi":
bizhiCalled=getData("bizhiCalled")
bizhiCalled+=1 #bizhiCalled计数
updateData(bizhiCalled,"bizhi")
updateUserCalled(groupId,sender,"bizhi",1)
if not getSetting(groupId,"bizhi"): #本群禁止要bizhi
record("bizhi","none",sender,groupId,False,"img")
return [Plain(text="bizhi功能被关闭了呐>^<,想打开的话联系下管理员呐~")]
if getSetting(groupId,"tribute") and not getTributeInfo(sender,"VIP"): #是否已完成上贡
return [
At(target=sender),
Plain(text="啊嘞嘞~你还不是VIP哦~不能使用bizhi功能呐~\n"),
Plain(text="快来完成每月指标吧~\n"),
Plain(text="群内发送'开始上贡'即可开始发送图片哦~\n"),
Plain(text="群内发送'停止上贡'即可停止发送图片哦~")
]
# else:
# if sender in blackList: #发送人在黑名单中
# record("bizhi",dist,sender,groupId,False,"img")
# return [Plain(text="要要要你🐎？大胆妖孽！我一眼就看出来你不是人！大威天龙！世尊地藏！般若诸佛！般若巴麻空！")]
dist=randomPic(bizhiDist)
print("本地bizhi图片地址：",dist)
record("bizhi",dist,sender,groupId,True,"img")
return [Image.fromFileSystem(dist)]
# 批量pic功能
elif messageText[:5]=="setu*" or messageText[:5]=="real*":
aim=messageText[:4]
if aim=="setu":
aimDist=setuDist
else:
aimDist=realDist
if not ((getSetting(groupId,"setu") and aim=="setu") or (getSetting(groupId,"real") and aim=="real")): #本群禁止要setu
forbiddenCount=getSetting(groupId,"forbiddenCount")
forbiddenCount+=1
updateSetting(groupId,"forbiddenCount",forbiddenCount)
record(messageText,"none",sender,groupId,False,"img")
if forbiddenCount<=3:
return [Plain(text="我们是正规群呐，不搞那一套哦，想看去辣种群看哟~")]
elif forbiddenCount<=6:
return [Plain(text="Kora!都说了是正规群啦！怎么老要这种东西呀，真是够讨厌的呢！再问我就生气啦！")]
elif forbiddenCount<=9:
return [Plain(text="爬爬爬，天天脑子里都是些什么玩意儿，滚呐！爷生气啦！打你哦！")]
else:
return [Image.fromFileSystem(angryDist)]
else:
try:
num=int(message.toString()[5:])
if sender in getAdmin(groupId):
if sender == HostQQ or num <= 5:
if aim=="setu":
setuCalled=getData("setuCalled")
setuCalled+=num
updateData(setuCalled,"setu")
else:
realCalled=getData("realCalled")
realCalled+=num
updateData(realCalled,"real")
picMsg=["pic*"]
for i in range(num):
# if setting[groupId]["setuLocal"]:
dist=randomPic(aimDist)
picMsg.append(Image.fromFileSystem(dist))
record("%s*%d"%(aim,num),dist,sender,groupId,True,"img")
# else:
# pass
return picMsg
else:
record("%s*%d"%(aim,num),"none",sender,groupId,False,"img")
return [Plain(text="管理最多也只能要5张呐~我可不会被轻易玩儿坏呢！！！！")]
elif num <= 5:
record("%s*%d"%(aim,num),"none",sender,groupId,False,"img")
return [Plain(text="只有主人和管理员可以使用%s*num命令哦~你没有权限的呐~"%aim)]
else:
record("%s*%d"%(aim,num),"none",sender,groupId,False,"img")
return [Plain(text="不是管理员你要你🐎呢？老色批！还要那么多？给你🐎一拳，给爷爬！")]
except ValueError:
return [Plain(text="命令错误！%s*后必须跟数字！"%aim)]
# 搜图功能
elif messageText in searchCallText:
if not getSetting(groupId,"search"):
return [
At(target=sender),
Plain(text="搜图功能关闭了呐~想要打开就联系机器人管理员吧~")
]
if getSetting(groupId,"tribute") and not getTributeInfo(sender,"VIP"): #是否已完成上贡
return [
At(target=sender),
Plain(text="啊嘞嘞~你还不是VIP哦~不能使用搜图功能呐~\n"),
Plain(text="快来完成每月指标吧~\n"),
Plain(text="群内发送'开始上贡'即可开始发送图片哦~\n"),
Plain(text="群内发送'停止上贡'即可停止发送图片哦~")
]
setReady(groupId,sender,True,"searchReady")
return [
At(target=sender),
Plain(text="请发送要搜索的图片呐~")
]
elif message.hasComponent(Image) and getSetting(groupId,"search") and getReady(groupId,sender,"searchReady"):
print("searching")
img = message.getFirstComponent(Image)
return searchImage(groupId,sender,img)
# 图片预测功能
elif messageText=="这张图里是什么":
if not getSetting(groupId,"imgPredict"):
return [
At(target=sender),
Plain(text="图片预测功能关闭了呐~想要打开就联系机器人管理员吧~")
]
if getSetting(groupId,"tribute") and not getTributeInfo(sender,"VIP"): #是否已完成上贡
return [
At(target=sender),
Plain(text="啊嘞嘞~你还不是VIP哦~不能使用图片预测功能呐~\n"),
Plain(text="快来完成每月指标吧~\n"),
Plain(text="群内发送'开始上贡'即可开始发送图片哦~\n"),
Plain(text="群内发送'停止上贡'即可停止发送图片哦~")
]
setReady(groupId,sender,True,"PredictReady")
return [
At(target=sender),
Plain(text="请发送要预测的图片呐(推荐真实图片呐)~")
]
elif message.hasComponent(Image) and getSetting(groupId,"imgPredict") and getReady(groupId,sender,"predictReady"):
print("predicting")
updateUserCalled(groupId,sender,"imgPredict",1)
img = message.getFirstComponent(Image)
return predictImage(groupId,sender,img)
# 黄图评价功能
elif messageText=="这张图涩吗":
if not getSetting(groupId,"yellowPredict"):
return [
At(target=sender),
Plain(text="图片涩度评价功能关闭了呐~想要打开就联系机器人管理员吧~")
]
if getSetting(groupId,"tribute") and not getTributeInfo(sender,"VIP"): #是否已完成上贡
return [
At(target=sender),
Plain(text="啊嘞嘞~你还不是VIP哦~不能使用涩度评价功能呐~\n"),
Plain(text="快来完成每月指标吧~\n"),
Plain(text="群内发送'开始上贡'即可开始发送图片哦~\n"),
Plain(text="群内发送'停止上贡'即可停止发送图片哦~")
]
setReady(groupId,sender,True,"yellowPredictReady")
return [
At(target=sender),
Plain(text="请发送要预测的图片呐~")
]
elif message.hasComponent(Image) and getSetting(groupId,"yellowPredict") and getReady(groupId,sender,"yellowPredictReady"):
print("judging")
updateUserCalled(groupId,sender,"yellowPredict",1)
img = message.getFirstComponent(Image)
return judgeImageYellow(groupId,sender,img.url,yellowJudgeDist)
# 上贡图片Judge
elif message.hasComponent(Image) and getSetting(groupId,"tribute") and getReady(groupId,sender,"tributeReady"):
# setReady(groupId,sender,False,"tributeready")
print("judging")
img = message.getFirstComponent(Image)
result=judgeImageYellow(groupId,sender,img.url,tributeDist)
Normal=float(re.findall(r'Normal :(.*?)%',result[2].text)[0])
Hot=float(re.findall(r'Hot :(.*?)%',result[3].text)[0])
Sexy=float(re.findall(r'Sexy :(.*?)%',result[4].text)[0])
Total=float(re.findall(r'Total :(.*?)%',result[5].text)[0])
if Normal<=55 or Hot>50 or Sexy>50 or Total>20:
code=getData("searchCount")
hammRes=imgSimilarJudge(imgHash("%s%d.png"%(tributeDist,code)),"tribute",3)
if hammRes[0]:
removedFile="%s%d.png"%(tributeDist,code)
shutil.move(removedFile,tributeSimilarDist)
return [
"上贡",
At(target=sender),
Plain(text="\n有相似图片在tribute图库中呐~\n"),
Plain(text="汉明距离%d\n"%hammRes[1]),
Plain(text="再找些别的图吧~")
]
else:
hammRes=imgSimilarJudge(imgHash("%s%d.png"%(tributeDist,code)),"setu",3)
if hammRes[0]:
removedFile="%s%d.png"%(tributeDist,code)
shutil.move(removedFile,tributeSimilarDist)
return [
"上贡",
At(target=sender),
Plain(text="\n有相似图片在setu图库中呐~\n"),
Plain(text="汉明距离%d\n"%hammRes[1]),
Plain(text="再找些别的图吧~")
]
else:
hammRes=imgSimilarJudge(imgHash("%s%d.png"%(tributeDist,code)),"setu18",3)
if hammRes[0]:
removedFile="%s%d.png"%(tributeDist,code)
shutil.move(removedFile,tributeSimilarDist)
return [
"上贡",
At(target=sender),
Plain(text="\n有相似图片在setu18图库中呐~\n"),
Plain(text="汉明距离%d\n"%hammRes[1]),
Plain(text="再找些别的图吧~")
]
insertHash("%s%d.png"%(tributeDist,code),imgHash("%s%d.png"%(tributeDist,code)),"tribute")
record("tribute accepted","%s%d.png"%(tributeDist,code),sender,groupId,1,"img")
if int(getTributeInfo(sender,"tributeCount"))+1>=int(getSetting(groupId,"tributeQuantity")):
setTributeInfo(sender,getTributeInfo(sender,"tributeCount")+1,"tributeCount")
now=datetime.datetime.now()
end=now + relativedelta(months=+1)
now=now.strftime("%Y-%m-%d %H:%M:%S")
end=end.strftime("%Y-%m-%d %H:%M:%S")
setTributeInfo(sender,now,"startTime")
setTributeInfo(sender,end,"endTime")
setTributeInfo(sender,True,"VIP")
setReady(groupId,sender,False,"tributeready")
return [
"上贡",
At(target=sender),
Plain(text="恭喜~这张图通过了测试呐~\n"),
Plain(text="本月目标%d张，现在已经传了%d张呐~\n"%(getSetting(groupId,"tributeQuantity"),getTributeInfo(sender,"tributeCount"))),
Plain(text="你已获得VIP身份辣~\n"),
Plain(text="VIP从现在开始到%s哦"%end)
]
else:
setTributeInfo(sender,getTributeInfo(sender,"tributeCount")+1,"tributeCount")
return [
"上贡",
At(target=sender),
Plain(text="恭喜~这张图通过了测试呐~\n"),
Plain(text="本月目标%d张，现在已经传了%d张呐~\n"%(getSetting(groupId,"tributeQuantity"),getTributeInfo(sender,"tributeCount"))),
Plain(text="群内发送'停止上贡'即可停止发送图片哦~")
]
else:
code=getData("searchCount")
insertHash("%s%d.png"%(tributeDist,code),imgHash("%s%d.png"%(tributeDist,code)),"tribute")
removedFile="%s%d.png"%(tributeDist,code)
shutil.move(removedFile,tributeDelDist)
record("tribute not accepted","%s%d.png"%(tributeDelDist,code),sender,groupId,1,"img")
return [
"上贡",
At(target=sender),
Plain(text="很遗憾，这张图没有达标呢~\n"),
Plain("本图评价结果：\n"),
result[2],
result[3],
result[4],
result[5],
Plain(text="\n评判标准为：\n"),
Plain(text="Normal<=55 or Hot>50 or Sexy>50 or Total>20\n"),
Plain(text="再找些别的图吧~\n"),
Plain(text="群内发送'停止上贡'即可停止发送图片哦~")
]
# 笑话功能
elif "来点" in messageText and "笑话" in messageText:
jokeDict={"苏联":"soviet","法国":"french","法兰西":"french","美国":"america","美利坚":"america"}
name=re.findall(r'来点(.*?)笑话',messageText,re.S)
if name==['']:
return [
At(target=sender),
Plain(text="来点儿啥笑话啊，你又不告诉人家！哼！")
]
elif name[0] in jokeDict.keys():
record("joke","none",sender,groupId,True,"function")
return getKeyJoke(jokeDict[name[0]])
else:
record("joke","none",sender,groupId,True,"function")
return getJoke(name[0])
# 群语录功能 celebrityQuotes
elif messageText=="群语录":
return getCelebrityQuotes(groupId,memberList,"none","random")
# 群语录精准迫害功能
elif "来点" in messageText and "语录" in messageText:
name=re.findall(r'来点(.*?)语录',messageText,re.S)
if name==[]:
return "noneReply"
else:
record("quotes","none",sender,groupId,True,"function")
return getCelebrityQuotes(groupId,memberList,name,"select")
# 网抑云
elif messageText=="网抑云" or messageText=="wyy":
return getWyy()
# 群友平安
elif messageText=="平安":
return safe(groupId,memberList)
# 历史上的今天
elif messageText=="历史上的今天":
return getHistoryToday()
# 开始上贡
elif messageText=="开始上贡":
if not getSetting(groupId,"tribute"):
return [
At(target=sender),
Plain(text="本群未开启上贡功能哦~")
]
if int(getTributeInfo(sender,"tributeCount"))>=int(getSetting(groupId,"tributeQuantity")):
setReady(groupId,sender,False,"tributeready")
return [
At(target=sender),
Plain(text="你这个月已经发够%s张符合标准的图了呐~可以不用发啦~"%getSetting(groupId,"tributeQuantity"))
]
else:
setReady(groupId,sender,True,"tributeready")
return [
At(target=sender),
Plain(text="现在可以发图啦~本月目标%d张，现在已经传了%d张呐~"%(getSetting(groupId,"tributeQuantity"),getTributeInfo(sender,"tributeCount")))
]
# 停止上贡
elif messageText=="停止上贡":
if not getSetting(groupId,"tribute"):
return [
At(target=sender),
Plain(text="本群未开启上贡功能哦~")
]
setReady(groupId,sender,False,"tributeready")
return [
At(target=sender),
Plain(text="上贡暂停啦~本月目标%d张，现在已经传了%d张呐~"%(getSetting(groupId,"tributeQuantity"),getTributeInfo(sender,"tributeCount")))
]
# 获取时间功能（可选背景）
elif messageText in timeCallText:
clockCalled=getData("clockCalled")
clockCalled+=1
updateData(clockCalled,"clock")
if getClockChoice(groupId,sender)=="none":
clockMessage=[
At(target=sender),
Plain(text="你还没有选择表盘哦~快来选择一个吧~\n"),
Plain(text="看中后直接发送选择表盘+序号即可哦~\n"),
Plain(text="如:选择表盘1\n"),
Plain(text="表盘预览:")
]
clockList = os.listdir(clockPreviewDist)
clockList.sort(key=lambda x:int(x[:-4]))
index=1
for i in clockList:
clockMessage.append(Plain(text="\n%s."%index))
clockMessage.append(Image.fromFileSystem(clockPreviewDist+i))
index+=1
return clockMessage
else:
t = datetime.datetime.now() #目前时间
t = t.strftime("%H:%M")
t = t.replace(":","")
dist=timeDist+str(getClockChoice(groupId,sender))+"/%s.png"%t
return [Image.fromFileSystem(dist)]
# 选择表盘（获取时间功能）
elif messageText[:4]=="选择表盘":
if messageText=="选择表盘":
return showClock(sender)
else:
code=messageText[4:]
if code.isdigit() and int(code)<=int(getData("dialsCount")):
recordClock(groupId,sender,int(code))
return[
Plain(text="已经选择了表盘%s呢!\n现在可以问我时间啦~"%code)
]
else:
return [
Plain(text="看中后直接发送选择表盘+序号即可哦~\n"),
Plain(text="再检查下有没有输错呢~\n")
]
# 天气查询功能
elif "[At::target=%i] 天气"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
weatherCalled=getData("weatherCalled")
weatherCalled+=1
updateData(weatherCalled,"weather")
return getWeather(message,sender)
# 碧蓝航线wiki查询功能
elif "[At::target=%i] blhx："%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
name=messageText[28:]
return blhxWiki(sender,name)
# 营销号生成器
elif "[At::target=%i] 营销号"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
_,somebody,something,other_word=messageText.split('、')
# print(something,somebody,other_word)
return yingxiaohao(somebody,something,other_word)
# 问你点儿事儿
elif "[At::target=%i] 问你点儿事儿："%BotQQ in message.toString():
updateUserCalled(groupId,sender,"at",1)
question=message.toString()[30:]
question=parse.quote(question)
return askSth(sender,question)
# myInfo功能
elif "[At::target=%i] myInfo"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
return getUserCalled(groupId,sender,memberList)
# lspRank功能
elif "[At::target=%i] rank"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
return getRank(groupId,memberList)
# linux命令查询功能
elif "[At::target=%i] linux"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
if '：' in messageText:
messageText=messageText.replace('：',':')
command=messageText.replace("[At::target=%i] linux:"%BotQQ,"")
print("get linux:%s"%command)
text=getLinuxExplanation(command)
if text=="error!no command!":
return [
At(target=sender),
Plain(text="未搜索到命令%s!请检查拼写！"%command)
]
else:
return [
At(target=sender),
Plain(text="%s:%s"%(command,text))
]
# 翻译功能
elif "[At::target=%i] "%BotQQ in messageText and "用" in messageText and "怎么说" in messageText:
updateUserCalled(groupId,sender,"at",1)
supportLanguage={"中文":"zh","英文":"en","日文":"jp","韩文":"kr","法文":"fr","西班牙文":"es","意大利文":"it","德文":"de","土耳其文":"tr","俄文":"ru","葡萄牙文":"pt","越南文":"vi","印度尼西亚文":"id","马来西亚文":"ms","泰文":"th"}
tp=re.findall(r'\] (.*?)用(.*?)怎么说',messageText,re.S)[0]
text=tp[0]
target=tp[1]
print("text:%s,target:%s"%(text,target))
source=textDetect(text.encode("utf-8"))
if target not in supportLanguage.keys():
sL=""
for i in supportLanguage.keys():
sL+=i
sL+='、'
return [
At(target=sender),
Plain(text="目前只支持翻译到%s哦~\n要全字匹配哦~看看有没有打错呐~\n翻译格式：text用（目标语言）怎么说"%sL)
]
target=supportLanguage[target]
# print(target)
return translate(groupId,sender,text,source,target)
# 设置处理
elif "[At::target=%i] setting."%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
command=messageText[16:]
try:
print(command)
name,config,change=command.split('.')
print(name,'-->'," config:",config,"set to",change)
return settingProcess(groupId,sender,config,change)
except ValueError:
return [
At(target=sender),
Plain(text="Command error! Use the '@bot command' command to query the commands you can use!")
]
# 获取信息处理
elif "[At::target=%i] info."%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
command=messageText[16:]
# try:
print(command)
info,check=command.split('.')
print(info,'-->'," info:",check)
return infoProcess(groupId,sender,check)
# wiki处理
elif "[At::target=%i] wiki"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
if '：' in messageText:
messageText=messageText.replace('：',':')
print(messageText)
print("get wiki:%s"%messageText.replace("[At::target=%i] wiki"%BotQQ,""))
return wikiProcess(groupId,sender,messageText)
# 添加群语录处理 @bot func.addQuote.content.target
elif "[At::target=%i] func.addQuote."%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
if message.hasComponent(Image):
try:
_,func,target,img=messageText.split(".")
quotesCount=getData("quotesCount")+1
print(quotesCount)
updateData(quotesCount,"quotes")
dist="%s%s.png"%(quotesDist,quotesCount)
print(dist)
img=message.getFirstComponent(Image)
img_content=requests.get(img.url).content
image=IMG.open(BytesIO(img_content))
image.save(dist)
return funcProcess(groupId,sender,func,pymysql.escape_string(dist),target,"img")
except Exception:
pass
else:
try:
_,func,content,target=messageText.split(".")
return funcProcess(groupId,sender,func,content,target,"text")
except Exception:
pass
# 添加别名处理 @bot func.addQuote.content.target
elif "[At::target=%i] func.addNickname."%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
try:
_,func,nickname,target=messageText.split(".")
return funcProcess(groupId,sender,func,nickname,target,"nickname")
except Exception:
pass
# 添加管理员处理
elif "[At::target=%i] addAdmin"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
target=int(re.findall(r'At::target=(.*?)]',message.toString()[19:],re.S)[0])
print("add admin:%d in group %d"%(target,groupId))
return addAdmin(groupId,target)
# 删除管理员处理
elif "[At::target=%i] deleteAdmin"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
target=int(re.findall(r'At::target=(.*?)]',message.toString()[19:],re.S)[0])
print("delete admin:%d in group %d"%(target,groupId))
return deleteAdmin(groupId,target)
# 加入黑名单处理
elif "[At::target=%i] addBlacklist"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
target=int(re.findall(r'At::target=(.*?)]',message.toString()[19:],re.S)[0])
if target in blackList:
return [Plain(text="%s is already in the blacklist")%qq2name(memberList,target)]
else:
blackList.append(target)
print("%s is added to the blacklist"%qq2name(memberList,target))
return addToBlacklist(target)
# 移出黑名单处理
elif "[At::target=%i] removeBlacklist"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
target=int(re.findall(r'At::target=(.*?)]',messageText[19:],re.S)[0])
if target not in blackList:
return [Plain(text="%s is not in the blacklist originally")%qq2name(memberList,target)]
else:
blackList.remove(target)
print("%s is removed from the blacklist"%qq2name(memberList,target))
return removeFromBlacklist(target)
# 添加监听处理
elif "[At::target=%i] addListen"%BotQQ in messageText:
updateUserCalled(groupId,sender,"at",1)
target=int(re.findall(r'At::target=(.*?)]',message.toString()[19:],re.S)[0])
print("add listen:%d in group %d"%(target,groupId))
Msg=addListen(groupId,target)
Msg.insert(0,"addListen.%d.%d"%(groupId,target))
return Msg
# 获取疫情统计
elif messageText=="疫情" or messageText=="疫情统计":
return [LightApp(getEpidemic())]
# 点歌
elif messageText[:3]=="点歌 ":
if getSetting(groupId,"music")!="off":
target=messageText[3:]
updateUserCalled(groupId,sender,"songOrder",1)
return [LightApp(songOrder(target))]
else:
return [
At(target=sender),
Plain(text="点歌功能关闭了呐~想要打开就去找主人吧~")
]
# 微博热搜
elif messageText=="weibo" or messageText=="微博" or messageText=="微博热搜":
return getWeiboHot()
# 超度
elif "超度" in messageText and sender==HostQQ:
target=int(re.findall(r'\[At::target=(.*?)\]',messageText,re.S)[0])