Add html export support
This commit is contained in:
Erik Snoeijs
committed by
cncfanatics
cncfanatics
parent
447e0c9349
commit
3270432f0b
@@ -1,182 +1,215 @@
|
||||
#===============================================================================
|
||||
# Copyright (C) 2010 Diego Duclos
|
||||
#
|
||||
# This file is part of pyfa.
|
||||
#
|
||||
# pyfa 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 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# pyfa 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 pyfa. If not, see <http://www.gnu.org/licenses/>.
|
||||
#===============================================================================
|
||||
|
||||
import cPickle
|
||||
import os.path
|
||||
import config
|
||||
import urllib2
|
||||
|
||||
class SettingsProvider():
|
||||
BASE_PATH = os.path.join(config.savePath, "settings")
|
||||
settings = {}
|
||||
_instance = None
|
||||
@classmethod
|
||||
def getInstance(cls):
|
||||
if cls._instance == None:
|
||||
cls._instance = SettingsProvider()
|
||||
|
||||
return cls._instance
|
||||
|
||||
def __init__(self):
|
||||
if not os.path.exists(self.BASE_PATH):
|
||||
os.mkdir(self.BASE_PATH);
|
||||
|
||||
def getSettings(self, area, defaults=None):
|
||||
s = self.settings.get(area)
|
||||
if s is None:
|
||||
p = os.path.join(self.BASE_PATH, area)
|
||||
|
||||
if not os.path.exists(p):
|
||||
info = {}
|
||||
if defaults:
|
||||
for item in defaults:
|
||||
info[item] = defaults[item]
|
||||
|
||||
else:
|
||||
try:
|
||||
f = open(p, "rb")
|
||||
info = cPickle.load(f)
|
||||
for item in defaults:
|
||||
if item not in info:
|
||||
info[item] = defaults[item]
|
||||
|
||||
except:
|
||||
info = {}
|
||||
if defaults:
|
||||
for item in defaults:
|
||||
info[item] = defaults[item]
|
||||
|
||||
self.settings[area] = s = Settings(p, info)
|
||||
|
||||
return s
|
||||
|
||||
def saveAll(self):
|
||||
for settings in self.settings.itervalues():
|
||||
settings.save()
|
||||
|
||||
class Settings():
|
||||
def __init__(self, location, info):
|
||||
self.location = location
|
||||
self.info = info
|
||||
|
||||
def save(self):
|
||||
f = open(self.location, "wb")
|
||||
cPickle.dump(self.info, f, cPickle.HIGHEST_PROTOCOL)
|
||||
|
||||
def __getitem__(self, k):
|
||||
return self.info[k]
|
||||
|
||||
def __setitem__(self, k, v):
|
||||
self.info[k] = v
|
||||
|
||||
def __iter__(self):
|
||||
return self.info.__iter__()
|
||||
|
||||
def iterkeys(self):
|
||||
return self.info.iterkeys()
|
||||
|
||||
def itervalues(self):
|
||||
return self.info.itervalues()
|
||||
|
||||
def iteritems(self):
|
||||
return self.info.iteritems()
|
||||
|
||||
def keys(self):
|
||||
return self.info.keys()
|
||||
|
||||
def values(self):
|
||||
return self.info.values()
|
||||
|
||||
def items(self):
|
||||
return self.info.items()
|
||||
|
||||
class ProxySettings():
|
||||
_instance = None
|
||||
@classmethod
|
||||
def getInstance(cls):
|
||||
if cls._instance == None:
|
||||
cls._instance = ProxySettings()
|
||||
|
||||
return cls._instance
|
||||
|
||||
def __init__(self):
|
||||
|
||||
# mode
|
||||
# 0 - No proxy
|
||||
# 1 - Auto-detected proxy settings
|
||||
# 2 - Manual proxy settings
|
||||
serviceProxyDefaultSettings = {"mode": 1, "type": "https", "address": "", "port": ""}
|
||||
|
||||
self.serviceProxySettings = SettingsProvider.getInstance().getSettings("pyfaServiceProxySettings", serviceProxyDefaultSettings)
|
||||
|
||||
def getMode(self):
|
||||
return self.serviceProxySettings["mode"]
|
||||
|
||||
def getAddress(self):
|
||||
return self.serviceProxySettings["address"]
|
||||
|
||||
def getPort(self):
|
||||
return self.serviceProxySettings["port"]
|
||||
|
||||
def getType(self):
|
||||
return self.serviceProxySettings["type"]
|
||||
|
||||
def setMode(self, mode):
|
||||
self.serviceProxySettings["mode"] = mode
|
||||
|
||||
def setAddress(self, addr):
|
||||
self.serviceProxySettings["address"] = addr
|
||||
|
||||
def setPort(self, port):
|
||||
self.serviceProxySettings["port"] = port
|
||||
|
||||
def setType(self, type):
|
||||
self.serviceProxySettings["type"] = type
|
||||
|
||||
def autodetect(self):
|
||||
|
||||
proxy = None
|
||||
proxAddr = proxPort = ""
|
||||
proxydict = urllib2.ProxyHandler().proxies
|
||||
txt = "Auto-detected: "
|
||||
|
||||
validPrefixes = ("http", "https")
|
||||
|
||||
for prefix in validPrefixes:
|
||||
if not prefix in proxydict:
|
||||
continue
|
||||
proxyline = proxydict[prefix]
|
||||
proto = "{0}://".format(prefix)
|
||||
if proxyline[:len(proto)] == proto:
|
||||
proxyline = proxyline[len(proto):]
|
||||
proxAddr, proxPort = proxyline.split(":")
|
||||
proxPort = int(proxPort.rstrip("/"))
|
||||
proxy = (proxAddr, proxPort)
|
||||
break
|
||||
|
||||
return proxy
|
||||
|
||||
def getProxySettings(self):
|
||||
|
||||
if self.getMode() == 0:
|
||||
return None
|
||||
if self.getMode() == 1:
|
||||
return self.autodetect()
|
||||
if self.getMode() == 2:
|
||||
return (self.getAddress(), int(self.getPort()))
|
||||
#===============================================================================
|
||||
# Copyright (C) 2010 Diego Duclos
|
||||
#
|
||||
# This file is part of pyfa.
|
||||
#
|
||||
# pyfa 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 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# pyfa 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 pyfa. If not, see <http://www.gnu.org/licenses/>.
|
||||
#===============================================================================
|
||||
|
||||
import cPickle
|
||||
import os.path
|
||||
import config
|
||||
import urllib2
|
||||
|
||||
class SettingsProvider():
|
||||
BASE_PATH = os.path.join(config.savePath, "settings")
|
||||
settings = {}
|
||||
_instance = None
|
||||
@classmethod
|
||||
def getInstance(cls):
|
||||
if cls._instance == None:
|
||||
cls._instance = SettingsProvider()
|
||||
|
||||
return cls._instance
|
||||
|
||||
def __init__(self):
|
||||
if not os.path.exists(self.BASE_PATH):
|
||||
os.mkdir(self.BASE_PATH);
|
||||
|
||||
def getSettings(self, area, defaults=None):
|
||||
s = self.settings.get(area)
|
||||
if s is None:
|
||||
p = os.path.join(self.BASE_PATH, area)
|
||||
|
||||
if not os.path.exists(p):
|
||||
info = {}
|
||||
if defaults:
|
||||
for item in defaults:
|
||||
info[item] = defaults[item]
|
||||
|
||||
else:
|
||||
try:
|
||||
f = open(p, "rb")
|
||||
info = cPickle.load(f)
|
||||
for item in defaults:
|
||||
if item not in info:
|
||||
info[item] = defaults[item]
|
||||
|
||||
except:
|
||||
info = {}
|
||||
if defaults:
|
||||
for item in defaults:
|
||||
info[item] = defaults[item]
|
||||
|
||||
self.settings[area] = s = Settings(p, info)
|
||||
|
||||
return s
|
||||
|
||||
def saveAll(self):
|
||||
for settings in self.settings.itervalues():
|
||||
settings.save()
|
||||
|
||||
class Settings():
|
||||
def __init__(self, location, info):
|
||||
self.location = location
|
||||
self.info = info
|
||||
|
||||
def save(self):
|
||||
f = open(self.location, "wb")
|
||||
cPickle.dump(self.info, f, cPickle.HIGHEST_PROTOCOL)
|
||||
|
||||
def __getitem__(self, k):
|
||||
return self.info[k]
|
||||
|
||||
def __setitem__(self, k, v):
|
||||
self.info[k] = v
|
||||
|
||||
def __iter__(self):
|
||||
return self.info.__iter__()
|
||||
|
||||
def iterkeys(self):
|
||||
return self.info.iterkeys()
|
||||
|
||||
def itervalues(self):
|
||||
return self.info.itervalues()
|
||||
|
||||
def iteritems(self):
|
||||
return self.info.iteritems()
|
||||
|
||||
def keys(self):
|
||||
return self.info.keys()
|
||||
|
||||
def values(self):
|
||||
return self.info.values()
|
||||
|
||||
def items(self):
|
||||
return self.info.items()
|
||||
|
||||
|
||||
|
||||
class ProxySettings():
|
||||
_instance = None
|
||||
@classmethod
|
||||
def getInstance(cls):
|
||||
if cls._instance == None:
|
||||
cls._instance = ProxySettings()
|
||||
|
||||
return cls._instance
|
||||
|
||||
def __init__(self):
|
||||
|
||||
# mode
|
||||
# 0 - No proxy
|
||||
# 1 - Auto-detected proxy settings
|
||||
# 2 - Manual proxy settings
|
||||
serviceProxyDefaultSettings = {"mode": 1, "type": "https", "address": "", "port": ""}
|
||||
|
||||
self.serviceProxySettings = SettingsProvider.getInstance().getSettings("pyfaServiceProxySettings", serviceProxyDefaultSettings)
|
||||
|
||||
def getMode(self):
|
||||
return self.serviceProxySettings["mode"]
|
||||
|
||||
def getAddress(self):
|
||||
return self.serviceProxySettings["address"]
|
||||
|
||||
def getPort(self):
|
||||
return self.serviceProxySettings["port"]
|
||||
|
||||
def getType(self):
|
||||
return self.serviceProxySettings["type"]
|
||||
|
||||
def setMode(self, mode):
|
||||
self.serviceProxySettings["mode"] = mode
|
||||
|
||||
def setAddress(self, addr):
|
||||
self.serviceProxySettings["address"] = addr
|
||||
|
||||
def setPort(self, port):
|
||||
self.serviceProxySettings["port"] = port
|
||||
|
||||
def setType(self, type):
|
||||
self.serviceProxySettings["type"] = type
|
||||
|
||||
def autodetect(self):
|
||||
|
||||
proxy = None
|
||||
proxAddr = proxPort = ""
|
||||
proxydict = urllib2.ProxyHandler().proxies
|
||||
txt = "Auto-detected: "
|
||||
|
||||
validPrefixes = ("http", "https")
|
||||
|
||||
for prefix in validPrefixes:
|
||||
if not prefix in proxydict:
|
||||
continue
|
||||
proxyline = proxydict[prefix]
|
||||
proto = "{0}://".format(prefix)
|
||||
if proxyline[:len(proto)] == proto:
|
||||
proxyline = proxyline[len(proto):]
|
||||
proxAddr, proxPort = proxyline.split(":")
|
||||
proxPort = int(proxPort.rstrip("/"))
|
||||
proxy = (proxAddr, proxPort)
|
||||
break
|
||||
|
||||
return proxy
|
||||
|
||||
def getProxySettings(self):
|
||||
|
||||
if self.getMode() == 0:
|
||||
return None
|
||||
if self.getMode() == 1:
|
||||
return self.autodetect()
|
||||
if self.getMode() == 2:
|
||||
return (self.getAddress(), int(self.getPort()))
|
||||
|
||||
|
||||
|
||||
"""
|
||||
Settings used by the HTML export feature.
|
||||
"""
|
||||
class HTMLExportSettings():
|
||||
_instance = None
|
||||
|
||||
@classmethod
|
||||
def getInstance(cls):
|
||||
if cls._instance == None:
|
||||
cls._instance = HTMLExportSettings()
|
||||
|
||||
return cls._instance
|
||||
|
||||
def __init__(self):
|
||||
serviceHTMLExportDefaultSettings = {"enabled": False, "path": config.pyfaPath + os.sep + 'pyfaFits.html' }
|
||||
self.serviceHTMLExportSettings = SettingsProvider.getInstance().getSettings("pyfaServiceHTMLExportSettings", serviceHTMLExportDefaultSettings)
|
||||
|
||||
def getEnabled(self):
|
||||
return self.serviceHTMLExportSettings["enabled"]
|
||||
|
||||
def setEnabled(self, enabled):
|
||||
self.serviceHTMLExportSettings["enabled"] = enabled
|
||||
|
||||
def getPath(self):
|
||||
return self.serviceHTMLExportSettings["path"]
|
||||
|
||||
def setPath(self, path):
|
||||
self.serviceHTMLExportSettings["path"] = path
|
||||
Reference in New Issue
Block a user