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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
import os
import sys
import re
from configparser import ConfigParser
def update_optiscaler_config(file_path):
if not os.path.exists(file_path):
print(f"Error: File '{file_path}' not found.")
return
with open(file_path, 'r') as f:
lines = f.readlines()
config = ConfigParser()
config.optionxform = str # Preserve case for keys (otherwise PATH could match Path)
config.read(file_path)
# Because we want to support unprefixed env variables, we need to count key occurrences across all sections of the ini file
# Keys that appear multiple times should be prefixed like Section_Key by the user for them to be targeted properly
# Normalize section names: strip - and . so V-Sync becomes VSync
# This allows env vars like VSync_Key to match INI section [V-Sync]
def normalize_section(section_name):
return section_name.replace('-', '').replace('.', '')
key_occurrences = {}
key_to_sections = {}
section_normalized_to_actual = {} # Maps normalized section name to actual section name
for section in config.sections():
normalized = normalize_section(section)
section_normalized_to_actual[normalized] = section
for key in config.options(section):
key_occurrences[key] = key_occurrences.get(key, 0) + 1
if key not in key_to_sections:
key_to_sections[key] = []
key_to_sections[key].append(section)
env_updates = []
# Handle OptiScaler_Section_Key format
optiscaler_vars = {k: v for k, v in os.environ.items() if k.startswith("OptiScaler_")}
for env_name, env_value in optiscaler_vars.items():
parts = env_name.split('_', 2)
if len(parts) >= 3:
env_updates.append(('optiscaler', parts[1], parts[2], env_value, env_name))
# Handle Section_Key and Key formats
other_vars = {k: v for k, v in os.environ.items() if not k.startswith("OptiScaler_")}
for env_name, env_value in other_vars.items():
# Try Section_Key format
if '_' in env_name:
parts = env_name.split('_', 1)
section_from_env = parts[0]
key = parts[1]
# Try exact section match first
if config.has_section(section_from_env) and config.has_option(section_from_env, key):
env_updates.append(('section_key', section_from_env, key, env_value, env_name))
continue
# Try section match with normalized section names
if section_from_env in section_normalized_to_actual:
actual_section = section_normalized_to_actual[section_from_env]
if config.has_option(actual_section, key):
env_updates.append(('section_key', actual_section, key, env_value, env_name))
continue
# Try Key format (only if key appears exactly once across all sections)
if env_name in key_occurrences and key_occurrences[env_name] == 1:
section = key_to_sections[env_name][0]
env_updates.append(('key', section, env_name, env_value, env_name))
print(f"Found {len(env_updates)} updates to apply")
for entry in env_updates:
print(f"> {entry}")
for update_type, section_target, key_target, env_value, env_name in env_updates:
found_section = False
# Regex to match [Section] and Key=Value (case-sensitive)
section_pattern = re.compile(rf'^\s*\[{re.escape(section_target)}]\s*')
key_pattern = re.compile(rf'^(\s*{re.escape(key_target)}\s*)=.*')
for i, line in enumerate(lines):
# Track if we are inside the correct section
if section_pattern.match(line):
found_section = True
continue
# If we hit a new section before finding the key, the key doesn't exist in the target section
if found_section and line.strip().startswith('[') and not section_pattern.match(line):
break
# Replace the value if the key is found within the correct section
if found_section and key_pattern.match(line):
lines[i] = key_pattern.sub(r'\1=' + env_value, line)
print(f"Updated: [{section_target}] {key_target} = {env_value} (from {env_name})")
break
# Write the modified content back
with open(file_path, 'w') as f:
f.writelines(lines)
if __name__ == "__main__":
if len(sys.argv) < 2:
print("Usage: python update-optiscaler-config.py <path_to_ini>")
else:
update_optiscaler_config(sys.argv[1])
|
