python 下Warning的使用

起因是这样的,当我使用pymysql模块执行建表的sql语句时获,在控制台输出了红色的消息,但是程序并没有终止而是继续运行了

sql语句如下:

CREATE TABLE IF NOT EXISTS test(age int);

整体代码:

import pymysql

conn = pymysql.connect(

user= "root",

password="",

database="test"

)

c = conn.cursor()

c.execute("create table if not exists abc(age int)")

得到的警告信息如下:

"""

NoneType: None

/Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packages/pymysql/cursors.py:170: Warning: (1050, "Table 'abc' already exists")

result = self._query(query)

"""

Process finished with exit code 0

sql语句是如果表存在则放弃创建,这意味着这个警告是应该的,并没有逻辑上的错误,但是这个红色的消息,让人看着不爽,于是找到了如下解决方案:

from warnings import filterwarnings

filterwarnings("ignore",category=pymysql.Warning)

问题得到解决,作为强迫症患者,必须在深入研究一下,得出以下结论:

warning模块中主要的两个方法

1.warn

该方法用于输出一个警告信息

#参数一: message 表示警告的详细信息

#参数二: category 类别,需要一个类作为参数,该类必须是Warning的子类,Warning类是Exception的子类,用于给警告设置一个具体的类型

#参数三: stacklevel 指定调用栈的层级,用于确定要从哪一级获取行号

#参数四: 没研究

stacklevel在源码中的调用:

上述代码获得一个frame,用于获取行号

由以上源码可以get到一个小技能即获取行号

python获取行号:

import sys

f = sys._getframe() # f 是一个对象class为 frame 但是该类无法直接访问,猜想是系统底层隐藏的毕竟与解释器核心数据相关

print(f.f_lineno) # f_lineno就是用于获取行号的属性

"""

在源码中可以看到_getframe函数是可以给一个整型参数的,

经过测试,得出结论,该参数用于,指定从栈的哪一个位置获取行号,

"""

#测试代码:

1 import sys

2 def task():

3 f = sys._getframe(2)

4 print(f.f_lineno)

5

6 def task2():

7 task()

8

9 task2()

"""

0表示从当前位置,即执行_getframe的地方

1表示获取上一级调用位置的行号 即第7行

2表示上上级 即第9行 以此类推...

"""

2.filterwarnings

该方法用于过滤警告信息

#参数action: 表示如何处理这个警告

可选的值 "error", "ignore", "always", "default", "module","once"

error会将警告作为异常抛出

ignore表示忽略该类警告不会打印

once表示该相同类型相同消息的警告仅打印一次

#参数category: 需要一个类,表示是要过滤的警告类别

#参数module: 需要一个模块,表示是要过滤的模块名称

#参数lineno:一个整型,表示要过滤哪一行的警告 0表示所有行 其他值貌似没有作用

意外发现

在查看源码是,发现了一个用于表示警告消息的类

该类非常简单,就是保存了一些警告信息,然后覆盖了 __str__函数,用于自定义字符串形式

其中的初始化函数引起了我的注意,分析了一下,发现这个写法不错,可以简化初始化中的属性赋值操作于是做了一下实验:

# 通用的类初始化方法

class Person():

def __init__(self,a,b,c):

for k,v in locals().items():

if k != "self":

setattr(self,k,v)

p = Person(1,2,3)

print(p.a)

print(p.b)

该方法最大的好处在于,无论初始化方法中需要多少参数都可以一一赋值,对于初始化参数多的类非常友好;

其原理是通过locals()来获取函数中的局部变量的字典形式,然后遍历字典,排除self之后其他的都设置为self的属性,用形参的名字作为熟悉名,大大减少了一堆重复的赋值代码!

否则你的代码可能是这样的:

class Person():

def __init__(self,a,b,c,d):

self.a = a

self.b = b

self.c = c

self.d = d

p = Person(1,2,3,4)

print(p.a)

print(p.b)

print(p.c)

如果参数特比多的话,你懂的......