utils/autopep8_clean_config.py

   1 # SPDX-License-Identifier: GPL-2.0-or-later
   2
   3 import os
   4
   5 from typing import (
   6     Generator,
   7     Callable,
   8     Set,
   9     Tuple,
  10 )
  11
  12 PATHS: Tuple[str, ...] = (
  13     "build_files",
  14     "doc",
  15     "release/datafiles",
  16     "release/lts",
  17     "release/scripts/freestyle",
  18     "release/scripts/modules",
  19     "release/scripts/presets",
  20     "release/scripts/startup",
  21     "release/scripts/templates_py",
  22     "source/blender",
  23     "source/tools",
  24     "tests",
  25 )
  26
  27 SOURCE_DIR = os.path.normpath(os.path.abspath(os.path.normpath(
  28     os.path.join(os.path.dirname(__file__), "..", "..", ".."))))
  29
  30 PATHS = tuple(
  31     os.path.join(SOURCE_DIR, p.replace("/", os.sep))
  32     for p in PATHS
  33 )
  34
  35 PATHS_EXCLUDE: Set[str] = set(
  36     os.path.join(SOURCE_DIR, p.replace("/", os.sep))
  37     for p in
  38     (
  39         "source/tools/svn_rev_map/sha1_to_rev.py",
  40         "source/tools/svn_rev_map/rev_to_sha1.py",
  41         "source/tools/svn_rev_map/rev_to_sha1.py",
  42         "release/scripts/modules/rna_manual_reference.py",
  43     )
  44 )
  45
  46
  47 def files(path: str, test_fn: Callable[[str], bool]) -> Generator[str, None, None]:
  48     for dirpath, dirnames, filenames in os.walk(path):
  49         # skip '.git'
  50         dirnames[:] = [d for d in dirnames if not d.startswith(".")]
  51         for filename in filenames:
  52             if test_fn(filename):
  53                 filepath = os.path.join(dirpath, filename)
  54                 yield filepath