mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-11 13:43:37 +00:00
0d9fbc1ad7
This version of SickBeard uses both TVDB and TVRage to search and gather it's series data from allowing you to now have access to and download shows that you couldn't before because of being locked into only what TheTVDB had to offer. Also this edition is based off the code we used in our XEM editon so it does come with scene numbering support as well as all the other features our XEM edition has to offer. Please before using this with your existing database (sickbeard.db) please make a backup copy of it and delete any other database files such as cache.db and failed.db if present, we HIGHLY recommend starting out with no database files at all to make this a fresh start but the choice is at your own risk! Enjoy!
144 lines
4.1 KiB
Python
144 lines
4.1 KiB
Python
import os, sys, time
|
|
import lib.hachoir_core.config as config
|
|
from lib.hachoir_core.i18n import _
|
|
|
|
class Log:
|
|
LOG_INFO = 0
|
|
LOG_WARN = 1
|
|
LOG_ERROR = 2
|
|
|
|
level_name = {
|
|
LOG_WARN: "[warn]",
|
|
LOG_ERROR: "[err!]",
|
|
LOG_INFO: "[info]"
|
|
}
|
|
|
|
def __init__(self):
|
|
self.__buffer = {}
|
|
self.__file = None
|
|
self.use_print = True
|
|
self.use_buffer = False
|
|
self.on_new_message = None # Prototype: def func(level, prefix, text, context)
|
|
|
|
def shutdown(self):
|
|
if self.__file:
|
|
self._writeIntoFile(_("Stop Hachoir"))
|
|
|
|
def setFilename(self, filename, append=True):
|
|
"""
|
|
Use a file to store all messages. The
|
|
UTF-8 encoding will be used. Write an informative
|
|
message if the file can't be created.
|
|
|
|
@param filename: C{L{string}}
|
|
"""
|
|
|
|
# Look if file already exists or not
|
|
filename = os.path.expanduser(filename)
|
|
filename = os.path.realpath(filename)
|
|
append = os.access(filename, os.F_OK)
|
|
|
|
# Create log file (or open it in append mode, if it already exists)
|
|
try:
|
|
import codecs
|
|
if append:
|
|
self.__file = codecs.open(filename, "a", "utf-8")
|
|
else:
|
|
self.__file = codecs.open(filename, "w", "utf-8")
|
|
self._writeIntoFile(_("Starting Hachoir"))
|
|
except IOError, err:
|
|
if err.errno == 2:
|
|
self.__file = None
|
|
self.info(_("[Log] setFilename(%s) fails: no such file") % filename)
|
|
else:
|
|
raise
|
|
|
|
def _writeIntoFile(self, message):
|
|
timestamp = time.strftime("%Y-%m-%d %H:%M:%S")
|
|
self.__file.write(u"%s - %s\n" % (timestamp, message))
|
|
self.__file.flush()
|
|
|
|
def newMessage(self, level, text, ctxt=None):
|
|
"""
|
|
Write a new message : append it in the buffer,
|
|
display it to the screen (if needed), and write
|
|
it in the log file (if needed).
|
|
|
|
@param level: Message level.
|
|
@type level: C{int}
|
|
@param text: Message content.
|
|
@type text: C{str}
|
|
@param ctxt: The caller instance.
|
|
"""
|
|
|
|
if level < self.LOG_ERROR and config.quiet or \
|
|
level <= self.LOG_INFO and not config.verbose:
|
|
return
|
|
if config.debug:
|
|
from lib.hachoir_core.error import getBacktrace
|
|
backtrace = getBacktrace(None)
|
|
if backtrace:
|
|
text += "\n\n" + backtrace
|
|
|
|
_text = text
|
|
if hasattr(ctxt, "_logger"):
|
|
_ctxt = ctxt._logger()
|
|
if _ctxt is not None:
|
|
text = "[%s] %s" % (_ctxt, text)
|
|
|
|
# Add message to log buffer
|
|
if self.use_buffer:
|
|
if not self.__buffer.has_key(level):
|
|
self.__buffer[level] = [text]
|
|
else:
|
|
self.__buffer[level].append(text)
|
|
|
|
# Add prefix
|
|
prefix = self.level_name.get(level, "[info]")
|
|
|
|
# Display on stdout (if used)
|
|
if self.use_print:
|
|
sys.stdout.flush()
|
|
sys.stderr.write("%s %s\n" % (prefix, text))
|
|
sys.stderr.flush()
|
|
|
|
# Write into outfile (if used)
|
|
if self.__file:
|
|
self._writeIntoFile("%s %s" % (prefix, text))
|
|
|
|
# Use callback (if used)
|
|
if self.on_new_message:
|
|
self.on_new_message (level, prefix, _text, ctxt)
|
|
|
|
def info(self, text):
|
|
"""
|
|
New informative message.
|
|
@type text: C{str}
|
|
"""
|
|
self.newMessage(Log.LOG_INFO, text)
|
|
|
|
def warning(self, text):
|
|
"""
|
|
New warning message.
|
|
@type text: C{str}
|
|
"""
|
|
self.newMessage(Log.LOG_WARN, text)
|
|
|
|
def error(self, text):
|
|
"""
|
|
New error message.
|
|
@type text: C{str}
|
|
"""
|
|
self.newMessage(Log.LOG_ERROR, text)
|
|
|
|
log = Log()
|
|
|
|
class Logger(object):
|
|
def _logger(self):
|
|
return "<%s>" % self.__class__.__name__
|
|
def info(self, text):
|
|
log.newMessage(Log.LOG_INFO, text, self)
|
|
def warning(self, text):
|
|
log.newMessage(Log.LOG_WARN, text, self)
|
|
def error(self, text):
|
|
log.newMessage(Log.LOG_ERROR, text, self)
|