轻松导入和执行Python模块的3种技巧-轻识

引言

作为一名程序员，你肯定希望与团队其他用户共享你所创建的有用模块。尽管你的模块可能有用，但是如果其他人需要花费大量精力来访问模块中的有用功能，则其他人将不会使用它。

因此，如果想使用户易于使用你的模块，导入和运行模块的代码应该简短。本文将向你展示3种简化导入和执行Python模块的方法。

导入

场景

假设我们有一个名为 utils.py 的文件，其中包含所有重要的函数和类

def add_two(num: int):    return num + 2   
def multiply_by_two(num: int):    return num * 2 
a = 5 在另一个脚本中，我们使用 from utils import * 从 utils.py 中导入所有内容，除了变量 a 。那我们该怎么做呢？

解决方案

要做到这点很容易，只要加入__all__ = [“add_two”, “multiply_by_two”]即可。使用 import * 时，__all__ 中指定的函数、类和包将被导入。

__all__ = ["add_two", "multiply_by_two"]
def add_two(num: int):    return num + 2   
def multiply_by_two(num: int):    return num * 2 
a = 5 现在尝试在另一个脚本中使用 import *

from utils import *
num = 3 print(add_two(num))print(multiply_by_two(num))print(a)
56Traceback (most recent call last):  File "main.py", line 6, in <module>    print(a)NameError: name 'a' is not definedCool！这个错误告诉我们，只从 utils.py 中导入了 add_two 和 multiply_by_two ，而并未导入变量 a 。

合并多个文件

场景

想象一下我们目录中文件的结构如下：

.├── data_modules│   ├── load_data.py│   └── process_data.py└── main.pyload_data.py 如下所示：

class DataLoader:    def __init__(self, data_dir: str):        self.data_dir = data_dir
    def load_data(self):        print(f"Loading data from {self.data_dir}")process _ data.py 如下所示：

class DataProcessor:     def __init__(self, data_name: str):        self.data_name = data_name        def process_data(self):        print(f"Processing {self.data_name}")要使用来自两个不同文件的类，我们需要从每个文件导入每个类。

from data_modules.load_data import DataLoaderfrom data_modules.process_data import DataProcessor
data_loader = DataLoader('data/')data_loader.load_data()
data_processor = DataProcessor('data1')data_processor.process_data()这个方法很好，但是使用2个导入语句是很繁琐的。有没有一种方法，我们可以把两个导入语句变成一个导入语句，就像下面这样。

from data_modules import DataLoader, DataProcessor解决方案

在 data _ modules 目录下插入 _ init _. py 文件可以很容易地解决这个问题:

touch data_modules/__init__.py然后将前面提到的 import 语句插入到 _ init _. py 文件中:

from .load_data import DataLoaderfrom .process_data import DataProcessor我们在这里使用一个 "." ，因为 load_data.py 和 _init _. py 在同一个目录中。

现在，让我们尝试从 data_modules 导入 DataLoader 和 DataProcessor 。

from data_modules import DataLoader, DataProcessor
data_loader = DataLoader('data/')data_loader.load_data()
data_processor = DataProcessor('data1')data_processor.process_data()
Loading data from data/Processing data1太棒了! 成功了！

像脚本一样运行目录

场景

我们的目录是这样的:

.└── data_modules    ├── __init__.py    ├── load_data.py    ├── main.py    └── process_data.py当我们运行 main.py 的时候是这样的：

$ python data_modules/main.py我们可能希望简化用户或队友运行 main.py 代码的步骤，只需运行父目录:

$ python data_modules这比运行 data_modules/main.py 更好，因为它更短，用户不需要知道 data_modules 中有哪些文件。

我们怎么才能做到呢？

解决方案

这时，__main__.py 就派上用场了。只需将运行顶级目录时要运行的脚本更改为__main__.py  即可。在我们的例子中，我们需要将 main.py 变为__main__.py。

# Rename main.py to __main__.py$ mv data_modules/main.py data_modules/__main__.py我们的新目录将如下所示

.└── data_modules    ├── __init__.py    ├── load_data.py    ├── __main__.py    └── process_data.py现在让我们测试下

$ python data_modulesLoading data from data/Processing data1效果非常好！

总结

在本文中，我们了解了如何让你和那些使用你的软件包的人更容易使用它。希望可以帮到你~

·  END  ·

HAPPY LIFE