aboutsummaryrefslogtreecommitdiffstats
path: root/settings.py
blob: 32bb1f5994115c7cc8714afd5f84f6ce7810abfb (plain)
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import ast
default_settings = '''route_tor = False
port_number = 80
allow_foreign_addresses = False

# 0 - off by default
# 1 - only manually created subtitles on by default
# 2 - enable even if automatically generated is all that's available
subtitles_mode = 0

# ISO 639 language code: https://en.wikipedia.org/wiki/List_of_ISO_639-1_codes
subtitles_language = "en"

enable_related_videos = True
enable_comments = True
enable_comment_avatars = True

# 0 to sort by top
# 1 to sort by newest
default_comment_sorting = 0

# developer use to debug 403s
gather_googlevideo_domains = False
'''
exec(default_settings)
allowed_targets = set(("route_tor", "port_number", "allow_foreign_addresses", "subtitles_mode", "subtitles_language", "enable_related_videos", "enable_comments", "enable_comment_avatars", "default_comment_sorting", "gather_googlevideo_domains"))

def log_ignored_line(line_number, message):
    print("settings.txt: Ignoring line " + str(node.lineno) + " (" + message + ")")

try:
    with open('settings.txt', 'r', encoding='utf-8') as file:
        settings_text = file.read()
except FileNotFoundError:
    with open('settings.txt', 'a', encoding='utf-8') as file:
        file.write(default_settings)
else:
    attributes = {
        ast.NameConstant: 'value',
        ast.Num: 'n',
        ast.Str: 's',
    }
    module_node = ast.parse(settings_text)
    for node in module_node.body:
        if type(node) != ast.Assign:
            log_ignored_line(node.lineno, "only assignments are allowed")
            continue
        
        if len(node.targets) > 1:
            log_ignored_line(node.lineno, "only simple single-variable assignments allowed")
            continue

        target = node.targets[0]
        if type(target) != ast.Name:
            log_ignored_line(node.lineno, "only simple single-variable assignments allowed")
            continue
        
        if target.id not in allowed_targets:
            log_ignored_line(node.lineno, "target is not a valid setting")
            continue
        
        if type(node.value) not in (ast.NameConstant, ast.Num, ast.Str):
            log_ignored_line(node.lineno, "only literals allowed for values")
            continue

        locals()[target.id] = node.value.__getattribute__(attributes[type(node.value)])