You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
148 lines
4.2 KiB
148 lines
4.2 KiB
#!/usr/bin/python3 -OO
|
|
# Copyright 2012-2021 The SABnzbd-Team <team@sabnzbd.org>
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
"""
|
|
util.apireg - Registration of API connection info
|
|
"""
|
|
|
|
import winreg
|
|
|
|
|
|
def reg_info(user):
|
|
"""Return the reg key for API"""
|
|
if user:
|
|
# Normally use the USER part of the registry
|
|
section = winreg.HKEY_CURRENT_USER
|
|
keypath = r"Software\SABnzbd"
|
|
else:
|
|
# A Windows Service will use the service key instead
|
|
section = winreg.HKEY_LOCAL_MACHINE
|
|
keypath = r"SYSTEM\CurrentControlSet\Services\SABnzbd"
|
|
return section, keypath
|
|
|
|
|
|
def get_connection_info(user=True):
|
|
"""Return URL of the API running SABnzbd instance
|
|
'user' == True will first try user's registry, otherwise system is used
|
|
"""
|
|
section, keypath = reg_info(user)
|
|
url = None
|
|
|
|
try:
|
|
hive = winreg.ConnectRegistry(None, section)
|
|
key = winreg.OpenKey(hive, keypath + r"\api")
|
|
for i in range(0, winreg.QueryInfoKey(key)[1]):
|
|
name, value, val_type = winreg.EnumValue(key, i)
|
|
if name == "url":
|
|
url = value
|
|
|
|
winreg.CloseKey(key)
|
|
except OSError:
|
|
pass
|
|
finally:
|
|
winreg.CloseKey(hive)
|
|
|
|
# Nothing in user's registry, try system registry
|
|
if user and not url:
|
|
url = get_connection_info(user=False)
|
|
|
|
return url
|
|
|
|
|
|
def set_connection_info(url, user=True):
|
|
"""Set API info in register"""
|
|
section, keypath = reg_info(user)
|
|
try:
|
|
hive = winreg.ConnectRegistry(None, section)
|
|
try:
|
|
winreg.CreateKey(hive, keypath)
|
|
except OSError:
|
|
pass
|
|
key = winreg.OpenKey(hive, keypath)
|
|
mykey = winreg.CreateKey(key, "api")
|
|
winreg.SetValueEx(mykey, "url", None, winreg.REG_SZ, url)
|
|
winreg.CloseKey(mykey)
|
|
winreg.CloseKey(key)
|
|
except OSError:
|
|
if user:
|
|
set_connection_info(url, user=False)
|
|
finally:
|
|
winreg.CloseKey(hive)
|
|
|
|
|
|
def del_connection_info(user=True):
|
|
"""Remove API info from register"""
|
|
section, keypath = reg_info(user)
|
|
try:
|
|
hive = winreg.ConnectRegistry(None, section)
|
|
key = winreg.OpenKey(hive, keypath)
|
|
winreg.DeleteKey(key, "api")
|
|
winreg.CloseKey(key)
|
|
except OSError:
|
|
if user:
|
|
del_connection_info(user=False)
|
|
finally:
|
|
winreg.CloseKey(hive)
|
|
|
|
|
|
def get_install_lng():
|
|
"""Return language-code used by the installer"""
|
|
lng = 0
|
|
try:
|
|
hive = winreg.ConnectRegistry(None, winreg.HKEY_CURRENT_USER)
|
|
key = winreg.OpenKey(hive, r"Software\SABnzbd")
|
|
for i in range(0, winreg.QueryInfoKey(key)[1]):
|
|
name, value, val_type = winreg.EnumValue(key, i)
|
|
if name == "Installer Language":
|
|
lng = value
|
|
winreg.CloseKey(key)
|
|
except OSError:
|
|
pass
|
|
finally:
|
|
winreg.CloseKey(hive)
|
|
|
|
if lng in LanguageMap:
|
|
return LanguageMap[lng]
|
|
return "en"
|
|
|
|
|
|
# Map from NSIS-codepage to our language-strings
|
|
LanguageMap = {
|
|
"1033": "en",
|
|
"1036": "fr",
|
|
"1031": "de",
|
|
"1043": "nl",
|
|
"1035": "fi",
|
|
"1045": "pl",
|
|
"1053": "sv",
|
|
"1030": "da",
|
|
"2068": "nb",
|
|
"1048": "ro",
|
|
"1034": "es",
|
|
"1046": "pr_BR",
|
|
"3098": "sr",
|
|
"1037": "he",
|
|
"1049": "ru",
|
|
"2052": "zh_CN",
|
|
}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("URL = %s" % get_connection_info())
|
|
print("Language = %s" % get_install_lng())
|
|
# del_connection_info()
|
|
# set_connection_info('localhost', '8080', 'blabla', user=False)
|
|
|