python中from module import * 的一個陷阱

lzprgmr發表於2014-07-20

from module import *把module中的成員全部導到了當前的global namespace,訪問起來就比較方便了。當然,python style一般不建議這麼做,因為可能引起name conflict。

但還有另外一個問題 - 你以為你修改了某個變數,其實,被from module import *後的那個並沒有被更新,非常危險,因為程式有可能還可以正常執行, 只不過結果錯了,到了production才被發現就比較慘了。

舉個例子:

你定義了一些變數在base模組中:

# reference data type
class Demo:
        def __init__(self, name):
                self.name = name
demo = Demo('Demo')

# primitive type
foo = 1

 

然後在一個模組中用from  module import 的方式讀它:

from base import *
def read():
        print 'reference data id: ' + str(id(demo))
        print 'reference data value : ' + demo.name

        print 'primitive data id: ' + str(id(foo))
        print 'primitive data value: ' + str(foo)

在另外一個模組中寫它:

import base

def write():
        print "\nOriginal:"
        print "Original reference data id: " + str(id(base.demo))
        base.demo.name = "Updated Demo" # this will reflect that change
        #base.demo = base.Demo("Updated Demo") # this won't relfect the change
        print "Original data id: " + str(id(base.foo))
        base.foo = 1000
        print "Original data id after assignment: " + str(id(base.foo))

 

然後先寫,後讀,看寫的內容是否有效:

import read
import write

print "before write"
read.read()

write.write()

print "\nafter write"
read.read()

 

結論是沒有,原因是:

  • 當你用from module import時,其實是copy了一份reference或者pointer,指向一份記憶體,var和module.var都指向同一份記憶體
  • 當你修改module.var時,其實你是讓它指向了另外一份記憶體,此時var和module.var指向的是不同的記憶體
  • 所以,雖然module.var的值變了,var還是指向原來那份記憶體,原來的值

這個對於object,比較容易理解,你可以直接修改object裡的值,這個是有效的,但是當你指向另外一個object時就無效了。 對於primitive型別來講,其實也是一個道理,因為每次賦值,都是讓其指向一個不同的記憶體地址,而不是inplace修改已有的那份記憶體 -  這個很容易驗證:

In [1]: a = 10

In [2]: id(a)
Out[2]: 20429204

In [3]: a = 100

In [4]: id(a)
Out[4]: 20430108

 

所以,建議是除非是一個quick and dirty的指令碼,否則不要使用from module import *!

 

例子:https://github.com/baiyanhuang/blog/tree/master/arena/python/from_module_import

相關文章