forked from s0md3v/roop
-
Notifications
You must be signed in to change notification settings - Fork 554
/
settings.py
69 lines (56 loc) · 2.68 KB
/
settings.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
54
55
56
57
58
59
60
61
62
63
64
65
66
import yaml
class Settings:
def __init__(self, config_file):
self.config_file = config_file
self.load()
def default_get(_, data, name, default):
value = default
try:
value = data.get(name, default)
except:
pass
return value
def load(self):
try:
with open(self.config_file, 'r') as f:
data = yaml.load(f, Loader=yaml.FullLoader)
except:
data = None
self.selected_theme = self.default_get(data, 'selected_theme', "Default")
self.server_name = self.default_get(data, 'server_name', "")
self.server_port = self.default_get(data, 'server_port', 0)
self.server_share = self.default_get(data, 'server_share', False)
self.output_image_format = self.default_get(data, 'output_image_format', 'png')
self.output_video_format = self.default_get(data, 'output_video_format', 'mp4')
self.output_video_codec = self.default_get(data, 'output_video_codec', 'libx264')
self.video_quality = self.default_get(data, 'video_quality', 14)
self.clear_output = self.default_get(data, 'clear_output', True)
self.max_threads = self.default_get(data, 'max_threads', 2)
self.memory_limit = self.default_get(data, 'memory_limit', 0)
self.provider = self.default_get(data, 'provider', 'cuda')
self.force_cpu = self.default_get(data, 'force_cpu', False)
self.output_template = self.default_get(data, 'output_template', '{file}_{time}')
self.use_os_temp_folder = self.default_get(data, 'use_os_temp_folder', False)
self.output_show_video = self.default_get(data, 'output_show_video', True)
self.launch_browser = self.default_get(data, 'launch_browser', True)
def save(self):
data = {
'selected_theme': self.selected_theme,
'server_name': self.server_name,
'server_port': self.server_port,
'server_share': self.server_share,
'output_image_format' : self.output_image_format,
'output_video_format' : self.output_video_format,
'output_video_codec' : self.output_video_codec,
'video_quality' : self.video_quality,
'clear_output' : self.clear_output,
'max_threads' : self.max_threads,
'memory_limit' : self.memory_limit,
'provider' : self.provider,
'force_cpu' : self.force_cpu,
'output_template' : self.output_template,
'use_os_temp_folder' : self.use_os_temp_folder,
'output_show_video' : self.output_show_video
}
with open(self.config_file, 'w') as f:
yaml.dump(data, f)