Рефакторинг кода нейросетью — ИИ перепишет и упростит код

Файл: orders/handlers.py


import json
from typing import Any, Dict, Tuple, Optional

# Assumes Order and notify are available in the import context (kept as in original code).


def _parse_raw(raw: str) -> Tuple[Dict[str, Any], Optional[str], Optional[int], float, str]:
    data = json.loads(raw)
    status = data.get("status")
    order = data.get("order", {}) or {}
    order_id = order.get("id")
    amount = int(order.get("amount", 0)) / 100
    currency = order.get("currency", "RUB")
    return data, status, order_id, amount, currency


def _get_order(order_id: int):
    return Order.objects.get(pk=order_id)


def _save_and_notify(order, event: str):
    order.save()
    notify(order, event)
    return order


def handle_payment_success(raw):
    data, status, order_id, amount, currency = _parse_raw(raw)
    if status != "succeeded":
        return None

    order = _get_order(order_id)
    order.paid = True
    order.amount = amount
    order.currency = currency
    return _save_and_notify(order, "paid")


def handle_payment_refund(raw):
    data, status, order_id, amount, currency = _parse_raw(raw)
    if status != "refunded":
        return None

    order = _get_order(order_id)
    order.paid = False
    order.refunded_amount = amount
    order.currency = currency
    return _save_and_notify(order, "refunded")


def handle_payment_failed(raw):
    data, status, order_id, amount, currency = _parse_raw(raw)
    if status != "failed":
        return None

    order = _get_order(order_id)
    order.paid = False
    order.amount = amount
    order.currency = currency
    order.last_error = data.get("error", {}).get("message", "")
    return _save_and_notify(order, "failed")


Список правок

Важно — рефакторинг, orders/handlers.py строки 1-4 — Добавлен импорт json и typing; json раньше использовался без явного импорта в фрагменте, что могло привести к NameError в файле при запуске. Добавление импорта устраняет этот риск и не меняет поведение.
Важно — удаление дублирования, orders/handlers.py строки 6-14 — Вынесена общая логика парсинга полезной нагрузки в функцию _parse_raw. Ранее одинаковый код повторялся в трёх функциях, что затрудняло поддержку; теперь код централизован, поведение не изменено.
Важно — удаление дублирования, orders/handlers.py строки 17-22 — Вынесена логика сохранения и уведомления в функцию _save_and_notify. Ранее вызов save() и notify(…) повторялся; это уменьшает дублирование без изменения последовательности действий.
Замечание — рефакторинг, orders/handlers.py строки 16-18 — Введена вспомогательная функция _get_order для явного получения заказа по id. Это мелкая структурная правка, упрощающая чтение кода и облегчающая возможное покрытие моками в тестах.
Замечание — рефакторинг/стилистика, orders/handlers.py строки 6-40 — Упорядочена логика каждого обработчика: сначала парсинг и проверка статуса, затем изменение полей и единый save/notify. Это убирает повторяющиеся части и делает код компактнее, поведение осталось прежним (те же поля меняются и те же уведомления отправляются).

Пояснение: все изменения направлены на удаление дублирования и небольшую организацию кода без изменения логики: те же проверки статуса, те же присваивания полей, то же сохранение и уведомление. Форматирование намеренно не тронуто иначе, чем необходимо для корректной работы (импорт json и мелкие вспомогательные функции).

// cart/checkout.js

export function createCheckoutHandler({ db, gateway, mailer, now = () => Date.now() }) {
  async function fetchCart(cartId) {
    return db.carts.findOne({ id: cartId });
  }

  async function fetchProduct(productId) {
    return db.products.findOne({ id: productId });
  }

  async function calculateTotal(cart) {
    let total = 0;
    for (const item of cart.items) {
      const product = await fetchProduct(item.productId);
      if (!product) {
        return { error: { status: 400, body: { error: 'bad item' } } };
      }

      let price = product.price;

      if (product.sale && product.saleEndsAt > now()) {
        price = Math.round(price * (1 - product.salePercent / 100));
      }

      if (cart.promo === 'FIRST10' && !cart.usedPromo) {
        price = Math.round(price * 0.9);
      }

      total += price * item.qty;
    }

    if (total > 500000) {
      total = total - Math.round(total * 0.05);
    }

    return { total };
  }

  async function createOrder(cartId, total) {
    return db.orders.insert({ cartId, total, status: 'new' });
  }

  async function markOrder(orderId, status) {
    return db.orders.update({ id: orderId }, { status });
  }

  async function processPayment(order) {
    return gateway.charge({ orderId: order.id, amount: order.total ?? order.amount ?? 0 });
  }

  return async function checkout(req, res) {
    const cart = await fetchCart(req.body.cartId);
    if (!cart) {
      res.status(404).json({ error: 'cart not found' });
      return;
    }

    const totalResult = await calculateTotal(cart);
    if (totalResult.error) {
      res.status(totalResult.error.status).json(totalResult.error.body);
      return;
    }
    const total = totalResult.total;

    const order = await createOrder(cart.id, total);

    try {
      const payment = await processPayment(order);

      if (payment.status === 'succeeded') {
        await markOrder(order.id, 'paid');
        await mailer.send(cart.email, 'Заказ оплачен', `Заказ ${order.id} на сумму ${total / 100}`);
        res.json({ ok: true, orderId: order.id });
      } else {
        await markOrder(order.id, 'failed');
        res.status(402).json({ error: 'payment failed' });
      }
    } catch (e) {
      console.log(e);
      res.status(500).json({ error: 'server error' });
    }
  };
}

// Backwards-compatible default handler using existing globals (if present)
export const checkout = createCheckoutHandler({ db, gateway, mailer, now: () => Date.now() });


Список правок:

важно, структурная правка, cart/checkout.js строки ~1-3, “Вынес создание обработчика в createCheckoutHandler”: Разделил создание обработчика и внедрение зависимостей (db, gateway, mailer, now) — ранее функции напрямую использовали глобалы, что мешало тестируемости и переиспользованию.
важно, структурная правка, cart/checkout.js строки ~5-17, “Вынес логику подсчёта в calculateTotal”: Цикл и логика расчёта цены и скидок вытянуты в отдельную функцию — это уменьшает размер основной функции и упрощает покрытие тестами; раньше длинная функция была трудна для понимания и поддержки.
замечание, структурная правка, cart/checkout.js строки ~6-9, “Разделил доступ к данным на fetchCart/fetchProduct”: Обёртки для чтения из db вынесены в отдельные функции, чтобы разделить ответственность и упростить мокирование в тестах — раньше прямые вызовы db затрудняли подмену во время тестирования.
важно, структурная правка, cart/checkout.js строки ~18-23, “Вынес создание и обновление заказа в createOrder/markOrder”: Операции с заказами вынесены в отдельные функции для ясности и возможности централизованной обработки ошибок/логирования — раньше эти вызовы были в теле основной функции, создавая когезионный блок.
замечание, структурная правка, cart/checkout.js строки ~24-27, “Вынес оплату в processPayment”: Вызов шлюза оплаты вынесен в отдельную функцию, что упрощает замену/mock и делает код основного обработчика компактнее.
замечание, поведенческая/структурная правка, cart/checkout.js строки ~28-31, “Сохранил поведение ошибок и статусы HTTP”: Вместо бросания исключений для ожидаемых ошибок (cart not found, bad item) теперь функции возвращают структуру с ошибкой, а обработчик отвечает с теми же статусами — поведение сохранено, но теперь централизовано.
замечание, структурная правка, cart/checkout.js строки ~1-37, “Добавил параметр now для определения текущего времени”: Параметр now (по умолчанию Date.now) введён для тестируемости условий распродажи; это не меняет поведения в проде, но упрощает тесты.
замечание, совместимость, cart/checkout.js строки ~39-41, “Добавил экспорт checkout для обратной совместимости”: Оставил экспорт checkout, созданный из createCheckoutHandler с предполагаемыми глобалами, чтобы существующий код не ломался — это упрощает постепенный рефактор.

// reports/build.ts
export function buildReport(rows: any[], opts: any) {
  const result: any = {};
  for (let i = 0; i < rows.length; i++) {
    const r = rows[i];
    const key = r.department + '/' + r.manager;
    if (result[key] == undefined) {
      result[key] = { total: 0, count: 0, avg: 0, items: [] };
    }
    const amount = parseFloat(r.amount);
    result[key].total = result[key].total + (isNaN(amount) ? 0 : amount);
    result[key].count++;
    result[key].avg = result[key].total / result[key].count;
    result[key].items.push(r);
  }
  if (opts.withTax) {
    for (const key in result) {
      result[key].total = result[key].total * 1.2;
    }
  }
  return result;
}

Список правок

Правка 1

Уровень: Критично
Тип: Logic bug — неверное место применения налога
Файл: reports/build.ts
Строки: 14–16 (оригинал)
Заголовок: Налог * 1.2 применяется внутри цикла — накапливается с каждой новой строкой группы

Объяснение.
Блок if (opts.withTax) { result[key].total *= 1.2 } находился внутри for-цикла. При каждом добавлении следующей строки той же группы total умножался на 1,2 ещё раз, поверх уже умноженного значения. Для группы из трёх строк с одинаковыми суммами итоговый total оказывался завышен в 1.2² и 1.2³ раз — данные необратимо искажались. Исправление: блок вынесен в отдельный for…in после завершения основного цикла, чтобы коэффициент применялся ровно один раз к финальному total каждой группы.

Правка 2

Уровень: Критично
Тип: Data loss — беззвучное распространение NaN
Файл: reports/build.ts
Строки: 10–11 (оригинал)
Заголовок: parseFloat возвращает NaN при отсутствующем или некорректном r.amount, после чего total и avg группы безвозвратно становятся NaN

Объяснение.
parseFloat(undefined) и parseFloat("abc") возвращают NaN. После первого такого значения NaN попадает в total, и все последующие арифметические операции над ним тоже дают NaN — все накопленные данные группы теряются без каких-либо ошибок в рантайме. Исправление: добавлен guard isNaN(amount) ? 0 : amount, который заменяет невалидное значение нулём и не прерывает накопление суммы по группе.