Python线程网页.py和pysnmp来创建一个web服务

2024-06-09 23:23:50 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图为我的家庭网络创建一个小型服务,从交换机读取路由器所在端口的SNMP值,进行一些计算并返回当前的平均带宽利用率。在

我让pysnmp运行得很好,每两秒钟读取一次snmp值。。。然后我决定整合网页.py创建一个非常简单的web服务,以千位为单位返回上传和下载的平均JSON。在

为此,我必须使用线程来保持SNMP轮询在后台进行。我是一个NOOOB,但是它不能正常工作——当我在我的浏览器中加载这个值时,它似乎因为某种原因而创建了SNMP进程的第二个线程,我真的不知道为什么。在

代码如下:

from pysnmp.entity.rfc3413.oneliner import cmdgen
from time import sleep
import threading
import json
import web

LOCK = threading.Lock()

class getBW(threading.Thread):

    def __init__(self):
        threading.Thread.__init__(self)
        self.data = dict(upload="0",download="0",)
        self.lock = LOCK
        self.start()

    def run(self):

        router_ip = "192.168.2.2"
        snmp_community = "public"
        upload_mib = "1.3.6.1.2.1.2.2.1.16.1"
        download_mib = "1.3.6.1.2.1.2.2.1.10.1"
        read_interval = 2
        up_octets = 0
        dn_octets = 0
        last_up_octets = 0
        last_dn_octets = 0
        up_change = 0
        dn_change = 0
        avg_up =0
        avg_dn =0
        up_hist = [0,0,0,0,0] 
        dn_hist = [0,0,0,0,0]
        i = 0
        cmdGen = cmdgen.CommandGenerator()

        while (1):
            errorIndication, errorStatus, errorIndex, varBinds = cmdGen.getCmd(
                cmdgen.CommunityData(snmp_community),
                cmdgen.UdpTransportTarget((router_ip, 161)),
                upload_mib,
                download_mib
            )

            # Check for errors and print out results
            if errorIndication:
                print(errorIndication)
            else:
                if errorStatus:
                    print('%s at %s' % (
                        errorStatus.prettyPrint(),
                        errorIndex and varBinds[int(errorIndex)-1] or '?'
                        )
                    )
                else:
                    for name, val in varBinds:
                        if str(name) == upload_mib:
                            if up_octets > 0: 
                                last_up_octets = up_octets
                                up_change = val-last_up_octets
                                # convert to kilobits/s
                                up_change = (((up_change*8)/1000)/read_interval)
                                up_hist[i] = up_change
                                avg_up=(sum(up_hist)/len(up_hist))
                            up_octets = val
                            #print ('%d kilobit per second upload, %d average' % (up_change,avg_up))

                        elif str(name) == download_mib:
                            if dn_octets > 0:
                                last_dn_octets = dn_octets
                                dn_change = val-last_dn_octets
                                # convert to kilobits/s
                                dn_change = (((dn_change*8)/1000)/read_interval)
                                dn_hist[i] = dn_change
                                avg_dn=(sum(dn_hist)/len(dn_hist))
                                i += 1
                                i = i % 5 # if I is 5 reset to 0
                            dn_octets = val
                            #print ('%d kilobit per second download, %d average' % (dn_change,avg_dn))

            #output values to dict
            self.lock.acquire()
            self.data = dict(upload=str(avg_up),download=str(avg_dn),)
            self.lock.release()
            print("UPLOAD: %d change, %d average; DOWNLOAD: %d change, %d average" % (up_change,avg_up,dn_change,avg_dn))
            print("sleeping for %i secs...\r\n" % read_interval)
            sleep(read_interval)

try:
    BW = getBW()
except:
    print "Error: unable to start thread"

urls = ("/.*", "hello")
app = web.application(urls, globals())

class hello:
    def __init__(self):
        self.lock = LOCK
    def GET(self):
        self.lock.acquire()
        thedata = BW.data
        self.lock.release()
        return json.dumps(thedata)

if __name__ == "__main__":
    #start web service
    app.run()

我真的很感激你的帮助。请记住,我是一个完全不懂线程的人,只是在学习Python。谢谢!在


Tags: importselflockifdownloadchangeoctetshist
1条回答
网友
1楼 · 发布于 2024-06-09 23:23:50

你正在使用网页.py的内置Web服务器。默认情况下,它启用“模块重新加载”,并且重载程序加载主模块两次。因为您是在导入时启动线程的,所以有两个线程在运行。在

一个简单的解决方法是将线程创建移动到__main__块,这样它只创建一次:

if __name__ == "__main__":
    BW = getBW()
    #start web service
    app.run()

相关问题 更多 >