python程序中将常量放在同一个文件里并防止修改

Posted on

在看《编写高质量代码-改善Python程序的91个建议》一书，是国内Python圈和CPyUG里的名人赖总和人合著的，里面建议了很多比较好，比较Pythonic的代码写法，打算边看边实践，挑选一些记录下，这算是第一篇吧。

常量也就是那些一般不会变的数据，建议的做法如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
'''
Best practice 01

Put all constants in one file, and protect them from changing value.

'''
# File name: constants.py

class Const(object):
    class ConstError(TypeError):
        pass
    
    class ConstCaseError(ConstError):
        pass
    
    def __setattr__(self, name, value):
        if name in self.__dict__: # 判断是否已经被赋值，如果是则报错
            raise self.ConstError("Can't change const.%s" % name)
        if not name.isupper(): # 判断所赋值是否是全部大写，用来做第一次赋值的格式判断，也可以根据需要改成其他判断条件
            raise self.ConstCaseError('const name "%s" is not all supercase' % name)

        self.__dict__[name] = value

const = Const()
const.MY_CONSTANT = "CHINA"
const.MY_SECOND_CONSTANT = "RUSSIA"
1
2
3
4
#File name: test.py
from constants import const
print(const.MY_CONSTANT)
const.MY_CONSTANT = 2 #此处尝试再赋值会触发ConstError