Ejemplo n.º 1
0
    def is_video_file_ready(self, torrent_handle, is_fast, log_enabled=True):
        if torrent_handle:
            status     = torrent_handle.status()
            video_file = self._get_video_file_from_torrent(torrent_handle)

            if int(status.state) >= 3 and video_file:
                complete_pieces = utils.get_video_file_complete_pieces(torrent_handle, video_file)
                total_pieces    = utils.get_video_file_total_pieces(torrent_handle, video_file)
                needed_pieces   = utils.get_preload_buffer_piece_count(torrent_handle, video_file)

                if is_fast or complete_pieces >= needed_pieces:
                    return True
                else:
                    if log_enabled:
                        self.bus.log('[Downloader] Not enough pieces yet: {0}/{1} (total: {2}) @ {3} kB/s'.format(complete_pieces, needed_pieces, total_pieces, status.download_rate / 1024))
            else:
                if log_enabled:
                    self.bus.log('[Downloader] Not ready yet: {0}'.format(str(status.state)))
        else:
            if log_enabled:
                self.bus.log('[Downloader] Not ready yet')

        return False
Ejemplo n.º 2
0
    def get_status(self):
        result = {}

        if self.session:
            result['session'] = self.torrent_config
            
            result['session']['connection_sets'] = []
            for info_hash, connection_set in self.bus.connection_monitor.torrent_connections.iteritems():
                connection_set_status = { 'info_hash': info_hash, 'timestamp': connection_set['timestamp'], 'connections':[] }
                
                for connection in connection_set['set']:
                    connection_status = { 'info_hash': info_hash, 'connection': connection }
                    connection_set_status['connections'].append(connection_status)

                result['session']['connection_sets'].append(connection_set_status)

            result['session']['torrents'] = []
            for torrent_handle in self.torrent_handles:
                torrent_valid  = torrent_handle.has_metadata()
                torrent_status = torrent_handle.status() if torrent_valid else None
                
                try:
                    torrent = {}
                    torrent['paused']        = torrent_status.paused
                    torrent['state']         = str(torrent_status.state)
                    torrent['state_index']   = int(torrent_status.state)
                    torrent['progress']      = math.trunc(torrent_status.progress * 100.0) / 100.0
                    torrent['download_rate'] = torrent_status.download_rate / 1024
                    torrent['upload_rate']   = torrent_status.upload_rate / 1024
                    torrent['num_seeds']     = torrent_status.num_seeds
                    torrent['total_seeds']   = torrent_status.num_complete
                    torrent['num_peers']     = torrent_status.num_peers
                    torrent['total_peers']   = torrent_status.num_incomplete
                    torrent['info_hash']     = str(torrent_handle.info_hash())

                    video_file = self._get_video_file_from_torrent(torrent_handle)
                    if video_file:
                        torrent['video_file']                          = {}
                        torrent['video_file']['path']                  = video_file.path
                        torrent['video_file']['size']                  = video_file.size
                        torrent['video_file']['start_piece_index']     = utils.piece_from_offset(torrent_handle, video_file.offset)
                        torrent['video_file']['end_piece_index']       = utils.piece_from_offset(torrent_handle, video_file.offset + video_file.size)
                        torrent['video_file']['total_pieces']          = utils.get_video_file_total_pieces(torrent_handle, video_file)
                        torrent['video_file']['preload_buffer_pieces'] = utils.get_preload_buffer_piece_count(torrent_handle, video_file)
                        torrent['video_file']['is_ready_fast']         = self.is_video_file_ready(torrent_handle, True, False)
                        torrent['video_file']['is_ready_slow']         = self.is_video_file_ready(torrent_handle, False, False)
                        torrent['video_file']['complete_pieces']       = utils.get_video_file_complete_pieces(torrent_handle, video_file)

                        piece_map = ''
                        for piece_index in range(torrent['video_file']['start_piece_index'], torrent['video_file']['end_piece_index'] + 1):
                            if torrent_handle.have_piece(piece_index):
                                piece_map = piece_map + '*'
                            elif torrent_handle.piece_priority(piece_index) == 0:
                                piece_map = piece_map + '0'
                            elif torrent_handle.piece_priority(piece_index) == 7:
                                piece_map = piece_map + '7'
                            else:
                                piece_map = piece_map + '.'
                        torrent['video_file']['piece_map'] = piece_map
                except:
                    pass

                result['session']['torrents'].append(torrent)

        return result