介绍Python设置属性值方法setattr()和判断变量是否存在的方法

以属性名为变量的方式给一个对象添加属性

使用 setattr 方法

obj = SomeClass()
key = 'key_name'
val = 'key_value'
setattr(obj, key, val)
print(obj.key_name)

示例：给报文字段赋值

from scapy.all import *
layer = {"ether":"dst='33:33:00:01:00:02',src='00:0c:29:d9:98:c7'"}
Ethernet_field = layer.setdefault('ether')
ethernet = Ether(type=0x86dd)
fields = dict(((lambda a:(a[0].strip("'"),a[1].strip("'"))) (field.split('=')) for field in Ethernet_field.split(',')))
for key, val in fields.items():
    setattr(ethernet, key, val) 
ethernet.show()

out:

>>> ethernet.show()
###[ Ethernet ]###
  dst       = 33:33:00:01:00:02
  src       = 00:0c:29:d9:98:c7
  type      = IPv6

判断变量是否存在

三种方法：

locals().keys()
dir()
vars()

ethernet = 'Ether'
dhcpv6 = 'dhcpv6_solicit'
message = ['ethernet','ip','udp','dhcpv6']
del_message = []
for layer_i, value in enumerate(message):             
    if message[layer_i] not in locals().keys():
        del_message.append(value)
for m in del_message:
    message.remove(m)
print(f'message:{message}')

运行结果：

1 2	C:\Users\DELL>python3 test_locals.py message:['ethernet', 'dhcpv6']

--THE END--
Python setattr

本文标题:Python笔记：属性值设置和判断变量是否存在

文章作者:hiyo

文章链接:https://hiyongz.github.io/posts/python-notes-for-variable-and-attribute/

许可协议:本博客文章除特别声明外，均采用CC BY-NC-ND 4.0 许可协议。转载请保留原文链接及作者。