将 Python 中的字典解析为我当前的 table

Question

我有一个 table，其中包含几个类别，其中两个是：mac 地址和设备名称。我有一个用我的代码（硬编码）编写的 mac 地址列表及其相应的设备名称（即 deviceDict['00:00:00:00:00:00']= name）

现在，我将那些 mac 地址和设备名称传递到一个文本文件，以便从相同的 Python 代码中读取并将其解析到我的 table 中。该代码当前可以识别文本文件，但不会将该信息解析到 table.

代码如下：

# File: WapLogParser.py
# Desc: Parses a WAP log file and pulls out information relating to connected clients
# Usage: python WapLogParser.py [file glob]

import re
import sys
import glob
import os

deviceDict = dict()

# Base table for storing client info
# All names must match what is in the Wap Log file
#   Exceptions: Date, Wap Name, Device Name - which are provided outside of the result parsing
table = [["Ssid", "Vlan", "Mac Address", "Connected Time", "Ip Address", "Rssi", "Date", "Wap Name", "Device Name"]]

def ParseResult(result, date, wapName):
    lines = result.split('\n')
    lines = list(filter(None, lines))
    # Any useful info will be at least 2 lines long
    if len(lines) == 1:
        return
    # create empty row
    data = [""] * len(table[0])
    # for each item in the result place it in the correct spot in the row 
    for line in lines:
        if line != "":
            # Parse the key/value pair
            m = re.match(r"(.*):\s\.*\s?(.*)", line)
            if m is not None:
                for idx in range(len(table[0])):
                    if table[0][idx].lower() == m[1].lower():
                         data[idx] = m[2]
        else:
            break

    # Remove the '(dBm)' from the RSSI value
    data[5] = data[5].split()[0]

    # Append WAP specific items to row
    data[6] = date
    data[7] = wapName
    data[8] = GetDeviceName(data[2].upper())

    # Add row to table
    table.append(data)


def ParseFile(path):
    with open(path) as f:
        lines = f.readlines()
        result = ""
        command = ""
        date = ""
        # WAP name is always on the first line 16 characters in with 4 
        # unnecessary characters trailing
        wapName = lines[0].strip()[16:-4]
        for line in lines:
            line = line.strip()
            # Is an issued command?
            if line.startswith("/#"):
                if command != "":
                    ParseResult(result, date, wapName) 
                command = "" 
                # reset the result for the new command
                result = ""
                m = re.match(r"^/#.*show\sclient.*stats$", line)
                if m is not None:
                    command = line
            # Anything that is not a command add to the result
            else:
                result += line + "\n"

            # Do we have the date?
            if line.startswith("Current date:"):
                date = line.replace("Current date: ", "")

# Print output to stderr
def eprint(*args, **kwargs):
    print(*args, file=sys.stderr, **kwargs)

# Print a 2d array in a csv format
def PrintAsCsv(table):
    for row in table:
        print(",".join(row))


def Main():
    InitDeviceDict()
    numArgs = len(sys.argv)
    for filename in glob.iglob(sys.argv[numArgs - 1], recursive=True):
        # Globs get directories too
        if os.path.isfile(filename):
            eprint("Parsing " + filename)
            try:
                ParseFile(filename)
            except Exception as e: # Mainly for if we see a binary file
                eprint("Bad file: " + e)

    # Print in a format we can use
    PrintAsCsv(table)


def GetDeviceName(macAddress):
    if macAddress in deviceDict:
        return deviceDict[macAddress]

    manufacturerPart = macAddress[:8]
    if manufacturerPart in deviceDict:
        return deviceDict[manufacturerPart]

    return 'Unknown Device'


def InitDeviceDict():
    with open('try.txt','r') as fo:
        for line in fo:
           deviceDict = {}
           line = line.split(',')
           macAddress = line[0].strip()
           manufacturerPart = line[1].strip()
           if macAddress in deviceDict:
               deviceDict[macAddress].append(manufacturerPart)
           else:
               deviceDict[macAddress]=(manufacturerPart)


           print(deviceDict)

# entry point
# script arguments:
#   WapLogParser.py [file glob]
if __name__ == "__main__":
    Main()

问题出在函数 GetDeviceName 和 InitDeviceDict 上。当我运行代码然后是一个批处理文件来显示我在 excel 上的信息时，我不断收到 "unknown device" （好像它没有识别我输入的 mac 地址产生设备名称）

有什么办法可以纠正这个问题吗？谢谢

Answer 1

InitDeviceDict 中填充的 deviceDict 不是全局 deviceDict。您只是在修改一个函数本地字典（并且也在每一行重置它）。从该函数中删除 deviceDict = {}，并在函数顶部使用 global deviceDict 声明您正在修改全局。

def InitDeviceDict():
    global deviceDict
    with open('try.txt','r') as fo:
        for line in fo:
           line = line.split(',')
           macAddress = line[0].strip()
           manufacturerPart = line[1].strip()
           if macAddress in deviceDict:
               deviceDict[macAddress].append(manufacturerPart)
           else:
               deviceDict[macAddress]=[manufacturerPart]

将 Python 中的字典解析为我当前的 table

Parsing a dictionary in Python to my current table

python

text

hashtable

hashcode