使用Hatchling轻松实现Python项目打包_pip install hatchling

本文将会介绍如何使用hatchling轻松实现Python打包。

在日常工作中，将Python代码打包成第三方模块，并上传至托管网站，并不是一个高频的需求，但学会Python代码的打包，也是一项必备技能。

所谓Python打包，就是将我们自己写的Python代码打包成第三方模块，方便后续使用或用于开源分享，比如常用的requests, numpy等第三方模块。常见的Python打包工具有：

• setuptools
• hatchling
• Flit
• PDM

这也是PyPI官网打包教程中给出的打包工具。本文将会介绍其中的hatchling。

Hatchling是一个Python包管理工具，主要用于方便地管理依赖关系和环境隔离。它与Hatch一起使用，用于配置、版本化、指定依赖关系以及将 Python 包发布到PyPI上。Hatchling的主要功能包括：

• 配置项目：使用pyproject.toml文件配置项目
• 版本化：管理项目的版本化过程
• 指定依赖关系：为项目指定所需的依赖关系
• 发布包到 PyPI：将项目发布到 Python Package Index (PyPI) 上

Hatchling 的插件系统允许用户轻松扩展其功能。

项目介绍

本文将会以一个简单的小项目为例，介绍如何使用hatchling来实现Python打包。我们的小项目主要使用tiktoken模块来实现文本或文本列表的token数计算。

项目的整体代码如下：

.
├── README.md
├── dist
│   ├── token_counter-0.0.1-py3-none-any.whl
│   └── token_counter-0.0.1.tar.gz
├── pyproject.toml
├── src
│   └── token_counter
│       ├── __init__.py
│       └── token_count.py
└── tests
    ├── __init__.py
    ├── package_test.py
    └── test_token_count.py

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

我们的功能实现放在src/token_counter模块中的token_count.py脚本，代码如下：

# -*- coding: utf-8 -*-
# @place: Pudong, Shanghai
# @file: token_count.py
# @time: 2024/1/22 17:45
import tiktoken

from typing import List, Union

class TokenCounter(object):
    def __init__(self, model="gpt-3.5-turbo"):
        """
        :param model: name of model, type: string        """
        self.model = model

    def count(self, _input: Union[List, str]) -> Union[List[int], int]:
        """
        :param _input: user input, type: str or List[str]        :return: Return the number of tokens used by text, type int or List[int]        """
        try:
            encoding = tiktoken.encoding_for_model(self.model)
        except KeyError:
            print("Warning: model not found. Using cl100k_base encoding.")
            encoding = tiktoken.get_encoding("cl100k_base")

        if isinstance(_input, list):
            token_count_list = []
            for text in _input:
                token_count_list.append(len(encoding.encode(text)))
            return token_count_list
        elif isinstance(_input, str):
            return len(encoding.encode(_input))
        else:
            raise NotImplementedError(f"not support data type for {type(_input)}, please use str or List[str].")

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
27
28
29
30
31
32
33

该脚本主要使用tiktoken模块来实现输入文本或文本列表的token数的计算。

在tests模块下，使用单元测试(unittest模块)对代码进行测试，代码（tests/test_token_count.py）如下：

# -*- coding: utf-8 -*-
# @place: Pudong, Shanghai
# @file: test_token_count.py
# @time: 2024/1/22 17:53
import unittest

from src.token_counter.token_count import TokenCounter

class TestTokenCounter(unittest.TestCase):
    def setUp(self):
        self.token_cnt = TokenCounter()

    def test_case1(self):
        text = "who are you?"
        tokens_cnt = self.token_cnt.count(_input=text)
        self.assertEqual(tokens_cnt, 4)

    def test_case2(self):
        texts = ["who are you?", "How's it going on?"]
        tokens_cnt = self.token_cnt.count(_input=texts)
        self.assertEqual(tokens_cnt, [4, 6])

    def test_case3(self):
        with self.assertRaises(NotImplementedError) as cm:
            self.token_cnt.count(_input=23)
        the_exception = cm.exception
        self.assertEqual(the_exception.__str__(), "not support data type for <class 'int'>, please use str or List[str].")

if __name__ == '__main__':
    suite = unittest.TestSuite()
    suite.addTest(TestTokenCounter('test_case1'))
    suite.addTest(TestTokenCounter('test_case2'))
    suite.addTest(TestTokenCounter('test_case3'))
    run = unittest.TextTestRunner()
    run.run(suite)

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
27
28
29
30
31
32
33
34
35
36

单元测试并不影响项目打包，但为了项目的完整性，需要把测试过程加上。

项目打包

对于Python打包，还需要一个配置文件（pyproject.toml），配置如下：

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"
[project]
name = "token_counter"
version = "0.0.1"
dependencies = [
  "tiktoken >= 0.5.0",
]
authors = [
  { name="jclian", email="jclian91@126.com" },
]
description = "A package for token count using tiktoken"
readme = "README.md"
requires-python = ">=3.9"
classifiers = [
    "Programming Language :: Python :: 3",
    "License :: OSI Approved :: MIT License",
    "Operating System :: OS Independent",
]

[project.urls]
Homepage = "https://github.com/percent4/package_python_project"

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

在这个配置文件中，声明了build系统使用hatchling，以及常见的打包配置项，比如：项目名称，版本号，依赖关系等等，这里不再过多介绍，上述配置应该是清晰明了的。

接下来是打包过程，我们需要安装依赖，命令如下：

python3 -m pip install --upgrade build
pip3 install hatchling
python -m build

1
2
3
4

使用python -m build即可打包，此时在dist目录下就会出现两个打包好的文件，如下：

token_counter-0.0.1.tar.gz
token_counter-0.0.1-py3-none-any.whl

1
2
3

这正是我们在平常安装Python第三方模块时所用到的安装文件。有了这些安装包，我们还需要将它们上传至第三方的托管网站，比如PyPI。

一般使用twine工具来实现安装包上传。

Twine模块是用于与 PyPI交互的实用工具，用于发布和管理Python包。它提供了上传代码到PyPI的功能，可以简化包的上传过程，无需执行setup.py。安装Twine模块后，你可以使用它来上传你的Python包到PyPI。要使用Twine模块在PyPI 上发布你自己的包，你需要准备好你的安装包，并使用twine upload命令上传它。

这里不再介绍如何上传安装包，后续有机会再介绍。

实验

将刚才的项目代码打包后，在本地安装好token_counter模块，命令如下：

pip3 install dist/token_counter-0.0.1-py3-none-any.whl

1
2

安装完毕后，我们来测试该模块的使用，示例代码如下：

# -*- coding: utf-8 -*-
# @place: Pudong, Shanghai
# @file: package_test.py
# @time: 2024/1/22 21:08
from token_counter.token_count import TokenCounter

text = "who are you?"
print(TokenCounter().count(_input=text))

1
2
3
4
5
6
7
8
9

输出结果：

4

以上就是“使用Hatchling轻松实现Python项目打包”的全部内容，希望对你有所帮助。
​
​关于Python技术储备

学好 Python 不论是就业还是做副业赚钱都不错，但要学会 Python 还是要有一个学习规划。最后大家分享一份全套的 Python 学习资料，给那些想学习 Python 的小伙伴们一点帮助！

一、Python所有方向的学习路线

Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照上面的知识点去找对应的学习资源，保证自己学得较为全面。

二、Python必备开发工具

三、Python视频合集

观看零基础学习视频，看视频学习是最快捷也是最有效果的方式，跟着视频中老师的思路，从基础到深入，还是很容易入门的。

四、实战案例

光学理论是没用的，要学会跟着一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

五、Python练习题

检查学习结果。

六、面试资料

我们学习Python必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有阿里大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

最后祝大家天天进步！！

上面这份完整版的Python全套学习资料已经上传至CSDN官方，朋友如果需要可以直接微信扫描下方CSDN官方认证二维码免费领取【保证100%免费】。