Войти Регистрация
Код-ревью

43. Загрузка зарплатного реестра в Альфа-Банке

Средняя

Python · 39 строк

Бухгалтер компании выгружает из 1С реестр на зарплату: CSV в cp1251 с заголовком «ФИО;Счёт;Сумма», суммы вида 45 000,50. Функция проверяет баланс компании, зачисляет деньги сотрудникам на счета в банке и списывает сумму с компании.

В реестре до 3 000 строк. credit_account сразу переводит деньги на счёт, get_connection — psycopg2. Если импорт упал, бухгалтер загружает тот же файл ещё раз. Отметьте строки с ошибками.

В работе: Альфа-Бизнес Онлайн: «Зарплатный проект» → «Загрузить реестр» 5-го и 20-го числа.

Как проверяем Нажмите на строку с ошибкой и коротко напишите, что не так. Баг засчитан, если отмечена хотя бы одна его строка. Зачёт — найдено не меньше 60 % багов с учётом серьёзности (критичный весит 3, обычный 2, мелкий 1), а лишних отметок не больше, чем багов в коде. Пустые строки, скобки и комментарии лишними не считаем. Ответ один на подход: после него показываем разбор.
CSVкодировкиDecimalидемпотентностьблокировкидоступ

Источник: наша задача на данных Альфа-Банка: зарплатный проект для бизнеса. Условие пересказано своими словами.

  1. import csv
  2. import logging
  4. from .bank import credit_account
  5. from .db import get_connection
  7. log = logging.getLogger(__name__)
  10. def import_payroll(path, company_id):
  11. """Зарплатный реестр от бухгалтерии клиента: ФИО;счёт;сумма."""
  12. total = 0
  13. rows = []
  14. f = open(path)
  15. for line in csv.reader(f, delimiter=";"):
  16. name, account, amount = line
  17. rows.append((name, account, float(amount)))
  18. total += float(amount)
  20. conn = get_connection()
  21. cur = conn.cursor()
  22. cur.execute("SELECT balance FROM companies WHERE id = %s", (company_id,))
  23. balance = cur.fetchone()[0]
  24. if total > balance:
  25. raise ValueError("Недостаточно средств на счёте компании")
  27. for name, account, amount in rows:
  28. cur.execute("SELECT id FROM employees WHERE account = %s", (account,))
  29. emp = cur.fetchone()
  30. if emp is None:
  31. log.warning("Сотрудник не найден: %s", line)
  32. continue
  33. credit_account(account, amount)
  34. cur.execute(
  35. "UPDATE companies SET balance = balance - %s WHERE id = %s", (amount, company_id)
  36. )
  37. conn.commit()
  38. log.info("Реестр %s: %d строк, %s ₽", path, len(rows), total)
  39. return total