File size: 2,379 Bytes
475a69a |
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 111 112 113 114 |
# -*- coding: UTF-8 -*-
# collecting settings to here
import json
import os
import modules.scripts as scripts
from . import util
name = "setting.json"
path = os.path.join(scripts.basedir(), name)
data = {
"model":{
"max_size_preview": True,
"skip_nsfw_preview": False
},
"general":{
"open_url_with_js": True,
"always_display": False,
"show_btn_on_thumb": True,
"proxy": "",
},
"tool":{
}
}
# save setting
# return output msg for log
def save():
print("Saving setting to: " + path)
json_data = json.dumps(data, indent=4)
output = ""
#write to file
try:
with open(path, 'w') as f:
f.write(json_data)
except Exception as e:
util.printD("Error when writing file:"+path)
output = str(e)
util.printD(str(e))
return output
output = "Setting saved to: " + path
util.printD(output)
return output
# load setting to global data
def load():
# load data into globel data
global data
util.printD("Load setting from: " + path)
if not os.path.isfile(path):
util.printD("No setting file, use default")
return
json_data = None
with open(path, 'r') as f:
json_data = json.load(f)
# check error
if not json_data:
util.printD("load setting file failed")
return
data = json_data
# check for new key
if "always_display" not in data["general"].keys():
data["general"]["always_display"] = False
if "show_btn_on_thumb" not in data["general"].keys():
data["general"]["show_btn_on_thumb"] = True
if "proxy" not in data["general"].keys():
data["general"]["proxy"] = ""
return
# save setting from parameter
def save_from_input(max_size_preview, skip_nsfw_preview, open_url_with_js, always_display, show_btn_on_thumb, proxy):
global data
data = {
"model":{
"max_size_preview": max_size_preview,
"skip_nsfw_preview": skip_nsfw_preview
},
"general":{
"open_url_with_js": open_url_with_js,
"always_display": always_display,
"show_btn_on_thumb": show_btn_on_thumb,
"proxy": proxy,
},
"tool":{
}
}
output = save()
if not output:
output = ""
return output
|