PyBitmessage/src/network/stats.py

import time

from bmconfigparser import BMConfigParser
from network.connectionpool import BMConnectionPool
from inventory import PendingDownloadQueue, PendingUpload
import asyncore_pollchoose as asyncore
import shared
import throttle

lastReceivedTimestamp = time.time()
lastReceivedBytes = 0
currentReceivedSpeed = 0
lastSentTimestamp = time.time()
lastSentBytes = 0
currentSentSpeed = 0

def connectedHostsList():
    if BMConfigParser().get("network", "asyncore"):
        retval = []
        for i in BMConnectionPool().inboundConnections.values() + BMConnectionPool().outboundConnections.values():
            if not i.fullyEstablished:
                continue
            try:
                retval.append(i)
            except AttributeError:
                pass
        return retval
    return shared.connectedHostsList.items()

def sentBytes():
    if BMConfigParser().get("network", "asyncore"):
        return asyncore.sentBytes
    return throttle.SendThrottle().total

def uploadSpeed():
    global lastSentTimestamp, lastSentBytes, currentSentSpeed
    if BMConfigParser().get("network", "asyncore"):
        currentTimestamp = time.time()
        if int(lastSentTimestamp) < int(currentTimestamp):
            currentSentBytes = asyncore.sentBytes
            currentSentSpeed = int((currentSentBytes - lastSentBytes) / (currentTimestamp - lastSentTimestamp))
            lastSentBytes = currentSentBytes
            lastSentTimestamp = currentTimestamp
        return currentSentSpeed
    return throttle.sendThrottle().getSpeed()

def receivedBytes():
    if BMConfigParser().get("network", "asyncore"):
        return asyncore.receivedBytes
    return throttle.ReceiveThrottle().total

def downloadSpeed():
    global lastReceivedTimestamp, lastReceivedBytes, currentReceivedSpeed
    if BMConfigParser().get("network", "asyncore"):
        currentTimestamp = time.time()
        if int(lastReceivedTimestamp) < int(currentTimestamp):
            currentReceivedBytes = asyncore.receivedBytes
            currentReceivedSpeed = int((currentReceivedBytes - lastReceivedBytes) / (currentTimestamp - lastReceivedTimestamp))
            lastReceivedBytes = currentReceivedBytes
            lastReceivedTimestamp = currentTimestamp
        return currentReceivedSpeed
    return throttle.ReceiveThrottle().getSpeed()

def pendingDownload():
    if BMConfigParser().get("network", "asyncore"):
        tmp = {}
        for connection in BMConnectionPool().inboundConnections.values() + BMConnectionPool().outboundConnections.values():
            for k in connection.objectsNewToMe.keys():
                tmp[k] = True
        return len(tmp)
    return PendingDownloadQueue.totalSize()

def pendingUpload():
    if BMConfigParser().get("network", "asyncore"):
        return 0
        tmp = {}
        for connection in BMConnectionPool().inboundConnections.values() + BMConnectionPool().outboundConnections.values():
            for k in connection.objectsNewToThem.keys():
                tmp[k] = True
        return len(tmp)
    return PendingUpload().len()