upgrade_pythoncapi.py: C++ files, ignore .git directory

vstinner · vstinner · commit b9cd1e593908 · 2021-12-06T23:40:48.000+01:00
diff --git a/README.rst b/README.rst
@@ -45,7 +45,8 @@ Upgrade ``mod.c`` file::
 
     python3 upgrade_pythoncapi.py mod.c
 
-Upgrade all ``.c`` and ``.h`` files of a project::
+Upgrade all C and C++ files (``.c``, ``.h``, ``.cc``, ``.cpp``, ``.cxx`` and
+``.hpp`` files) of a project::
 
     python3 upgrade_pythoncapi.py directory/
 
diff --git a/upgrade_pythoncapi.py b/upgrade_pythoncapi.py
@@ -15,6 +15,14 @@
 INCLUDE_PYTHONCAPI_COMPAT = f'#include "{PYTHONCAPI_COMPAT_H}"'
 INCLUDE_PYTHONCAPI_COMPAT2 = f'#include <{PYTHONCAPI_COMPAT_H}>'
 
+C_FILE_EXT = (
+    # C language
+    ".c", ".h",
+    # C++ language
+    ".cc", ".cpp", ".cxx", ".hpp",
+)
+IGNORE_DIRS = (".git", ".tox")
+
 
 # Match a C identifier: 'identifier', 'var_3', 'NameCamelCase'
 # Use \b to only match a full word: match "a_b", but not just "b" in "a_b".
@@ -64,7 +72,7 @@ def call_assign_regex(name):
 
 
 def is_c_filename(filename):
-    return filename.endswith((".c", ".h"))
+    return filename.endswith(C_FILE_EXT)
 
 
 class Operation:
@@ -496,10 +504,11 @@ def _walk_dir(self, path):
 
         for dirpath, dirnames, filenames in os.walk(path):
             # Don't walk into .tox
-            try:
-                dirnames.remove(".tox")
-            except ValueError:
-                pass
+            for ignore_name in IGNORE_DIRS:
+                try:
+                    dirnames.remove(ignore_name)
+                except ValueError:
+                    pass
             for filename in filenames:
                 if is_c_filename(filename):
                     yield os.path.join(dirpath, filename)
