SOLID и паттерны проектирования в Python

# ✗ Каждый новый формат — правка функции
def export_bad(data, fmt):
    if fmt == "json":
        ...
    elif fmt == "csv":
        ...

# ✓ Новый формат — новая запись в реестре, существующий код не меняется
import json

EXPORTERS = {}

def exporter(fmt):
    def register(func):
        EXPORTERS[fmt] = func
        return func
    return register

@exporter("json")
def to_json(rows):
    return json.dumps(rows, ensure_ascii=False)

@exporter("csv")
def to_csv(rows):
    return "\n".join(",".join(map(str, r.values())) for r in rows)

def export(rows, fmt):
    return EXPORTERS[fmt](rows)

rows = [{"id": 1, "name": "Анна"}]
print(export(rows, "json"))
print(export(rows, "csv"))

class Rectangle:
    def __init__(self, w, h):
        self.w, self.h = w, h
    def set_width(self, w):
        self.w = w
    def area(self):
        return self.w * self.h

class Square(Rectangle):
    def __init__(self, side):
        super().__init__(side, side)
    def set_width(self, w):            # квадрат обязан менять обе стороны
        self.w = self.h = w

def stretch(rect: Rectangle):
    rect.set_width(10)
    return rect.area()                 # ожидаем 10 * исходная высота

print(stretch(Rectangle(2, 5)))        # 50 — как ожидалось
print(stretch(Square(5)))              # 100 — сюрприз: Square не заменяет Rectangle

from typing import Protocol

class UserRepository(Protocol):
    def get(self, user_id: int) -> dict: ...

class SqlUserRepository:               # «настоящая» реализация (в проде)
    def get(self, user_id):
        raise RuntimeError("нет базы данных в этом примере")

class InMemoryUserRepository:          # для тестов
    def __init__(self, users):
        self.users = users
    def get(self, user_id):
        return self.users[user_id]

class UserService:
    def __init__(self, repo: UserRepository):   # зависимость передаётся снаружи
        self.repo = repo

    def greeting(self, user_id):
        return f"Здравствуйте, {self.repo.get(user_id)['name']}!"

service = UserService(InMemoryUserRepository({1: {"name": "Анна"}}))
print(service.greeting(1))

class CsvReader:
    def read(self, path): return f"читаю CSV {path}"

class JsonReader:
    def read(self, path): return f"читаю JSON {path}"

READERS = {".csv": CsvReader, ".json": JsonReader}   # классы — обычные значения словаря

def reader_for(path):
    ext = path[path.rfind("."):]
    try:
        return READERS[ext]()
    except KeyError:
        raise ValueError(f"неизвестный формат {ext}") from None

for p in ["data.csv", "conf.json"]:
    print(reader_for(p).read(p))

class OldPaymentApi:                     # сторонняя библиотека, менять нельзя
    def make_payment(self, kopecks, currency_code):
        return f"оплачено {kopecks / 100:.2f} {currency_code}"

class PaymentAdapter:                    # наш код ожидает pay(rub)
    def __init__(self, api):
        self.api = api
    def pay(self, rub):
        return self.api.make_payment(int(rub * 100), "RUB")

print(PaymentAdapter(OldPaymentApi()).pay(199.9))

class Order:
    def __init__(self):
        self._listeners = []

    def on_paid(self, callback):
        self._listeners.append(callback)
        return callback                      # можно использовать как декоратор

    def pay(self):
        print("Заказ оплачен")
        for cb in self._listeners:
            cb(self)

order = Order()

@order.on_paid
def send_email(o):
    print("  → письмо клиенту")

order.on_paid(lambda o: print("  → уведомление складу"))
order.pay()