-
Notifications
You must be signed in to change notification settings - Fork 0
/
channel_check.py
53 lines (43 loc) · 1.34 KB
/
channel_check.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import os
import sys
import yaml
outlawed_channels = [
'defaults',
]
def check_file(filename):
"""Load YAML file and check for outlawed channels under top-level 'channels' key"""
reject = False
try:
with open(filename, 'r') as file:
data = yaml.safe_load(file)
except Exception as e:
print(f"Error loading '{filename}': {e}")
return reject
if data:
if 'channels' in data:
for channel in data['channels']:
if channel in outlawed_channels:
print(f"Channel '{channel}' is not allowed in '{filename}'")
reject = True
return reject
def check_files(filenames):
"""Check multiple files"""
reject = False
for filename in filenames:
reject |= check_file(filename)
return reject
def recurse_folders(folder):
"""Recursively find all YAML files"""
for root, dirs, files in os.walk(folder):
for file in files:
if file.endswith('.yml') or file.endswith('.yaml'):
yield os.path.join(root, file)
if __name__ == '__main__':
print("Checking for outlawed channels")
if len(sys.argv) < 2:
print("Usage: channel_check.py <folder>")
sys.exit(1)
folder = sys.argv[1]
reject = check_files(recurse_folders(folder))
if reject:
sys.exit(1)