`build/scripts` ya style --py

1 files changed, 43 insertions, 22 deletions

diff --git a/build/scripts/resolve_java_srcs.py b/build/scripts/resolve_java_srcs.py
index a2e6c20012..8d3d6285e7 100644
--- a/build/scripts/resolve_java_srcs.py
+++ b/build/scripts/resolve_java_srcs.py
@@ -9,29 +9,36 @@ def list_all_files(directory, prefix='/', hidden_files=False):
     if os.path.exists(directory):
         for i in os.listdir(directory):
             abs_path = os.path.join(directory, i)
-            result += list_all_files(os.path.join(directory, abs_path), prefix + i + '/', hidden_files) \
-                if os.path.isdir(abs_path) else ([prefix + i] if (hidden_files or not i.startswith('.')) else [])
+            result += (
+                list_all_files(os.path.join(directory, abs_path), prefix + i + '/', hidden_files)
+                if os.path.isdir(abs_path)
+                else ([prefix + i] if (hidden_files or not i.startswith('.')) else [])
+            )
     return result
 
 
 def pattern_to_regexp(p):
-    return '^' + \
-           ('/' if not p.startswith('**') else '') + \
-           re.escape(p).replace(
-               r'\*\*\/', '[_DIR_]'
-           ).replace(
-               r'\*', '[_FILE_]'
-           ).replace(
-               '[_DIR_]', '(.*/)?'
-           ).replace(
-               '[_FILE_]', '([^/]*)'
-           ) + '$'
-
-
-def resolve_java_srcs(srcdir, include_patterns, exclude_patterns, all_resources, resolve_kotlin=False, resolve_groovy=False):
+    return (
+        '^'
+        + ('/' if not p.startswith('**') else '')
+        + re.escape(p)
+        .replace(r'\*\*\/', '[_DIR_]')
+        .replace(r'\*', '[_FILE_]')
+        .replace('[_DIR_]', '(.*/)?')
+        .replace('[_FILE_]', '([^/]*)')
+        + '$'
+    )
+
+
+def resolve_java_srcs(
+    srcdir, include_patterns, exclude_patterns, all_resources, resolve_kotlin=False, resolve_groovy=False
+):
     result = {'java': [], 'not_java': [], 'kotlin': [], 'groovy': []}
     include_patterns_normal, include_patterns_hidden, exclude_patterns_normal, exclude_patterns_hidden = [], [], [], []
-    for vis, hid, patterns in ((include_patterns_normal, include_patterns_hidden, include_patterns), (exclude_patterns_normal, exclude_patterns_hidden, exclude_patterns),):
+    for vis, hid, patterns in (
+        (include_patterns_normal, include_patterns_hidden, include_patterns),
+        (exclude_patterns_normal, exclude_patterns_hidden, exclude_patterns),
+    ):
         for pattern in patterns:
             if (pattern if pattern.find('/') == -1 else pattern.rsplit('/', 1)[1]).startswith('.'):
                 hid.append(pattern)
@@ -42,7 +49,7 @@ def resolve_java_srcs(srcdir, include_patterns, exclude_patterns, all_resources,
             re_patterns = [re.compile(i, re.IGNORECASE) for i in re_patterns]
         else:
             re_patterns = [re.compile(i) for i in re_patterns]
-        vis[:], hid[:] = re_patterns[:len(vis)], re_patterns[len(vis):]
+        vis[:], hid[:] = re_patterns[: len(vis)], re_patterns[len(vis) :]
 
     for inc_patterns, exc_patterns, with_hidden_files in (
         (include_patterns_normal, exclude_patterns_normal, False),
@@ -77,8 +84,22 @@ def resolve_java_srcs(srcdir, include_patterns, exclude_patterns, all_resources,
     return sorted(result['java']), sorted(result['not_java']), sorted(result['kotlin']), sorted(result['groovy'])
 
 
-def do_it(directory, sources_file, resources_file, kotlin_sources_file, groovy_sources_file, include_patterns, exclude_patterns, resolve_kotlin, resolve_groovy, append, all_resources):
-    j, r, k, g = resolve_java_srcs(directory, include_patterns, exclude_patterns, all_resources, resolve_kotlin, resolve_groovy)
+def do_it(
+    directory,
+    sources_file,
+    resources_file,
+    kotlin_sources_file,
+    groovy_sources_file,
+    include_patterns,
+    exclude_patterns,
+    resolve_kotlin,
+    resolve_groovy,
+    append,
+    all_resources,
+):
+    j, r, k, g = resolve_java_srcs(
+        directory, include_patterns, exclude_patterns, all_resources, resolve_kotlin, resolve_groovy
+    )
     mode = 'a' if append else 'w'
     open(sources_file, mode).writelines(i + '\n' for i in j)
     open(resources_file, mode).writelines(i + '\n' for i in r)
@@ -96,8 +117,8 @@ if __name__ == '__main__':
     parser.add_argument('-k', '--kotlin-sources-file', default=None)
     parser.add_argument('-g', '--groovy-sources-file', default=None)
     parser.add_argument('--append', action='store_true', default=False)
-    parser.add_argument('--all-resources',  action='store_true', default=False)
-    parser.add_argument('--resolve-kotlin',  action='store_true', default=False)
+    parser.add_argument('--all-resources', action='store_true', default=False)
+    parser.add_argument('--resolve-kotlin', action='store_true', default=False)
     parser.add_argument('--resolve-groovy', action='store_true', default=False)
     parser.add_argument('--include-patterns', nargs='*', default=[])
     parser.add_argument('--exclude-patterns', nargs='*', default=[])