pytg_script/replymet.py at master · chliny/pytg_script · GitHub

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
#!/usr/bin/python
#-*- coding=utf-8 -*-

import logging
import sys
import time
import os
from telegram_python import PyTelegram

class ReplyMet(PyTelegram):
    def __init__(self):
        super(ReplyMet, self).__init__()
        self.MAX_METLEN = 100
        self.listened_list = []
        self.metlist=[]
        self.get_met_history()
        self.dialogs = self.get_dialog_list()

    def print_help(self):
        print("python3 %s [ingress_id] [groupname1] [groupname2] ..." \
                % __file__)

    def get_dialoginfo(self, target):
        for dialog_info in self.dialogs:
            for key in ["title", "username", "print_name"]:
                if key in dialog_info and dialog_info[key] == target:
                    return dialog_info
        return {}

    def get_met_history(self):
        line_list = []
        if not os.path.exists("metlist.txt"):
            return

        with open("metlist.txt") as fd:
            for line in fd.readlines()[-1*self.MAX_METLEN:]:
                try:
                    line_sp = line.split("\t")
                    self.metlist.append(line_sp[-1].strip())
                except Exception as e :
                    logging.error(e)
        return

    def parse_argv(self):
        if len(sys.argv) < 2:
            return self.print_help()

        try:
            ingress_id = sys.argv[1].lower()
            self.selfname = [ingress_id]
            self_info = self.sender.get_self()
            self_username = self_info["username"].lower()
            self.selfname.append("@"+self_username)
            logging.debug(self.selfname)
        except Exception as e :
            logging.error(e)
            return False

        if len(sys.argv) > 2:
            for group in sys.argv[2:]:
                try:
                    group_info = self.get_dialoginfo(group)
                    self.listened_list.append(group_info["id"])
                except Exception as e:
                    logging.error(e)
        self.receive_loop()
        return True

    def met(self, username, group,mettype):
        try:
            word = "/met @%s" % username
            if mettype==0 :
                word= u"@%s Please only met me once!" % username
            self.sender.send_msg(group, word)
        except Exception as e:
            logging.error(e)
            return False
        return True

    def parse_recive(self, msg_dict):
        try:
            if msg_dict["event"] != "message":
                return False
        except Exception as e:
            logging.error(e)
            logging.debug(msg_dict)
            return False

        try:
            receive_info = msg_dict["receiver"]
            receive_type = receive_info["type"]
            if receive_type not in ["channel", "chat"] \
                    or (receive_type == "chat" and  receive_info["members_num"] <= 2):
                return False
        except Exception as e:
            logging.error(e)
            logging.debug(msg_dict)
            return False

        try:
            if "own" in msg_dict and msg_dict["own"]:
                return False
            if "mention" in msg_dict and not msg_dict["mention"]:
                return False
        except Exception as e:
            logging.error(e)
            logging.debug(msg_dict)
            return False

        try:
            logging.debug(msg_dict)
            sender_info = msg_dict["sender"]
            receiver_id = msg_dict["receiver"]["id"]
        except Exception as e:
            logging.error(e)
            logging.debug(msg_dict)
            return False

        if self.listened_list and receiver_id not in self.listened_list:
            return False

        try:
            msg_text = msg_dict["text"]
            msg_split = msg_text.strip().split()
            if len(msg_split)>=2 and msg_split[0].lower() == "/met" and msg_split[1].lower() in self.selfname:
                if sender_info["username"] in self.metlist:
                    self.met(sender_info["username"], receiver_id,0)
                else:
                    self.met(sender_info["username"], receiver_id,1)
                    f=open("metlist.txt","a")
                    f.write(time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())))
                    f.write('\t')
                    f.write(sender_info["username"])
                    f.write('\n')
                    f.close()
                    if len(self.metlist)>=self.MAX_METLEN :
                        del self.metlist[0]
                        self.metlist.append(sender_info["username"])
                    else :
                        self.metlist.append(sender_info["username"])
        except Exception as e :
            logging.error(e)
            return False
        return True

if __name__ == "__main__":
    rm = ReplyMet()
    rm.parse_argv()