adding more tests, refactoring

Author: nikolay-e

.github/workflows/ci.yml

@@ -1,3 +1,4 @@
+# .github/workflows/ci.yml
 name: TreeMapper CI
 
 on:
@@ -8,9 +9,54 @@ on:
       - main
 
 jobs:
+  # --- Новое задание для проверки качества кода ---
+  lint-type-check:
+    name: Lint & Type Check
+    runs-on: ubuntu-latest # Достаточно одной ОС/версии Python
+    steps:
+      - name: Checkout Code
+        uses: actions/checkout@v3
+
+      - name: Set up Python 3.11
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.11'
+
+      - name: Cache pip Dependencies
+        uses: actions/cache@v3
+        with:
+          # Используем setup.cfg для ключа кэша
+          path: ~/.cache/pip
+          key: ${{ runner.os }}-pip-${{ hashFiles('**/setup.cfg') }}
+          restore-keys: |
+            ${{ runner.os }}-pip-
+
+      - name: Install Linters and Type Checker
+        run: |
+          python -m pip install --upgrade pip
+          # Устанавливаем только dev-зависимости (включая линтеры, mypy)
+          # Предполагается, что они в extras_require [dev] в setup.cfg
+          pip install .[dev]
+          # Если линтеров нет в setup.cfg, установите их отдельно:
+          # pip install flake8 black isort mypy types-PyYAML types-pathspec
+
+      - name: Run Linters and Formatters Check
+        run: |
+          flake8 src tests
+          black --check src tests
+          isort --check-only src tests
+
+      - name: Run Type Checker (Mypy)
+        run: |
+          mypy src tests # Проверяем и исходники, и тесты
+
+  # --- Обновленное задание для тестов ---
   test:
+    needs: lint-type-check # Запускаем тесты ПОСЛЕ проверки качества кода
     strategy:
+      fail-fast: false # Не отменять другие тесты при падении одного
       matrix:
+        # Оставим 3 версии Ubuntu для надежности, можно сократить до latest и 20.04
         os: [ ubuntu-latest, ubuntu-20.04, ubuntu-22.04, macos-latest, windows-latest ]
         python-version: [ 3.9, '3.10', '3.11' ]
 
@@ -28,34 +74,47 @@ jobs:
       - name: Cache pip Dependencies
         uses: actions/cache@v3
         with:
+          # Путь к кэшу pip (работает для Linux/macOS, должно работать и для Windows в actions/cache@v3)
           path: ~/.cache/pip
-          key: ${{ runner.os }}-pip-${{ hashFiles('**/setup.cfg') }}
+          # Ключ кэша по setup.cfg
+          key: ${{ runner.os }}-pip-${{ matrix.python-version }}-${{ hashFiles('**/setup.cfg') }}
           restore-keys: |
+            ${{ runner.os }}-pip-${{ matrix.python-version }}-
             ${{ runner.os }}-pip-
 
       - name: Install Dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install -r requirements.txt
-          pip install -e .
+          # Устанавливаем пакет в editable режиме + dev зависимости (pytest, pytest-cov)
+          # Предполагает наличие extras_require [dev] в setup.cfg
+          pip install -e .[dev]
 
-      - name: Build with PyInstaller
-        run: |
-          python -m PyInstaller --clean -y --dist ./dist/${{ runner.os }} --workpath /tmp treemapper.spec
-
-      - name: Build with python -m build
-        run: |
-          python -m build
+      # --- УДАЛЕНЫ ШАГИ СБОРКИ PyInstaller и build ---
 
-      - name: Run Tests
+      - name: Run Tests with Coverage
         run: |
-          pytest
+          # Запускаем pytest с генерацией отчета покрытия в формате XML
+          pytest -v --cov=src/treemapper --cov-report=xml
+
+      # --- Новый шаг: Выгрузка покрытия (запускается только один раз) ---
+      - name: Upload coverage reports to Codecov
+        if: matrix.os == 'ubuntu-latest' && matrix.python-version == '3.11' # Запускаем только для одной комбинации
+        uses: codecov/codecov-action@v4 # Используем v4
+        # Не требует токена для публичных репозиториев на github.com
+        # env:
+        #   CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }} # Для приватных репо или других платформ
+        with:
+           files: ./coverage.xml # Путь к файлу отчета
+           fail_ci_if_error: true # Завалить сборку, если выгрузка не удалась
+           verbose: true # Для отладки
 
+  # --- Обновленное задание для PyPy ---
   test-pypy:
+    needs: lint-type-check
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [ pypy-3.9 ]
+        python-version: [ pypy-3.9 ] # Можно добавить pypy-3.10, если нужно
 
     steps:
       - name: Checkout Code
@@ -70,21 +129,16 @@ jobs:
         uses: actions/cache@v3
         with:
           path: ~/.cache/pip
-          key: pypy-pip-${{ hashFiles('**/setup.cfg') }}
+          key: pypy-${{ matrix.python-version }}-pip-${{ hashFiles('**/setup.cfg') }}
           restore-keys: |
-            pypy-pip-
+            pypy-${{ matrix.python-version }}-pip-
 
       - name: Install Dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install -r requirements.txt
-          pip install -e .
-          pip install build pytest
-
-      - name: Build with python -m build
-        run: |
-          python -m build
+          # Устанавливаем с dev зависимостями
+          pip install -e .[dev]
 
       - name: Run Tests
         run: |
-          pytest
+          pytest -v

requirements.txt

@@ -4,3 +4,4 @@ pathspec
 build
 twine
 pyinstaller
+pytest-cov

setup.cfg

@@ -18,12 +18,28 @@ package_dir =
 packages = find:
 python_requires = >=3.9
 install_requires =
-    pathspec
-    pyyaml
+    pathspec>=0.9
+    pyyaml>=5.4
 
 [options.packages.find]
 where = src
 
 [options.entry_points]
 console_scripts =
-    treemapper = treemapper.treemapper:main
+    treemapper = treemapper.treemapper:main
+
+[options.extras_require]
+dev =
+    pytest>=7.0
+    pytest-cov>=3.0
+    build>=0.10
+    twine>=4.0
+    pyinstaller>=5.0
+    flake8>=5.0
+    black>=23.0
+    isort>=5.10
+    mypy>=1.0
+    types-PyYAML
+    types-pathspec
+    pyyaml
+    pathspec

src/treemapper/cli.py

@@ -1,12 +1,14 @@
+# src/treemapper/cli.py
 import argparse
 import sys
 from pathlib import Path
-from typing import Tuple
+from typing import Tuple, Optional # <--- Добавлен Optional
 
-
-def parse_args() -> Tuple[Path, Path, Path, bool, int]:
+# ---> ИЗМЕНЕНИЕ: Заменяем | None на Optional[...] <---
+def parse_args() -> Tuple[Path, Optional[Path], Optional[Path], bool, int]:
     """Parse command line arguments."""
     parser = argparse.ArgumentParser(
+        prog='treemapper',
         description="Generate a YAML representation of a directory structure.",
         formatter_class=argparse.ArgumentDefaultsHelpFormatter)
 
@@ -29,7 +31,7 @@ def parse_args() -> Tuple[Path, Path, Path, bool, int]:
     parser.add_argument(
         "--no-default-ignores",
         action="store_true",
-        help="Disable all default ignores (including .gitignore and .treemapperignore)")
+        help="Disable default ignores (.treemapperignore, .gitignore, output file)")
 
     parser.add_argument(
         "-v", "--verbosity",
@@ -41,15 +43,26 @@ def parse_args() -> Tuple[Path, Path, Path, bool, int]:
 
     args = parser.parse_args()
 
-    root_dir = Path(args.directory).resolve()
-    if not root_dir.is_dir():
-        print(f"Error: The path '{root_dir}' is not a valid directory.")
-        sys.exit(1)
+    try:
+        root_dir = Path(args.directory).resolve(strict=True)
+        if not root_dir.is_dir():
+            print(f"Error: The path '{root_dir}' is not a valid directory.", file=sys.stderr)
+            sys.exit(1)
+    except FileNotFoundError:
+         print(f"Error: The directory '{args.directory}' does not exist.", file=sys.stderr)
+         sys.exit(1)
+    except Exception as e:
+         print(f"Error resolving directory path '{args.directory}': {e}", file=sys.stderr)
+         sys.exit(1)
 
     output_file = Path(args.output_file)
     if not output_file.is_absolute():
         output_file = Path.cwd() / output_file
 
-    ignore_file = Path(args.ignore_file) if args.ignore_file else None
+    ignore_file_path: Optional[Path] = None # Используем Optional для ясности
+    if args.ignore_file:
+        ignore_file_path = Path(args.ignore_file)
+        if not ignore_file_path.is_absolute():
+             ignore_file_path = Path.cwd() / ignore_file_path
 
-    return root_dir, ignore_file, output_file, args.no_default_ignores, args.verbosity
+    return root_dir, ignore_file_path, output_file, args.no_default_ignores, args.verbosity