import shutil
import subprocess
import sys
import tempfile
from pathlib import Path

# окрема тека для всіх файлів цієї практики — її ми видалимо в самому кінці
majsternya = Path(tempfile.mkdtemp(prefix="tema24_"))

print("майстерня:", majsternya)
print("тека існує:", majsternya.is_dir())
print("інтерпретатор, яким запускатимемо скрипт:", sys.executable)

майстерня: /tmp/tema24_0iojo3q3
тека існує: True
інтерпретатор, яким запускатимемо скрипт: /usr/bin/python3

prodazhi_tekst = """data,tovar,kilkist,suma
2026-07-01,кава,2,180.00
2026-07-01,чай,1,90.00
2026-07-02,кава,3,270.00
2026-07-03,тістечко,4,160.00
2026-07-03,кава,1,90.00
"""

fajl_danykh = majsternya / "prodazhi.csv"
fajl_danykh.write_text(prodazhi_tekst, encoding="utf-8")

# рахуємо очікувані числа тут, у зошиті, щоб було з чим порівнювати скрипт
ochikuvana_suma = 180.00 + 90.00 + 270.00 + 160.00 + 90.00
ochikuvana_kilkist = 2 + 1 + 3 + 4 + 1

print("створено:", fajl_danykh.name)
print("рядків із даними:", len(prodazhi_tekst.strip().split("\n")) - 1)
print("очікувана сума по колонці suma   :", ochikuvana_suma)
print("очікувана сума по колонці kilkist:", ochikuvana_kilkist)

створено: prodazhi.csv
рядків із даними: 5
очікувана сума по колонці suma   : 790.0
очікувана сума по колонці kilkist: 11

kod_skrypta = '''\
#!/usr/bin/env python3
"""Підсумок по числовій колонці CSV-файлу з продажами."""
import argparse
import csv
import sys

DIYI = {"suma": "сума", "serednye": "середнє", "maksymum": "максимум"}


def zrobyty_rozbyrach():
    """Опис аргументів окремо від роботи — так розбирач можна перевірити в тесті."""
    parser = argparse.ArgumentParser(
        prog="zvit.py",
        description="Рахує підсумок по числовій колонці CSV-файлу з продажами.")
    parser.add_argument("fajl", help="шлях до CSV-файлу")
    parser.add_argument("-k", "--kolonka", default="suma", metavar="НАЗВА",
                        help="назва числової колонки (типово: %(default)s)")
    parser.add_argument("-d", "--dia", default="suma",
                        choices=["suma", "serednye", "maksymum"],
                        help="що порахувати (типово: %(default)s)")
    parser.add_argument("-t", "--tochnist", type=int, default=2, metavar="N",
                        help="знаків після коми (типово: %(default)s)")
    parser.add_argument("--tyho", action="store_true",
                        help="друкувати лише число, без заголовка")
    return parser


def main(argv=None):
    """Повертає код виходу: 0 — успіх, 1 — не змогли порахувати."""
    args = zrobyty_rozbyrach().parse_args(argv)

    try:
        with open(args.fajl, encoding="utf-8", newline="") as f:
            ryadky = list(csv.DictReader(f))
    except OSError as pomylka:
        # повідомлення про помилку — у stderr, щоб не змішатися з результатом
        print(f"zvit.py: {pomylka}", file=sys.stderr)
        return 1

    if not ryadky:
        print("zvit.py: у файлі немає жодного рядка з даними", file=sys.stderr)
        return 1
    if args.kolonka not in ryadky[0]:
        nayavni = ", ".join(ryadky[0].keys())
        print(f"zvit.py: немає колонки {args.kolonka!r}; є: {nayavni}", file=sys.stderr)
        return 1

    try:
        chysla = [float(r[args.kolonka]) for r in ryadky]
    except ValueError:
        print(f"zvit.py: колонка {args.kolonka!r} не числова", file=sys.stderr)
        return 1

    if args.dia == "suma":
        rezultat = sum(chysla)
    elif args.dia == "serednye":
        rezultat = sum(chysla) / len(chysla)
    else:
        rezultat = max(chysla)

    # службовий рядок — теж у stderr: це звіт про перебіг, а не результат
    print(f"zvit.py: оброблено {len(chysla)} рядків", file=sys.stderr)

    # і тільки це — результат, тільки він іде в stdout
    tekst = f"{rezultat:.{args.tochnist}f}"
    if args.tyho:
        print(tekst)
    else:
        print(f"{DIYI[args.dia]} по колонці {args.kolonka}: {tekst}")
    return 0


if __name__ == "__main__":
    sys.exit(main())
'''

fajl_skrypta = majsternya / "zvit.py"
fajl_skrypta.write_text(kod_skrypta, encoding="utf-8")

print("створено:", fajl_skrypta.name)
print("рядків у скрипті:", len(kod_skrypta.strip().split("\n")))

створено: zvit.py
рядків у скрипті: 74

def zapustyty(*argumenty):
    """Запускає zvit.py з переданими аргументами й повертає результат subprocess."""
    # text=True — щоб stdout і stderr прийшли рядками, а не байтами
    return subprocess.run(
        [sys.executable, str(fajl_skrypta), *argumenty],
        capture_output=True, text=True, encoding="utf-8")


sproba = zapustyty(str(fajl_danykh))

print("код виходу:", sproba.returncode)
print("--- stdout ---")
print(sproba.stdout, end="")
print("--- stderr ---")
print(sproba.stderr, end="")

код виходу: 0
--- stdout ---
сума по колонці suma: 790.00
--- stderr ---
zvit.py: оброблено 5 рядків

udacha = zapustyty(str(fajl_danykh))

assert udacha.returncode == 0, f"успішний запуск має давати 0, а дав {udacha.returncode}"

# у stdout має бути рівно один рядок — результат
ryadky_stdout = udacha.stdout.strip().split("\n")
assert len(ryadky_stdout) == 1, f"у stdout мав бути один рядок, а їх {len(ryadky_stdout)}"

# дістаємо число з кінця рядка й порівнюємо з порахованим у зошиті
chyslo_zi_skrypta = float(ryadky_stdout[0].split(":")[-1])
assert chyslo_zi_skrypta == ochikuvana_suma, \
    f"скрипт дав {chyslo_zi_skrypta}, а ми чекали {ochikuvana_suma}"

print("скрипт надрукував:", ryadky_stdout[0])
print("ми порахували руками:", ochikuvana_suma)
print("✅ скрипт рахує те саме, що й ми")

скрипт надрукував: сума по колонці suma: 790.00
ми порахували руками: 790.0
✅ скрипт рахує те саме, що й ми

dovidka = zapustyty("-h")

print("код виходу:", dovidka.returncode)
print("--- stdout ---")
print(dovidka.stdout)

код виходу: 0
--- stdout ---
usage: zvit.py [-h] [-k НАЗВА] [-d {suma,serednye,maksymum}] [-t N] [--tyho]
               fajl

Рахує підсумок по числовій колонці CSV-файлу з продажами.

positional arguments:
  fajl                  шлях до CSV-файлу

options:
  -h, --help            show this help message and exit
  -k НАЗВА, --kolonka НАЗВА
                        назва числової колонки (типово: suma)
  -d {suma,serednye,maksymum}, --dia {suma,serednye,maksymum}
                        що порахувати (типово: suma)
  -t N, --tochnist N    знаків після коми (типово: 2)
  --tyho                друкувати лише число, без заголовка

# довідка — це результат роботи (нас про неї попросили), тому вона в stdout і код 0
assert dovidka.returncode == 0, "запит довідки — це успіх, код має бути 0"
assert dovidka.stderr == "", "довідка не має нічого писати в stderr"

# усі описані аргументи мають бути в тексті довідки — і ми їх туди не вписували
for chastyna in ["usage:", "fajl", "--kolonka", "--dia", "--tochnist", "--tyho", "-h"]:
    assert chastyna in dovidka.stdout, f"у довідці немає {chastyna!r}"

# типові значення підставились через %(default)s — довідка не може застаріти
assert "типово: suma" in dovidka.stdout, "у довідці немає підставленого default"

print("✅ довідка згенерувалась сама й містить усі аргументи")

✅ довідка згенерувалась сама й містить усі аргументи

# (аргументи, що очікуємо побачити в stdout)
perevirky = [
    ([str(fajl_danykh)], "сума по колонці suma: 790.00"),
    ([str(fajl_danykh), "--kolonka", "kilkist"], "сума по колонці kilkist: 11.00"),
    ([str(fajl_danykh), "--dia", "serednye"], "середнє по колонці suma: 158.00"),
    ([str(fajl_danykh), "--dia", "maksymum", "-t", "1"], "максимум по колонці suma: 270.0"),
    (["-k", "kilkist", "-d", "maksymum", "--tyho", str(fajl_danykh)], "4.00"),
]

for argumenty, ochikuvano in perevirky:
    vyhid = zapustyty(*argumenty)
    otrymano = vyhid.stdout.strip()
    assert vyhid.returncode == 0, f"запуск {argumenty} мав завершитись успішно"
    assert otrymano == ochikuvano, f"чекали {ochikuvano!r}, отримали {otrymano!r}"
    # у показі замінюємо довгий тимчасовий шлях на просту назву файлу
    pokaz = " ".join(a.replace(str(fajl_danykh), "prodazhi.csv") for a in argumenty)
    print(f"✔ zvit.py {pokaz:<52} → {otrymano}")

print("✅ усі комбінації аргументів дали очікуваний результат")

✔ zvit.py prodazhi.csv                                         → сума по колонці suma: 790.00

✔ zvit.py prodazhi.csv --kolonka kilkist                       → сума по колонці kilkist: 11.00

✔ zvit.py prodazhi.csv --dia serednye                          → середнє по колонці suma: 158.00

✔ zvit.py prodazhi.csv --dia maksymum -t 1                     → максимум по колонці suma: 270.0

✔ zvit.py -k kilkist -d maksymum --tyho prodazhi.csv           → 4.00
✅ усі комбінації аргументів дали очікуваний результат

polamani = [
    ("без обовʼязкового аргументу", []),
    ("значення поза choices", [str(fajl_danykh), "--dia", "medi"]),
    ("не число там, де чекали int", [str(fajl_danykh), "--tochnist", "bagato"]),
    ("невідомий аргумент", [str(fajl_danykh), "--kolir", "chervonyj"]),
]

for opys, argumenty in polamani:
    vyhid = zapustyty(*argumenty)
    ostannij_ryadok = vyhid.stderr.strip().split("\n")[-1]
    print(f"{opys:<32} код {vyhid.returncode}")
    print(f"{'':<32} {ostannij_ryadok}")
    # argparse завжди відповідає двійкою на зіпсований командний рядок
    assert vyhid.returncode == 2, f"{opys}: чекали код 2, отримали {vyhid.returncode}"
    assert vyhid.stdout == "", f"{opys}: у stdout не має бути нічого"

print("✅ на кожен зіпсований командний рядок argparse відповів кодом 2")

без обовʼязкового аргументу      код 2
                                 zvit.py: error: the following arguments are required: fajl

значення поза choices            код 2
                                 zvit.py: error: argument -d/--dia: invalid choice: 'medi' (choose from 'suma', 'serednye', 'maksymum')

не число там, де чекали int      код 2
                                 zvit.py: error: argument -t/--tochnist: invalid int value: 'bagato'

невідомий аргумент               код 2
                                 zvit.py: error: unrecognized arguments: --kolir chervonyj
✅ на кожен зіпсований командний рядок argparse відповів кодом 2

nemaye_fajlu = zapustyty(str(majsternya / "nemaye.csv"))
nemaye_kolonky = zapustyty(str(fajl_danykh), "--kolonka", "cina")

for opys, vyhid in [("файлу немає", nemaye_fajlu), ("колонки немає", nemaye_kolonky)]:
    print(f"{opys:<16} код {vyhid.returncode} · stderr: {vyhid.stderr.strip()}")
    assert vyhid.returncode == 1, f"{opys}: чекали код 1, отримали {vyhid.returncode}"

# і найголовніше: у stderr має бути зрозуміле пояснення, а не порожнеча
assert "No such file" in nemaye_fajlu.stderr or "немає" in nemaye_fajlu.stderr, \
    "повідомлення про відсутній файл має пояснювати, що сталось"
assert "cina" in nemaye_kolonky.stderr, "у повідомленні має бути названа шукана колонка"
assert "data, tovar, kilkist, suma" in nemaye_kolonky.stderr, \
    "повідомлення має підказати, які колонки насправді є"

print("✅ помилки виконання дають код 1 і зрозуміле повідомлення")

файлу немає      код 1 · stderr: zvit.py: [Errno 2] No such file or directory: '/tmp/tema24_0iojo3q3/nemaye.csv'
колонки немає    код 1 · stderr: zvit.py: немає колонки 'cina'; є: data, tovar, kilkist, suma
✅ помилки виконання дають код 1 і зрозуміле повідомлення

vyhid = zapustyty(str(fajl_danykh))

print("--- stdout (результат) ---")
print(vyhid.stdout, end="")
print("--- stderr (службове) ---")
print(vyhid.stderr, end="")

# службовий рядок є — але не в stdout
assert "оброблено" in vyhid.stderr, "звіт про перебіг мав піти в stderr"
assert "оброблено" not in vyhid.stdout, "звіт про перебіг НЕ мав потрапити в stdout"

# а результат є рівно в stdout і ніде більше
assert "сума по колонці" in vyhid.stdout, "результат мав піти в stdout"
assert "сума по колонці" not in vyhid.stderr, "результат НЕ мав потрапити в stderr"

print("✅ потоки не змішались: результат окремо, повідомлення окремо")

--- stdout (результат) ---
сума по колонці suma: 790.00
--- stderr (службове) ---
zvit.py: оброблено 5 рядків
✅ потоки не змішались: результат окремо, повідомлення окремо

tyho = zapustyty(str(fajl_danykh), "--tyho")

# увесь stdout цілком має бути числом — жодного зайвого слова
chyslo = float(tyho.stdout.strip())

assert chyslo == ochikuvana_suma, f"чекали {ochikuvana_suma}, отримали {chyslo}"
assert tyho.stderr.strip() != "", "службове повідомлення нікуди не зникло — воно в stderr"

print("stdout цілком:", repr(tyho.stdout))
print("перетворили на число:", chyslo)
print("а службове повідомлення тим часом у stderr:", tyho.stderr.strip())
print("✅ вивід готовий до конвеєра: наступна програма отримає чисті дані")

stdout цілком: '790.00\n'
перетворили на число: 790.0
а службове повідомлення тим часом у stderr: zvit.py: оброблено 5 рядків
✅ вивід готовий до конвеєра: наступна програма отримає чисті дані

# додаємо майстерню в sys.path, щоб import побачив наш файл (як у темі 16)
sys.path.insert(0, str(majsternya))

import zvit  # noqa: E402 — імпорт посеред файлу тут навмисний, це частина демонстрації

# сам факт імпорту нічого не надрукував і нічого не порахував:
# уся робота захована під if __name__ == "__main__"
print("модуль імпортовано, __name__ усередині нього:", zvit.__name__)
print("а функція main на місці:", zvit.main)

модуль імпортовано, __name__ усередині нього: zvit
а функція main на місці: <function main at 0x7f6fd4f66ac0>

import io
from contextlib import redirect_stdout, redirect_stderr

# перехоплюємо обидва потоки окремо — так само, як це робив subprocess
buferu_stdout = io.StringIO()
buferu_stderr = io.StringIO()

with redirect_stdout(buferu_stdout), redirect_stderr(buferu_stderr):
    kod = zvit.main([str(fajl_danykh), "--dia", "serednye"])

print("код, який повернула main():", kod)
print("stdout:", buferu_stdout.getvalue().strip())
print("stderr:", buferu_stderr.getvalue().strip())

assert kod == 0, "успішний виклик main() має повернути 0"
assert "середнє по колонці suma: 158.00" in buferu_stdout.getvalue(), \
    "main() мала надрукувати те саме, що й запуск через subprocess"
print("✅ main() працює так само, як окремий процес — але вдесятеро швидше")

код, який повернула main(): 0
stdout: середнє по колонці suma: 158.00
stderr: zvit.py: оброблено 5 рядків
✅ main() працює так само, як окремий процес — але вдесятеро швидше

buferu_stderr = io.StringIO()

try:
    with redirect_stderr(buferu_stderr):
        zvit.main([str(fajl_danykh), "--dia", "medi"])
except SystemExit as vyhid_z_prohramy:
    kod_vyhodu = vyhid_z_prohramy.code
else:
    kod_vyhodu = None

print("SystemExit спіймано, код усередині:", kod_vyhodu)
print("а в stderr тим часом:", buferu_stderr.getvalue().strip().split("\n")[-1])

assert kod_vyhodu == 2, f"argparse мав завершити програму з кодом 2, а дав {kod_vyhodu}"
assert "invalid choice" in buferu_stderr.getvalue(), \
    "argparse мав пояснити, що значення не входить у choices"

# і ще одна перевірка, яка пояснює, чому ловити треба саме SystemExit
assert not isinstance(SystemExit(2), Exception), \
    "SystemExit навмисно не Exception — щоб except Exception його не з'їв"
print("✅ SystemExit несе код виходу й навмисно живе поза ієрархією Exception")

SystemExit спіймано, код усередині: 2
а в stderr тим часом: zvit.py: error: argument -d/--dia: invalid choice: 'medi' (choose from 'suma', 'serednye', 'maksymum')
✅ SystemExit несе код виходу й навмисно живе поза ієрархією Exception

# крім наших двох файлів там з'явився ще й __pycache__ — слід від import zvit
bulo = sorted(p.name for p in majsternya.iterdir())

sys.path.remove(str(majsternya))
shutil.rmtree(majsternya)

print("у майстерні лежало:", ", ".join(bulo))
print("тека існує:", majsternya.exists())

assert not majsternya.exists(), "тимчасова тека мала зникнути"
print("✅ проєкт лишився чистим")

у майстерні лежало: __pycache__, prodazhi.csv, zvit.py
тека існує: False
✅ проєкт лишився чистим

10 · Прибираємо за собоюТимчасова тека більше не потрібна. Видаляємо її цілком — і перевіряємо, що відпрактики не лишилось нічого.¶