如何使用importlib重寫字節碼？

Question

我正在尋找一種在Python 2.x中使用importlib的方式來即時重寫導入模塊的字節碼。換句話說，我需要在導入期間的編譯和執行步驟之間掛鉤我自己的函數。除此之外，我希望導入功能與內置功能一樣工作。

我已經做了imputil，但該圖書館並沒有涵蓋所有的情況，反正是不推薦使用。

Answer 1

有過通過importlib源代碼一看，我相信你可以在_bootstrap模塊中繼承PyLoader和覆蓋get_code：

class PyLoader: 
    ... 

    def get_code(self, fullname): 
    """Get a code object from source.""" 
    source_path = self.source_path(fullname) 
    if source_path is None: 
     message = "a source path must exist to load {0}".format(fullname) 
     raise ImportError(message) 
    source = self.get_data(source_path) 
    # Convert to universal newlines. 
    line_endings = b'\n' 
    for index, c in enumerate(source): 
     if c == ord(b'\n'): 
      break 
     elif c == ord(b'\r'): 
      line_endings = b'\r' 
      try: 
       if source[index+1] == ord(b'\n'): 
        line_endings += b'\n' 
      except IndexError: 
       pass 
      break 
    if line_endings != b'\n': 
     source = source.replace(line_endings, b'\n') 

    # modified here 
    code = compile(source, source_path, 'exec', dont_inherit=True) 
    return rewrite_code(code) 

我假設你知道你在做什麼，而是代表程序員無處不在，我相信我應該說：啊 = p