如何在Python中自动格式化代码、添加空格并修复语法错误以执行exec？

Great questions! Let's break them down one by one to give you clear, practical answers.

This all boils down to following PEP8, Python's official style guide, which ensures code is consistent and easy to read for everyone. Here are the key whitespace rules you should stick to:

• Around binary operators: Add spaces before and after operators like +, -, *, /, =, ==, etc. This makes equations and assignments easier to parse at a glance.

  • Bad: a=3+5 or if x>10:
  • Good: a = 3 + 5 or if x > 10:
  • Exception: Don't add spaces around = in keyword arguments (e.g., func(name="Bob")) or function default values (e.g., def greet(name="Guest"):).

• After commas and colons: Always add a space after commas (in lists, tuples, function arguments) and after colons (in if/for/while blocks, dictionary key-value pairs).

  • Bad: my_list=[1,2,3] or my_dict={"key":"value"}
  • Good: my_list = [1, 2, 3] or my_dict = {"key": "value"}
  • Exception: Skip spaces around colons in slicing (e.g., my_list[1:4], not my_list[1 : 4]).

• Indentation: Use 4 spaces per indent level (never tabs) to separate code blocks like if bodies, function definitions, or loops.

  • Bad:

    if True:
    print("Hello")
    

  • Good:

    if True:
        print("Hello")
    

Let's split this into two distinct scenarios: formatting valid code, and trying to fix broken code so exec() can run it.

Formatting Valid Python Code #

Absolutely! You can use popular formatting tools directly within Python via their APIs—no need to call external commands. Here are two reliable options:

Using black #

black is an opinionated formatter that enforces a consistent style with zero configuration. It's super easy to use:

import black

messy_code = "def calculate(x,y):return x*y+1"
formatted_code = black.format_str(messy_code, mode=black.Mode())
print(formatted_code)
# Output:
# def calculate(x, y):
#     return x * y + 1

Using autopep8 #

autopep8 adjusts code to match PEP8 standards, and it's more flexible if you want to tweak formatting rules:

import autopep8

messy_code = "a=10; print('Result:',a*2)"
formatted_code = autopep8.fix_code(messy_code)
print(formatted_code)
# Output: a = 10; print('Result:', a * 2)

Both tools work great for fixing whitespace, line breaks, and other formatting issues—but they only work if your code is syntactically valid (no missing colons, unclosed brackets, etc.).

Fixing Syntax Errors for exec() #

This is where things get tricky. If your string x contains code with syntax errors (like missing colons, mismatched parentheses, or invalid indentation), there's no reliable way to automatically fix it so exec() can run it. Here's why:

• Ambiguity: Syntax errors often have multiple possible fixes. For example, if True print("hi") could mean if True: print("hi") (missing colon) or even something else entirely—but a tool can't read your mind to know which you intended.
• Formatters don't fix errors: Tools like black and autopep8 will throw an error if they encounter invalid syntax, because they're designed to format valid code, not repair broken code.

There are experimental tools that attempt simple syntax fixes, but they're not trustworthy—they might introduce new errors or produce code that doesn't do what you want. The only safe way to make exec(x) work is to ensure the code in x is syntactically correct in the first place, or manually fix the errors before running it.

内容的提问来源于stack exchange，提问作者Jon