mirror of
https://gitlab.com/kelteseth/ScreenPlay.git
synced 2024-11-22 10:42:29 +01:00
74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
#!/usr/bin/python3
|
|
# SPDX-License-Identifier: LicenseRef-EliasSteurerTachiom OR AGPL-3.0-only
|
|
import subprocess
|
|
import sys
|
|
import os
|
|
from multiprocessing import cpu_count
|
|
from multiprocessing import Pool
|
|
from multiprocessing import dummy
|
|
from pathlib import Path
|
|
from sys import stdout
|
|
|
|
stdout.reconfigure(encoding='utf-8')
|
|
|
|
|
|
def find_all_files(path):
|
|
file_list = []
|
|
for root, dirnames, files in os.walk(path):
|
|
for filename in files:
|
|
file_list.append(os.path.join(root, filename))
|
|
return file_list
|
|
|
|
|
|
def find_all_git_staged_files():
|
|
process = subprocess.run("git diff --name-only --staged",
|
|
capture_output=True, shell=True)
|
|
out = process.stdout.decode("utf-8")
|
|
out = out.splitlines()
|
|
return out
|
|
|
|
|
|
def find_files(file_endings_tuple, path="", git_staged_only=False):
|
|
file_list = []
|
|
# Get the root folder by moving one up
|
|
root = Path(__file__).parent.absolute()
|
|
root = os.path.abspath(os.path.join(root, "../"))
|
|
|
|
root = os.path.join(root, path)
|
|
print(root)
|
|
|
|
if git_staged_only:
|
|
file_list = find_all_git_staged_files()
|
|
else:
|
|
file_list = find_all_files(root)
|
|
|
|
filtered_file_list = []
|
|
for filename in file_list:
|
|
if filename.endswith(file_endings_tuple):
|
|
filtered_file_list.append(os.path.join(root, filename))
|
|
|
|
return filtered_file_list
|
|
|
|
|
|
def execute_threaded(file_list, format_file_function):
|
|
p = Pool(cpu_count())
|
|
p.map(format_file_function, file_list)
|
|
p.close()
|
|
p.join()
|
|
|
|
|
|
def check_git_exit(caller_name):
|
|
# Check if all files are formatter
|
|
process = subprocess.run("git --no-pager diff",
|
|
capture_output=True, shell=True)
|
|
|
|
out = process.stdout.decode("utf-8")
|
|
|
|
if out != "":
|
|
print("\n########### %s DONE ###########\n" % caller_name)
|
|
print("Git diff is not empty. This means your files where not correctly formatted!")
|
|
out.replace("\\n", "\n")
|
|
# print(out)
|
|
sys.exit(1)
|
|
|