Понимание точки входа в Python: условие __name__ == '__main__'

Расширенные примеры использования if __name__ == '__main__'

Пример 1. Пакет с точкой входа через __main__.py

Когда проект оформлен как пакет (директория с __init__.py), удобно создать файл __main__.py, который запускается командой python -m package_name. В этом файле используется та же конструкция:

# структура проекта:
# mypackage/
#   __init__.py
#   __main__.py
#   core.py

# __main__.py
from mypackage.core import run

if __name__ == '__main__':
    run()

$ python -m mypackage
 # Результат: выполнение функции run()

Пример 2. Модульное тестирование с doctest

Конструкция позволяет встраивать тесты прямо в документацию и запускать их только при прямом запуске модуля:

# calculator.py
def multiply(a, b):
    """
    Возвращает произведение a и b.

    >>> multiply(3, 4)
    12
    >>> multiply(-1, 5)
    -5
    """
    return a * b

def divide(a, b):
    """
    Возвращает частное a и b.

    >>> divide(10, 2)
    5.0
    >>> divide(7, 0)
    Traceback (most recent call last):
        ...
    ZeroDivisionError: division by zero
    """
    if b == 0:
        raise ZeroDivisionError("division by zero")
    return a / b

if __name__ == '__main__':
    import doctest
    doctest.testmod(verbose=True)

$ python calculator.py
Trying:
    multiply(3, 4)
Expecting:
    12
ok
Trying:
    multiply(-1, 5)
Expecting:
    -5
ok
...
1 items passed all tests:
   2 tests in __main__.multiply
...
Test passed.

Пример 3. Асинхронный запуск с asyncio

При написании асинхронных скриптов точка входа должна содержать вызов asyncio.run():

# async_demo.py
import asyncio

async def fetch_data(url):
    # имитация загрузки
    await asyncio.sleep(1)
    return f"Данные из {url}"

async def main():
    urls = ["http://example.com", "http://python.org"]
    tasks = [fetch_data(url) for url in urls]
    results = await asyncio.gather(*tasks)
    print("Результаты:", results)

if __name__ == '__main__':
    asyncio.run(main())

$ python async_demo.py
Результаты: ['Данные из http://example.com', 'Данные из http://python.org']

Пример 4. Использование с флагами командной строки (argparse)

Современный способ создания CLI-утилит с помощью модуля argparse:

# cli_tool.py
import argparse

def process_file(filename, uppercase=False):
    with open(filename, 'r') as f:
        content = f.read()
    return content.upper() if uppercase else content

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="Обработчик файлов")
    parser.add_argument('filename', help="Имя файла для обработки")
    parser.add_argument('--uppercase', '-u', action='store_true',
                        help="Преобразовать в верхний регистр")
    args = parser.parse_args()

    result = process_file(args.filename, args.uppercase)
    print(result)

$ python cli_tool.py test.txt --uppercase
# выведет содержимое test.txt в верхнем регистре

Пример 5. Профилирование по запросу

Иногда полезно включать профилирование только при прямом запуске, сохраняя чистоту при импорте:

# optimized.py
import cProfile

def compute_heavy():
    # ресурсоёмкая операция
    total = 0
    for i in range(1000000):
        total += i ** 2
    return total

if __name__ == '__main__':
    cProfile.run('compute_heavy()', sort='cumulative')

$ python optimized.py
         4 function calls in 0.123 seconds

   Ordered by: cumulative time

   ncalls  tottime  percall  cumtime  percall filename:lineno(function)
        1    0.123    0.123    0.123    0.123 optimized.py:8(compute_heavy)
        1    0.000    0.000    0.123    0.123 {built-in method builtins.exec}
        1    0.000    0.000    0.000    0.000 {method 'disable' ...}

Пример 6. Разделение тестов и рабочего кода в одном файле

Для небольших проектов можно совмещать реализацию и тесты в одном модуле, защищая тесты условием:

# validator.py
def validate_email(email):
    import re
    pattern = r'^[\w\.-]+@[\w\.-]+\.\w+$'
    return bool(re.match(pattern, email))

# Тесты, выполняемые только при прямом запуске
if __name__ == '__main__':
    # Простые тесты без внешних фреймворков
    tests = [
        ("user@example.com", True),
        ("user@domain", False),
        ("@example.com", False),
        ("", False),
    ]
    for email, expected in tests:
        result = validate_email(email)
        status = "✅" if result == expected else "❌"
        print(f"{status} validate_email('{email}') = {result} (expected {expected})")

$ python validator.py
✅ validate_email('user@example.com') = True (expected True)
✅ validate_email('user@domain') = False (expected False)
✅ validate_email('@example.com') = False (expected False)
✅ validate_email('') = False (expected False)