예제 #1
0
class Statistic(QObject):

    # 定义信号
    # 第1个int表示requst包的数量,第2个int表示response包的数量
    tracker_pkt_cnt_changed = pyqtSignal(int, int)

    # 第1个int表示handshake包的数量,第2个int表示message包的数量
    peer_pkt_cnt_changed = pyqtSignal(int, int)

    # tracker中的信息增加时发送信号,更新UI信息
    tracker_info_changed = pyqtSignal()

    # peer中信息改变是发送的信号,更新UI信息
    peer_info_changed = pyqtSignal()



    def __init__(self):
        super(Statistic, self).__init__()
        # ip : tracker
        self.iip = '192.168.1.9'
        self.tracker_stat = {}
        self.peer_stat = {}

        self.tracker_req_pkt_cnt = 0
        self.tracker_res_pkt_cnt = 0
        self.peer_msg_pkt_cnt = 0
        self.peer_hs_pkt_cnt = 0
        self.peer_upload_time = {}
        self.peer_download_time = {}
        self.ctrl = Control()
        # log
        self.log = Log()

    def add_tracker_pkt(self, pkt, type):
        # # 发送信
        # self.dataChanged.emit("old status", "new status")
        pkt_info = pkt.packet_info
        cur_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
        if type == 'response':
            self.tracker_res_pkt_cnt+=1
            ip = pkt_info.sip
            port = pkt_info.sport
        else:
            self.tracker_req_pkt_cnt+=1
            ip = pkt_info.dip
            port = pkt_info.dport

        if ip in self.tracker_stat.keys():
            stat = self.tracker_stat[ip]
        else:
            stat = Tracker_Info('udp', ip, port)
            stat = Tracker_Info('udp', ip, port)
            self.tracker_stat[ip] = stat
        if type == 'response':
            if pkt.action == 0:
                self.log.add_log('[%s]:udp tracker[ip:%s port:%s]进行了connect response\n' % (cur_time, ip, port))
            elif pkt.action == 1:
                seeders = pkt.seeders
                user_cnt = len(pkt.ip_port_list)
                stat.set_seeders(seeders)
                stat.set_users(user_cnt)
                self.log.add_log('[%s]:udp tracker[ip:%s port:%s]进行了announce response\n' % (cur_time, ip, port))
            elif pkt.action == 2:
                self.log.add_log('[%s]:udp tracker[ip:%s port:%s]进行了scrapy response\n' % (cur_time, ip, port))
                pass
            elif pkt.action == 3:
                self.log.add_log('[%s]:udp tracker[ip:%s port:%s]进行了error response\n' % (cur_time, ip, port))
                pass
        else:
            if pkt.action == 0:
                pass
                self.log.add_log('[%s]:本地主机向udp tracker[ip:%s port:%s]进行了connect request\n' % (cur_time, ip, port))
            elif pkt.action == 1:
                pass
                self.log.add_log('[%s]:本地主机向udp tracker[ip:%s port:%s]进行了announce request\n' % (cur_time, ip, port))
            elif pkt.action == 2:
                pass
                self.log.add_log('[%s]:本地主机向udp tracker[ip:%s port:%s]进行了scrapy request\n' % (cur_time, ip, port))
        self.tracker_info_changed.emit()
        self.tracker_pkt_cnt_changed.emit(self.tracker_res_pkt_cnt, self.tracker_req_pkt_cnt)

    def add_http_tracker(self, pkt, type):
        # self.dataChanged.emit("old status", "new status")
        pkt_info = pkt.packet_info
        cur_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
        if type == 'response':
            self.tracker_res_pkt_cnt += 1
            ip = pkt_info.sip
            port = pkt_info.sport
            self.log.add_log('[%s]:http tracker[ip:%s port:%s]进行了announce response\n' % (cur_time, ip, port))
        else:
            self.tracker_req_pkt_cnt += 1
            ip = pkt_info.dip
            port = pkt_info.dport
            self.log.add_log('[%s]:本地主机向udp tracker[ip:%s port:%s]发送了annnounce request\n' % (cur_time, ip, port))

        if ip in self.tracker_stat.keys():
            stat = self.tracker_stat[ip]
        else:
            stat = Tracker_Info('http', ip, port)
            self.tracker_stat[ip] = stat

        self.tracker_info_changed.emit()
        self.tracker_pkt_cnt_changed.emit(self.tracker_res_pkt_cnt, self.tracker_req_pkt_cnt)

    def add_peer_pkt(self, pkt):
        ticks = time.time()
        cur_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
        pkt_info = pkt.packet_info

        if pkt_info.sip == self.iip:
            ip = pkt_info.dip
            port = pkt_info.dport
        else:
            ip = pkt_info.sip
            port = pkt_info.sport

        if Control.block_all_flag:
            self.ctrl.add_blocked_addr(ip, port)

        if ip in self.peer_stat.keys():
            stat = self.peer_stat[ip]
        else:
            stat = Peer_Info(ip, port)
            self.peer_stat[ip] = stat

        # 统计握手协议
        if isinstance(pkt, Peer_Handshake):
            self.peer_hs_pkt_cnt+=1
            self.log.add_log('[%s]:本地主机与peer[ip:%s port:%s]进行了handshake\n' % (cur_time, ip, port))
        else:
            self.log.add_log('[%s]:本地主机与peer[ip:%s port:%s]发送了数据\n' % (cur_time, ip, port))
            self.peer_msg_pkt_cnt+=1
            last_time = 0
            speed = self.peer_stat[ip].upload_speed
            # 上传
            if self.iip == pkt_info.sip:
                if ip in self.peer_upload_time.keys():
                    last_time = self.peer_upload_time[ip]
            # 下载
            else:
                if ip in self.peer_download_time.keys():
                    last_time = self.peer_download_time[ip]
            # 计算速度
            if ticks - last_time ==0:
                pass
            else:
                speed = round(pkt_info.pld_len / (ticks - last_time), 2)
            # 上传
            if self.iip == pkt_info.sip:
                self.peer_upload_time[ip] = ticks
                self.peer_stat[ip].upload_speed = speed
            # 下载
            else:
                self.peer_download_time[ip] = ticks
                self.peer_stat[ip].download_speed = speed
        self.peer_info_changed.emit()
        self.peer_pkt_cnt_changed.emit(self.peer_hs_pkt_cnt, self.peer_msg_pkt_cnt)
예제 #2
0
class Init:

    def __init__(self):

        self.log = Log()
        self.face_reg = FaceReg("file_path", self.log)
        self.arrangement_reader = ArrangementReader(self.log)
        self.camera_manager = CameraManager(self.log)
        self.tts = self.log.tts

        self.is_end = False

        self.sign_result = {
            "扫地": {},
            "拖地": {},
            "杂物间+柜台": {},
            "摆桌椅": {},
            "窗户+窗台": {},
            "图书角+黑板+讲台": {},
            "倒垃圾": {}
        }

    def time_count(self, second, callback):

        """
        计时器
        :param second: 计算的秒数
        :param callback: 时间到以后的callback
        :return:
        """
        self.log.add_log("Init: Start time count for" + str(second), 1)
        time.sleep(second)
        callback()

    def generate_report(self):

        """
        开始生成报告
        :return:
        """
        json.dump(self.sign_result, open("./data/json/report.json", "w", encoding="utf-8"))

    def init_sign_window(self):

        """
        初始化签名窗口
        :return:
        """

        # INIT THE SIGN WINDOW
        window = Tk()
        window.title("自动签到系统")
        window.geometry('1024x600')

        # SET THE NOTICE LABLE
        notice = Label(window, text="请点击按钮并签到", font=("Helvetica", 20))
        notice.pack(side=TOP)

        # 摆桌椅Frame
        desk_place = Frame(window, height=25)
        desc_desk = Label(desk_place, text="摆桌椅签名区", font=("Helvetica", 15))
        desc_desk.pack(side=TOP)
        row_1_button = Button(desk_place, height=10, width=10, command=lambda : self.sign(1, 1), text="列1")
        row_1_button.pack(side=LEFT)
        row_2_button = Button(desk_place, height=10, width=10, command=lambda: self.sign(1, 2), text="列2")
        row_2_button.pack(side=LEFT)
        row_3_button = Button(desk_place, height=10, width=10, command=lambda: self.sign(1, 3), text="列3")
        row_3_button.pack(side=LEFT)
        row_4_button = Button(desk_place, height=10, width=10, command=lambda: self.sign(1, 4), text="列4")
        row_4_button.pack(side=LEFT)
        row_5_button = Button(desk_place, height=10, width=10, command=lambda: self.sign(1, 5), text="列5")
        row_5_button.pack(side=LEFT)
        row_6_button = Button(desk_place, height=10, width=10, command=lambda: self.sign(1, 6), text="列6")
        row_6_button.pack(side=LEFT)
        row_7_button = Button(desk_place, height=10, width=10, command=lambda: self.sign(1, 7), text="列7")
        row_7_button.pack(side=LEFT)
        row_8_button = Button(desk_place, height=10, width=10, command=lambda: self.sign(1, 8), text="列8")
        row_8_button.pack(side=LEFT)

        desk_place.pack(side=TOP)

        # 窗户+窗台签名区
        window_place = Frame(window, height=25)
        window_desk = Label(window_place, text="窗户+窗台签名区", font=("Helvetica", 15))
        window_desk.pack(side=TOP)
        row_1_button = Button(window_place, height=10, width=10, command=lambda: self.sign(2, 1), text="窗1")
        row_1_button.pack(side=LEFT)
        row_2_button = Button(window_place, height=10, width=10, command=lambda: self.sign(2, 2), text="窗2")
        row_2_button.pack(side=LEFT)
        row_3_button = Button(window_place, height=10, width=10, command=lambda: self.sign(2, 3), text="窗3")
        row_3_button.pack(side=LEFT)
        row_4_button = Button(window_place, height=10, width=10, command=lambda: self.sign(2, 4), text="窗4")
        row_4_button.pack(side=LEFT)

        window_place.pack(side=TOP)

        # 黑板签名区
        bb_place = Frame(window, height=25, width=50)
        bb_notice = Label(bb_place, text="黑板+图书角签名区", font=("Helvetica", 15))
        bb_notice.pack(side=TOP)
        button = Button(bb_place, height=10, width=10, command=lambda: self.sign(3, 1), text="点ME")
        button.pack(side=LEFT)

        # ATTENTION: PACK IN THE END

        # 扫地签名
        sweep_place = Frame(window, height=25)
        sweep_notice = Label(sweep_place, text="扫地签名区", font=("Helvetica", 15))
        sweep_notice.pack(side=TOP)
        row_1_button = Button(sweep_place, height=10, width=10, command=lambda: self.sign(4, 1), text="列1")
        row_1_button.pack(side=LEFT)
        row_2_button = Button(sweep_place, height=10, width=10, command=lambda: self.sign(4, 2), text="列2")
        row_2_button.pack(side=LEFT)
        row_3_button = Button(sweep_place, height=10, width=10, command=lambda: self.sign(4, 3), text="列3")
        row_3_button.pack(side=LEFT)
        row_4_button = Button(sweep_place, height=10, width=10, command=lambda: self.sign(4, 4), text="列4")
        row_4_button.pack(side=LEFT)
        row_5_button = Button(sweep_place, height=10, width=10, command=lambda: self.sign(4, 5), text="列5")
        row_5_button.pack(side=LEFT)
        row_6_button = Button(sweep_place, height=10, width=10, command=lambda: self.sign(4, 6), text="列6")
        row_6_button.pack(side=LEFT)
        row_7_button = Button(sweep_place, height=10, width=10, command=lambda: self.sign(4, 7), text="列7")
        row_7_button.pack(side=LEFT)
        row_8_button = Button(sweep_place, height=10, width=10, command=lambda: self.sign(4, 8), text="列8")
        row_8_button.pack(side=LEFT)
        row_9_button = Button(sweep_place, height=10, width=10, command=lambda: self.sign(4, 9), text="后面+讲台")
        row_9_button.pack(side=LEFT)

        sweep_place.pack(side=TOP)

        # 拖地签名
        mop_place = Frame(window, height=25)
        mop_notice = Label(window, text="拖地签名区", font=("Helvetica", 15))
        mop_notice.pack(side=TOP)
        row_1_button = Button(mop_place, height=10, width=10, command=lambda: self.sign(5, 1), text="列1")
        row_1_button.pack(side=LEFT)
        row_2_button = Button(mop_place, height=10, width=10, command=lambda: self.sign(5, 2), text="列2")
        row_2_button.pack(side=LEFT)
        row_3_button = Button(mop_place, height=10, width=10, command=lambda: self.sign(5, 3), text="列3")
        row_3_button.pack(side=LEFT)
        row_4_button = Button(mop_place, height=10, width=10, command=lambda: self.sign(5, 4), text="列4")
        row_4_button.pack(side=LEFT)
        row_5_button = Button(mop_place, height=10, width=10, command=lambda: self.sign(5, 5), text="列5")
        row_5_button.pack(side=LEFT)
        row_6_button = Button(mop_place, height=10, width=10, command=lambda: self.sign(5, 6), text="列6")
        row_6_button.pack(side=LEFT)
        row_7_button = Button(mop_place, height=10, width=10, command=lambda: self.sign(5, 7), text="列7")
        row_7_button.pack(side=LEFT)
        row_8_button = Button(mop_place, height=10, width=10, command=lambda: self.sign(5, 8), text="列8")
        row_8_button.pack(side=LEFT)
        row_8_button = Button(mop_place, height=10, width=10, command=lambda: self.sign(5, 9), text="后面+讲台")
        row_8_button.pack(side=LEFT)

        mop_place.pack(side=TOP)

        # 倒垃圾签名
        trash_place = Frame(window, height=25, width=50)
        trash_notice = Label(trash_place, text="倒垃圾签名区", font=("Helvetica", 15))
        trash_notice.pack(side=TOP)
        button = Button(trash_place, height=10, width=10, command=lambda: self.sign(6, 1), text="点ME")
        button.pack(side=LEFT)

        trash_place.pack(side=LEFT)

        # 杂物间签名区
        room_place = Frame(window, height=25, width=50)
        room_notice = Label(room_place, text="杂物间签名区", font=("Helvetica", 15))
        room_notice.pack(side=TOP)
        button = Button(room_place, height=10, width=10, command=lambda: self.sign(7, 1), text="点ME")
        button.pack(side=LEFT)

        room_place.pack(side=LEFT)

        bb_place.pack(side=LEFT)

        # RUN THE WINDOW
        window.mainloop()

    def sign(self, command, row):

        """
        进行签名
        :param command: 签到模式
        :param row: 列数
        :return:
        """
        img = self.camera_manager.capture_image()
        self.camera_manager.show_image(img)

        self.generate_report()
        detect_res = self.face_reg.face_detect(self.face_reg.read_img("img.jpg"))
        try:
            face_num = detect_res["result"]["face_num"]
        except KeyError:
            print(detect_res)
            raise KeyError

        while True:
            if face_num == 0:
                self.tts.start("没有检测到人脸,请对准摄像头重拍")
            else:
                # self.tts.start("Face Detected. Start searching...")
                break

        search_res = self.face_reg.face_search(self.face_reg.read_img("img.jpg"))
        try:
            user_id = search_res["result"]["user_list"][0]["user_id"]
        except:
            print(search_res)
            raise KeyError

        if command == 1:
            index = self.arrangement_reader.get_index(self.arrangement_reader.get_name(int(user_id)),
                                                      self.arrangement_reader.class_arrangement["摆桌椅"])
            if index is False:
                self.tts.start("你可不是干这个的啊!")
                return
            if index + 1 == int(row):
                self.sign_result["摆桌椅"][index] = True
                self.tts.start("签到完成")

                next_name = self.arrangement_reader.class_arrangement["扫地"][index]
                self.tts.start("请" + next_name + "同学开始扫地")
                self.tts.start("你负责的是" + str(index + 1) + "列,要扫干净哦!第九列是指教室的后面和讲台")

            else:
                self.tts.start("错误的人脸和对应列数!")
        elif command == 2:
            index = self.arrangement_reader.get_index(self.arrangement_reader.get_name(int(user_id)),
                                                      self.arrangement_reader.class_arrangement["窗户+窗台"])
            if index is False:
                self.tts.start("你可不是干这个的啊!")
                return

            if index + 1 == int(row):
                self.sign_result["窗户+窗台"][index] = True
                self.tts.start("签到完成")
            else:
                self.tts.start("错误的人脸和对应窗户号!")
        elif command == 3:
            index = self.arrangement_reader.get_index(self.arrangement_reader.get_name(int(user_id)),
                                                      self.arrangement_reader.class_arrangement["图书角+黑板+讲台"])
            if index is False:
                self.tts.start("你可不是干这个的啊!")
                return

            self.sign_result["图书角+黑板+讲台"][index] = True
            self.tts.start("签到完成")
        elif command == 4:
            index = self.arrangement_reader.get_index(self.arrangement_reader.get_name(int(user_id)),
                                                      self.arrangement_reader.class_arrangement["扫地"])
            if index is False:
                self.tts.start("你可不是干这个的啊!")
                return

            if index + 1 == int(row):
                try:
                    if self.sign_result["摆桌椅"][index] is True:

                        self.sign_result["扫地"][index] = True
                        self.tts.start("签到完成")

                        next_name = self.arrangement_reader.class_arrangement["拖地"][index]
                        self.tts.start("请" + next_name + "同学开始拖地")
                        self.tts.start("你负责的是" + str(index + 1) + "列,要拖干净哦!第九列是指教室的后面和讲台")

                        is_end = True
                        for i in self.sign_result["扫地"].keys():
                            if self.sign_result["扫地"][i] is False:
                                is_end = False
                        if is_end:
                            trash = self.arrangement_reader.get_names(self.arrangement_reader.class_arrangement["倒垃圾"])

                            self.tts.start(trash)
                            self.tts.start("请这些同学现在开始倒垃圾")
                except KeyError:
                    self.tts.start("摆桌椅的还没摆完呢!")
                    return
                else:
                    self.tts.start("你的任务还没有开始呢!")
                    # c = str(input("ALLOW?: "))
                    # if c == 1:
                    #     self.sign_result["扫地"][index] = True
                    #     self.tts.start("签到完成")
            else:
                self.tts.start("错误的人脸和对应列数!")
        elif command == 5:
            index = self.arrangement_reader.get_index(self.arrangement_reader.get_name(int(user_id)),
                                                      self.arrangement_reader.class_arrangement["拖地"])
            if index is False:
                self.tts.start("你可不是干这个的啊!")
                return

            if index + 1 == int(row):
                try:
                    if self.sign_result["扫地"][index] is True:
                        self.sign_result["拖地"][index] = True
                        self.tts.start("签到完成")

                        is_end = True
                        for i in self.sign_result["拖地"].keys():
                            if self.sign_result["拖地"][i] is False:
                                is_end = False
                        if is_end:
                            room = self.arrangement_reader.get_names(self.arrangement_reader.class_arrangement["杂物间+柜台"])

                            self.tts.start(room)
                            self.tts.start("这些同学则现在开始清理杂物间和柜台")
                    else:
                        self.tts.start("你的任务还没有开始呢!")
                        # c = str(input("ALLOW?: "))
                        # if c == 1:
                        #     self.sign_result["拖地"][index] = True
                        #     self.tts.start("签到完成")
                except KeyError:
                    self.tts.start("扫地的还没扫呢!你怎么签名")
                    # c = str(input("ALLOW?: "))
                    # if c == 1:
                    #     self.sign_result["拖地"][index] = True
                    #     self.tts.start("签到完成")
                    return
            else:
                self.tts.start("错误的人脸和对应列数!")
        elif command == 6:
            index = self.arrangement_reader.get_index(self.arrangement_reader.get_name(int(user_id)),
                                                      self.arrangement_reader.class_arrangement["倒垃圾"])
            if index is False:
                self.tts.start("你可不是干这个的啊!")
                return

            self.sign_result["倒垃圾"][index] = True
            self.tts.start("签到完成")
        elif command == 7:
            index = self.arrangement_reader.get_index(self.arrangement_reader.get_name(int(user_id)),
                                                      self.arrangement_reader.class_arrangement["杂物间+柜台"])

            if index is False:
                self.tts.start("你可不是干这个的啊!")
                return
            self.sign_result["杂物间+柜台"][index] = True
            self.tts.start("签到完成")

    def phase1_start(self):

        """
        run
        :return:
        """
        self.log.add_log("Phase1: now start", 1)
        self.tts.start("现在,开始第一阶段。请以下同学留下,其余同学全部出去或在后面等着")
        table = self.arrangement_reader.get_names(self.arrangement_reader.class_arrangement["摆桌椅"])
        window = self.arrangement_reader.get_names(self.arrangement_reader.class_arrangement["窗户+窗台"])
        blackboard = self.arrangement_reader.get_names(self.arrangement_reader.class_arrangement["图书角+黑板+讲台"])

        self.tts.start(table)
        self.tts.start("这些同学负责摆桌椅,现在开始工作。按照摆完以后请过来选择列数完成签名,")
        self.tts.start("摆桌椅要求:全部拉开,保 证一定距离以方面扫地拖地;椅子要翻上去")

        time.sleep(0.5)
        self.tts.start(window)
        self.tts.start("然后,这些同学负责擦窗台和窗户。按照读的顺序,每人一个窗,从只有窗户的一侧从前往后数,门的一侧也是从前往后数")
        self.tts.start("窗户要求:用抹布擦干以后找报纸弄干,擦的时候要有规律,水不要太多。没有抹布要么等要么去对面借,完成以后过来签名")

        time.sleep(0.5)
        self.tts.start(blackboard)
        self.tts.start("这些同学负责整理黑板、图书柜、讲台")
        self.tts.start("黑板要擦干净,擦完以后一定打开屏幕,不要遮住了。讲台不要有垃圾,粉笔灰擦干净,完成后过来签名")

        video_recorder = threading.Thread(target=self.camera_manager.capture_video, args=("phase1-record1.mp4",))
        video_recorder.start()

        self.init_sign_window()

    def run(self):

        """
        启动程序
        :return:
        """
        self.log.add_log("Init: System is now running...", 1)
        self.log.add_log("Init: Initializing...", 1)
        time_count_thread = threading.Thread(target=self.time_count, args=(1200, self.generate_report))
        time_count_thread.start()

        self.tts.start("Hi~ 这里是袁翊闳开发的【自动劳动管理系统】")
        self.tts.start("主人~ 现在,是时候开始劳动了~")
        #
        self.arrangement_reader.read_in()

        # public = self.arrangement_reader.get_names(self.arrangement_reader.public_labor_arranagement)
        # self.tts.start(public)
        # self.tts.start("这些同学马上跟着正诺恒,拿上一个扫把和垃圾铲,还有两个垃圾袋去工地!")

        self.phase1_start()

        self.generate_report()