mirror of
https://gitlab.com/kelteseth/ScreenPlay.git
synced 2024-11-20 01:34:04 +01:00
64 lines
2.6 KiB
Python
64 lines
2.6 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
# SPDX-License-Identifier: LicenseRef-EliasSteurerTachiom OR AGPL-3.0-only
|
||
|
|
import os
|
||
|
|
from pathlib import Path
|
||
|
|
import util
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
def find_files(path, pattern, file_extensions):
|
||
|
|
files = []
|
||
|
|
for root, dirnames, filenames in os.walk(path):
|
||
|
|
for filename in filenames:
|
||
|
|
if any(filename.endswith(ext) for ext in file_extensions):
|
||
|
|
file_path = os.path.join(root, filename)
|
||
|
|
with open(file_path) as f:
|
||
|
|
contents = f.read()
|
||
|
|
if any(substring in contents for substring in pattern):
|
||
|
|
files.append(file_path)
|
||
|
|
print(f"Found file {file_path}")
|
||
|
|
return files
|
||
|
|
|
||
|
|
|
||
|
|
def check_moc_include(filename, add_include=False):
|
||
|
|
cppfile = Path(filename)
|
||
|
|
# replace cppfile ending with .cpp
|
||
|
|
cppfile = cppfile.with_suffix(".cpp").resolve()
|
||
|
|
# check if cpp file exists
|
||
|
|
if not cppfile.exists():
|
||
|
|
# check if cpp file exists but 3 directories up and then in src
|
||
|
|
cppfile = cppfile.joinpath(
|
||
|
|
"../../../../src/{}".format(cppfile.name)).resolve()
|
||
|
|
if not cppfile.exists():
|
||
|
|
print(f"FAIL: {cppfile} does not exist")
|
||
|
|
return
|
||
|
|
print(f"Check {cppfile}")
|
||
|
|
|
||
|
|
with open(cppfile) as f:
|
||
|
|
if '#include "moc_{}"'.format(os.path.basename(cppfile)) in f.read():
|
||
|
|
print("OK: {} HAS moc include".format(cppfile))
|
||
|
|
else:
|
||
|
|
print("FAIL: {} MISSES moc include".format(cppfile))
|
||
|
|
if add_include:
|
||
|
|
with open(cppfile, 'a') as f:
|
||
|
|
f.write("\n#include \"moc_{}\"\n".format(
|
||
|
|
os.path.basename(cppfile)))
|
||
|
|
print("Adding include...")
|
||
|
|
else:
|
||
|
|
print("Add this line to the end of the file:")
|
||
|
|
print("#include \"moc_{}\"".format(os.path.basename(cppfile)))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
parser = argparse.ArgumentParser(
|
||
|
|
description="Check if moc includes are present and add them if requested.")
|
||
|
|
parser.add_argument('-a', '--addinclude', action='store_true',
|
||
|
|
help='Automatically add missing moc includes')
|
||
|
|
args = parser.parse_args()
|
||
|
|
repo_root = util.repo_root_path()
|
||
|
|
filenames = find_files(repo_root, ['Q_OBJECT', 'Q_GADGET', 'Q_NAMESPACE'], [
|
||
|
|
'.cpp', '.h', "main.cpp"])
|
||
|
|
for filename in filenames:
|
||
|
|
check_moc_include(filename, args.addinclude)
|
||
|
|
if not args.addinclude:
|
||
|
|
print("Run this script with the '-a' or '--addinclude' flag to automatically add the moc includes")
|