mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-03-31 18:46:53 +00:00
93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
import argparse
|
|
import json
|
|
from typing import Optional
|
|
from typing import Sequence
|
|
from pathlib import Path
|
|
|
|
|
|
def _check_duplicate_entry(json_entries, pkeys):
|
|
""" Check duplicate entry based on pkey criteria.
|
|
|
|
:param json_entries: List of json entries
|
|
:param pkeys: List of Primary keys
|
|
:return: list of duplicated entry pkey value tuples
|
|
"""
|
|
unique_entries = set()
|
|
duplicate_entries = set()
|
|
for entry in json_entries:
|
|
pkey_value_tuple = tuple(entry[pkey] for pkey in pkeys)
|
|
if pkey_value_tuple not in unique_entries:
|
|
unique_entries.add(pkey_value_tuple)
|
|
else:
|
|
duplicate_entries.add(pkey_value_tuple)
|
|
return duplicate_entries, len(duplicate_entries)
|
|
|
|
|
|
def main(argv: Optional[Sequence[str]] = None) -> int:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('filenames', nargs='*', type=str,
|
|
help='Names of the JSON files to check duplicate entries'
|
|
)
|
|
table_uuid_mapping = {
|
|
'action': ['uuid'],
|
|
'env_property_group': ['uuid'],
|
|
'environment': ['uuid'],
|
|
'environment_property': ['code'],
|
|
'report_summary': ['uuid'],
|
|
'runner': ['uuid'],
|
|
'scenario': ['uuid'],
|
|
'sla': ['uuid'],
|
|
'sla_scenario_association': ['sla', 'scenario'],
|
|
'tag': ['uuid'],
|
|
'tag_action_association': ['tag_uuid', 'action_uuid'],
|
|
'tag_case_association': ['test_case_uuid', 'tag_uuid'],
|
|
'teams': ['uuid'],
|
|
'test_case': ['uuid'],
|
|
'test_suit': ['uuid'],
|
|
'test_supported_version': ['test_case_uuid', 'version'],
|
|
'testcase_workload_association': ['uuid'],
|
|
'user': ['uuid'],
|
|
'user_tokens': ['user_token'],
|
|
'workflow_task': ['workflow_id'],
|
|
'context': ['uuid'],
|
|
'test_sla_association': ['test_case', 'sla'],
|
|
'teams_association': ['user_uuid', 'team_uuid'],
|
|
'teams_resource_permission': ['team_uuid', 'resource_name'],
|
|
'label': ['uuid'],
|
|
'authentication_config_rules': ['auth_type'],
|
|
'authentication': ['uuid'],
|
|
'user_authentication_association':
|
|
['user_uuid', 'authentication_uuid'],
|
|
}
|
|
|
|
args = vars(parser.parse_args(argv))
|
|
filenames = args['filenames']
|
|
flag = False
|
|
|
|
for i in range(len(filenames)):
|
|
json_file = filenames[i]
|
|
file_name = Path(filenames[i]).stem
|
|
if file_name not in table_uuid_mapping:
|
|
print(
|
|
f"Table {file_name} has no primary key specified to validate "
|
|
f"duplicate entries. Please update the plugin code in "
|
|
f"https://git.voereir.io/voereir/pre-commit-hooks"
|
|
)
|
|
continue
|
|
|
|
primary_keys = table_uuid_mapping[file_name]
|
|
with open(json_file, encoding='UTF-8') as f:
|
|
json_entries = json.load(f)
|
|
duplicate_entries, status = _check_duplicate_entry(
|
|
json_entries, primary_keys)
|
|
|
|
if status:
|
|
print(f"Duplicate entries found - {duplicate_entries} in file "
|
|
f"{json_file}")
|
|
flag = True
|
|
|
|
return flag
|
|
|
|
|
|
if __name__ == "__main__":
|
|
exit(main())
|