DEV Community

Cover image for Clean Code and Good Practices in Python
Leon Martin
Leon Martin

Posted on

Clean Code and Good Practices in Python

Writing clean, maintainable Python code is an essential skill for any developer. Clean code not only makes your work more readable and efficient but also ensures your projects can be easily understood and maintained by others. In this article, we’ll explore key principles and good practices for writing clean Python code.


1. Follow PEP 8 Style Guidelines

PEP 8 is the official style guide for Python and provides conventions for writing readable and consistent code. Tools like pylint and flake8 can help ensure your code adheres to these standards.

Key PEP 8 Rules:

  • Use 4 spaces for indentation.
  • Limit lines to 79 characters.
  • Use meaningful names for variables and functions.

Example:

# Good
def calculate_total_price(price, tax_rate):
    return price + (price * tax_rate)
Enter fullscreen mode Exit fullscreen mode

2. Write Descriptive and Meaningful Names

Names should clearly describe the purpose of variables, functions, and classes. Avoid using single letters or vague terms.

❌ Bad:

def func(x, y):
    return x + y
Enter fullscreen mode Exit fullscreen mode

✅ Good:

def add_numbers(number1, number2):
    return number1 + number2
Enter fullscreen mode Exit fullscreen mode

Guidelines:

  • Use snake_case for variable and function names.
  • Use PascalCase for class names.

3. Keep Functions and Classes Small

Functions should do one thing and do it well. Similarly, classes should adhere to the Single Responsibility Principle (SRP).

❌ Bad:

def process_user_data(user):
    # Validating user
    if not user.get('name') or not user.get('email'):
        return "Invalid user"

    # Sending email
    print(f"Sending email to {user['email']}")
    return "Success"
Enter fullscreen mode Exit fullscreen mode

✅ Good:

def validate_user(user):
    return bool(user.get('name') and user.get('email'))

def send_email(email):
    print(f"Sending email to {email}")

def process_user_data(user):
    if validate_user(user):
        send_email(user['email'])
        return "Success"
    return "Invalid user"
Enter fullscreen mode Exit fullscreen mode

4. Use Constants for Magic Numbers and Strings

Avoid using hardcoded values directly in your code. Define them as constants for better readability and maintainability.

❌ Bad:

if order_total > 100:
    discount = 10
Enter fullscreen mode Exit fullscreen mode

✅ Good:

MINIMUM_DISCOUNT_THRESHOLD = 100
DISCOUNT_PERCENTAGE = 10

if order_total > MINIMUM_DISCOUNT_THRESHOLD:
    discount = DISCOUNT_PERCENTAGE
Enter fullscreen mode Exit fullscreen mode

5. Use List Comprehensions for Simple Transformations

List comprehensions make your code more concise and Pythonic. However, avoid overcomplicating them.

❌ Bad:

squared_numbers = []
for number in range(10):
    squared_numbers.append(number ** 2)
Enter fullscreen mode Exit fullscreen mode

✅ Good:

squared_numbers = [number ** 2 for number in range(10)]
Enter fullscreen mode Exit fullscreen mode

6. Avoid Mutable Default Arguments

Using mutable objects like lists or dictionaries as default arguments can lead to unexpected behavior.

❌ Bad:

def append_to_list(value, items=[]):
    items.append(value)
    return items
Enter fullscreen mode Exit fullscreen mode

✅ Good:

def append_to_list(value, items=None):
    if items is None:
        items = []
    items.append(value)
    return items
Enter fullscreen mode Exit fullscreen mode

7. Handle Exceptions Gracefully

Python encourages using exceptions for error handling. Use try...except blocks to handle errors and provide meaningful messages.

Example:

try:
    result = int(input("Enter a number: ")) / 0
except ZeroDivisionError:
    print("Cannot divide by zero!")
except ValueError:
    print("Invalid input. Please enter a valid number.")
Enter fullscreen mode Exit fullscreen mode

8. Write DRY (Don’t Repeat Yourself) Code

Avoid duplicating logic in your code. Extract common functionality into reusable functions or classes.

❌ Bad:

def calculate_area_circle(radius):
    return 3.14 * radius ** 2

def calculate_area_rectangle(length, width):
    return length * width
Enter fullscreen mode Exit fullscreen mode

✅ Good:

import math

def calculate_area(shape, **dimensions):
    if shape == "circle":
        return math.pi * dimensions["radius"] ** 2
    elif shape == "rectangle":
        return dimensions["length"] * dimensions["width"]
Enter fullscreen mode Exit fullscreen mode

9. Use Docstrings and Comments

Document your code with meaningful docstrings and comments to explain the "why" behind complex logic.

Example:

def calculate_area(radius):
    """
    Calculate the area of a circle given its radius.

    Args:
        radius (float): The radius of the circle.

    Returns:
        float: The area of the circle.
    """
    return 3.14 * radius ** 2
Enter fullscreen mode Exit fullscreen mode

10. Use Type Hints

Type hints make your code more readable and help tools like mypy catch errors early.

Example:

def add_numbers(a: int, b: int) -> int:
    return a + b
Enter fullscreen mode Exit fullscreen mode

11. Test Your Code

Always write tests to ensure your code works as expected. Use frameworks like unittest or pytest.

Example:

import unittest

def add(a, b):
    return a + b

class TestMathOperations(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(2, 3), 5)

if __name__ == "__main__":
    unittest.main()
Enter fullscreen mode Exit fullscreen mode

12. Use Virtual Environments

Isolate your project dependencies to avoid conflicts using virtual environments.

Commands:

# Create a virtual environment
python -m venv env

# Activate the virtual environment
source env/bin/activate  # macOS/Linux
env\\Scripts\\activate   # Windows
Enter fullscreen mode Exit fullscreen mode

Last Words

Clean code is more than just a set of rules—it’s a mindset. By adopting these good practices, you’ll write Python code that is readable, maintainable, and professional. Remember, clean code benefits not only you but everyone who works with your code.

What’s your favorite clean code practice in Python? Please share your tips in the comments below!

Top comments (0)