Added support for final classes

DJSchaffner · Jun 24, 2021 · f8caee6 · f8caee6
1 parent aa6e581
commit f8caee6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -23,7 +23,7 @@ def parse_file(path: pathlib.Path, package_set: set, import_set: set, content: l
         import_set.add(line)
 
       # Its a class definition, remove the visibility modifier
-      elif (class_definition_match := re.match(r"\s*(public|private|protected)\s+((abstract\s+)?(class|enum|interface).*\n)", line)):
+      elif (class_definition_match := re.match(r"\s*(public|private|protected)\s+(((abstract|final)\s+)?(class|enum|interface).*\n)", line)):
         content.append(class_definition_match.groups()[1])
 
       # Anythin else just add to content if not whitespace