快捷導(dǎo)航

通過字符串導(dǎo)入 Python 模塊的方法詳解

更新時間：2019年10月27日 09:52:43 作者：青南

這篇文章主要介紹了通過字符串導(dǎo)入 Python 模塊的方法詳解，本文通過實例結(jié)合，給大家介紹的非常詳細(xì)，具有一定的參考借鑒價值,需要的朋友可以參考下

我們平時導(dǎo)入第三方模塊的時候，一般使用的是 import 關(guān)鍵字，例如：

import scrapy
from scrapy.spider import Spider

但是如果各位同學(xué)看過 Scrapy 的 settings.py 文件，就會發(fā)現(xiàn)里面會通過字符串的方式來指定pipeline 和 middleware，例如：

DOWNLOADER_MIDDLEWARES = {
   'Test.middlewares.ExceptionRetryMiddleware': 545,
   'Test.middlewares.BOProxyMiddlewareV2': 543,
 }
 
 SPIDER_MIDDLEWARES = {
  'Test.middlewares.LoggingRequestMiddleware': 543,
 }

我們知道，這里的 Test.middlewares.ExceptionRetryMiddleware 實際上對應(yīng)了根目錄下面的 Test 文件夾里面的 middlewares.py 文件中的 ExceptionRetryMiddleware 類。那么 Scrapy 是如何根據(jù)這個字符串，導(dǎo)入這個類的呢？

在 Scrapy 源代碼中，我們可以找到相關(guān)的代碼：

def load_object(path):
  """Load an object given its absolute object path, and return it.
  object can be a class, function, variable or an instance.
  path ie: 'scrapy.downloadermiddlewares.redirect.RedirectMiddleware'
  """
  try:
    dot = path.rindex('.')
  except ValueError:
    raise ValueError("Error loading object '%s': not a full path" % path)
  module, name = path[:dot], path[dot+1:]
  mod = import_module(module)
  try:
    obj = getattr(mod, name)
  except AttributeError:
    raise NameError("Module '%s' doesn't define any object named '%s'" % (module, name))
  return obj

根據(jù)這段代碼，我們知道，它使用了 importlib 模塊的 import_module 函數(shù)：

首先根據(jù)字符串路徑最右側(cè)的 . 把字符串路徑分成兩個部分，例如： Test.middlewares.LoggingRequestMiddleware 分成 Test.middlewares 和 LoggingRequestMiddleware

使用 import_module 導(dǎo)入左邊的部分

從左邊部分通過 getattr 獲得具體的類

現(xiàn)在我們來測試一下。我們創(chuàng)建的測試文件結(jié)構(gòu)如下圖所示：