如何让你的Python代码更漂亮？yapf ——你的代码小助手

最新推荐文章于 2024-09-29 19:43:01 发布

原创最新推荐文章于 2024-09-29 19:43:01 发布 · 1k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

Python 专栏收录该内容

717 篇文章

订阅专栏

Yapf是一个Python代码格式化工具，能按照PEP8和Google代码风格对代码进行格式化。它不仅支持命令行使用，还可以作为模块在项目中集成，方便对用户提交的代码进行格式化处理。通过Yapf，你可以让代码更加整洁和规范。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

大部分人使用Python中的代码格式化工具时都会选择autopep8之类的工具，这些工具可能有一定的局限性，比如不会重新格式化已经符合 PEP8 的代码。

而本文即将介绍的 Yapf(https://github.com/google/yapf)，不仅能将代码格式化为符合 PEP8 代码指南的格式，还能格式化为符合 Google 代码指南的格式，可选项更多，让你的代码更加漂亮。

1.准备

开始之前，你要确保Python和pip已经成功安装在电脑上。

(可选1) 如果你用Python的目的是数据分析，可以直接安装Anaconda。

(可选2) 此外，推荐大家用VSCode编辑器，它有许多的优点。

请选择以下任一种方式输入命令安装依赖：

Windows 环境打开 Cmd (开始-运行-CMD)。
MacOS 环境打开 Terminal (command+空格输入Terminal)。
如果你用的是 VSCode编辑器或 Pycharm，可以直接使用界面下方的Terminal.

pip install yapf

YAPF 支持 Python 2.7 和 3.6.4+。

2.Yapf 使用方法

你可以在命令行中直接使用 Yapf, 参数如下：

yapf [-h] [-v] [-d | -i | -q] [-r | -l START-END] [-e PATTERN]
            [--style STYLE] [--style-help] [--no-local-style] [-p]
            [-vv]
            [files ...]

必选参数:

files 需要被格式化的文件

可选参数：

-h, --help 显示帮助信息
-v, --version 显示程序的版本号
-d, --diff 打印格式化前后差异
-i, --in-place 格式化文件并替换
-q, --quiet 什么都不输出
-r, --recursive 在目录上递归运行
-l START-END, --lines START-END 重新格式化的行范围，基于一个
-e PATTERN, --exclude PATTERN 需要排除的文件
–style STYLE 指定格式化样式（例如“pep8”或“google”）
-p, --parallel 并行格式化（多个文件时有效）
-vv, --verbose 在处理时打印出文件名

举个例子，如果我们有以下的混乱格式代码 1.py ：

x = { 'a':37,'b':42,
# pyhton一百本电子书、视频教程等等我都准备好了，直接在文末名片自取即可
'c':927}

y = 'hello ''world'
z = 'hello '+'world'
a = 'hello {}'.format('world')
class foo  ( object ):
  def f    (self ):
    return       37*-+2
  def g(self, x,y=42):
      return y
def f  ( a ) :
  return      37+-+a[42-x : y**3]

为了格式化它，我们可以在命令行中执行以下命令：

yapf -i 1.py

效果如下：

3.Yapf 作为模块使用

Yapf 不仅可以在命令行中使用，我们还可以在项目运行时使用。

一个比较常见的场景是用户在你的web网站上提交了python代码，你希望能一件格式化这些代码，那将yapf作为格式化的模块则再好不过。

Yapf 最主要的两个 API 是 FormatCode 和 FormatFile:

from yapf.yapflib.yapf_api import FormatCode

formatted_code, changed = FormatCode("f ( a = 1, b = 2 )")
print(formatted_code)
# 'f(a=1, b=2)\n'
print(changed)
# True

你还可以传递 style_config 决定使用哪种指南做格式化：

from yapf.yapflib.yapf_api import FormatCode

print(FormatCode("def g():\n return True", style_config='pep8')[0])
# 'def g():\n return True\n'

FormatFile，这个函数可以从文件中读取代码并格式化：

from yapf.yapflib.yapf_api import FormatFile
print(open("foo.py").read())
# a==b
reformatted_code, encoding, changed = FormatFile("foo.py")
print(formatted_code)
# 'a == b\n'
print(encoding)
# 'utf-8'
print(changed)
# True

如果你添加in_place参数，它将把格式化后的代码写回文件中：

from yapf.yapflib.yapf_api import FormatFile
print(FormatFile("foo.py", in_place=True)[:2])
# (None, 'utf-8')

print(open("foo.py").read())
# a == b