コード例 #1
0
    def update_driver_list(self, rid):
        service = DCModelService(get_db_conn())
        clients = self.server_socket.get_room_clients(rid)
        dids = []
        for item in clients:
            if item.driver and item.driver['did'] not in dids:
                dids.append(item.driver['did'])

        drivers = []

        for did in dids:
            driver = service.SearchDriverWithDid(did)
            if driver is not None:
                del driver['username']
                del driver['password']
                del driver['created_at']
                del driver['avatar']
                if driver['badge'] is None:
                    driver['badge'] = "1"
                drivers.append(driver)
        send_json = {
            'type': 'sys',
            'detail': 'driver list',
            'rid': rid,
            'drivers': drivers
        }
        self.server_socket.send_clients(
            min_json_dumps_to_bytes(send_json) + b'\n', clients)
コード例 #2
0
 def handle_detail_room_list(self, msg):
     self.log_request('room list')
     service = DCModelService(get_db_conn())
     rooms = service.ListRooms()
     for i in rooms:
         i['created_at'] = i['created_at'].strftime("%Y-%m-%d %H:%M:%S %z")
     send_json = {'type': 'sys', 'detail': 'room list', 'rooms': rooms}
     self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
コード例 #3
0
 def handle_detail_sign_up(self, msg):
     self.log_request('sign up')
     service = DCModelService(get_db_conn())
     driver = msg['driver']
     res = service.CreateDriver(driver['username'], driver['password'],
                                driver['name'])
     send_json = {
         "type": 'sys',
         'detail': 'sign up',
         'status': res['status'],
         'msg': 'Create Success' if res['status'] else 'Create Failed'
     }
     if res['status']:
         send_json['did'] = res['did']
     self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
コード例 #4
0
 def handle_detail_badge(self, msg):
     send_json = msg
     service = DCModelService(get_db_conn())
     if msg['updown'] == 'down':
         send_json['status'] = True
         badges = service.GetBadge(msg['did'])
         badge = badges[0] if len(badges) > 0 else None
         if badge == None:
             send_json['status'] = False
             self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
         else:
             with open(config.FILE_DIR + badge, 'rb') as f:
                 data_bytes = f.read()
             send_json['length'] = len(data_bytes)
             self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
             self.sendall(data_bytes)
             self.sendall(b'\n')
コード例 #5
0
 def handle_detail_driver_avatar(self, msg, data_bytes=None):
     send_json = msg
     service = DCModelService(get_db_conn())
     if msg['updown'] == 'up':
         # 验证身份
         if msg['driver']['did'] != self.driver['did']:
             send_json['status'] = False
             self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
             return
         name = gen_filename(data_bytes[0:20] +
                             str(time.clock()).encode('utf-8'))
         if name is None:
             name = 'driver_default.png'
         path = config.FILE_DIR + name
         # 写文件
         try:
             with open(path, 'wb') as f:
                 f.write(data_bytes)
         except Exception as e:
             logging.warn(str(e))
             send_json['status'] = False
             self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
             return
         service.SetAvatar(msg['driver']['did'], name)
     elif msg['updown'] == 'down':
         logging.info("dealing with down")
         send_json['status'] = True
         try:
             name = service.GetAvatar(msg['driver']['did'])
             if not name:
                 name = 'driver_default.png'
             path = config.FILE_DIR + name
             with open(path, 'rb') as f:
                 data_bytes = f.read()
             send_json['length'] = len(data_bytes)
             self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
             self.sendall(data_bytes)
             self.sendall(b'\n')
         except Exception as e:
             logging.warn(str(e))
             traceback.print_exc()
             send_json['status'] = False
             self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
             return
コード例 #6
0
 def handle_detail_room_avatar(self, msg, data_bytes=None):
     send_json = msg
     service = DCModelService(get_db_conn())
     if msg['updown'] == 'down':
         send_json['status'] = True
         try:
             name = service.GetRoomAvatar(msg['room']['rid'])
             if name is None:
                 name = 'room_default.png'
             path = config.FILE_DIR + name
             with open(path, 'rb') as f:
                 data_bytes = f.read()
             send_json['length'] = len(data_bytes)
             self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
             self.sendall(data_bytes)
             self.sendall(b'\n')
         except Exception as e:
             send_json['status'] = False
             self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
             return
コード例 #7
0
 def handle_detail_sign_in(self, msg):
     self.log_request('sign in')
     send_json = {
         'type': 'sys',
         'detail': 'sign in',
         'status': True,
         'msg': ''
     }
     service = DCModelService(get_db_conn())
     req_driver = msg['driver']
     logging.info(req_driver)
     driver = service.SearchDriverWithUsername(req_driver['username'])
     if driver != None and driver['password'] == hashlib.sha256(
             req_driver['password'].encode('utf-8') +
             config.DRIVER_SALT).hexdigest():
         self.driver = {
             'username': driver['username'],
             'did': driver['did']
         }
         # 登录成功
         send_json['driver'] = {
             'did': driver['did'],
             'name': driver['name'],
             'badge': driver['badge'],
             'created_at':
             driver['created_at'].strftime("%Y-%m-%d %H:%M:%S %z"),
             'avatar': driver['avatar']
         }
         send_json['msg'] = 'Sign in Success'
         send_json['status'] = True
         self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')
     else:
         # 登录失败
         send_json['status'] = False
         send_json['msg'] = 'Sign in Fail'
         self.sendall(min_json_dumps_to_bytes(send_json) + b'\n')