mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-25 17:28:50 -05: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!
261 lines
8.4 KiB
Python
261 lines
8.4 KiB
Python
"""
|
|
Parent of all (field) classes in Hachoir: Field.
|
|
"""
|
|
|
|
from lib.hachoir_core.compatibility import reversed
|
|
from lib.hachoir_core.stream import InputFieldStream
|
|
from lib.hachoir_core.error import HachoirError, HACHOIR_ERRORS
|
|
from lib.hachoir_core.log import Logger
|
|
from lib.hachoir_core.i18n import _
|
|
from lib.hachoir_core.tools import makePrintable
|
|
from weakref import ref as weakref_ref
|
|
|
|
class FieldError(HachoirError):
|
|
"""
|
|
Error raised by a L{Field}.
|
|
|
|
@see: L{HachoirError}
|
|
"""
|
|
pass
|
|
|
|
def joinPath(path, name):
|
|
if path != "/":
|
|
return "/".join((path, name))
|
|
else:
|
|
return "/%s" % name
|
|
|
|
class MissingField(KeyError, FieldError):
|
|
def __init__(self, field, key):
|
|
KeyError.__init__(self)
|
|
self.field = field
|
|
self.key = key
|
|
|
|
def __str__(self):
|
|
return 'Can\'t get field "%s" from %s' % (self.key, self.field.path)
|
|
|
|
def __unicode__(self):
|
|
return u'Can\'t get field "%s" from %s' % (self.key, self.field.path)
|
|
|
|
class Field(Logger):
|
|
# static size can have two differents value: None (no static size), an
|
|
# integer (number of bits), or a function which returns an integer.
|
|
#
|
|
# This function receives exactly the same arguments than the constructor
|
|
# except the first one (one). Example of function:
|
|
# static_size = staticmethod(lambda *args, **kw: args[1])
|
|
static_size = None
|
|
|
|
# Indicate if this field contains other fields (is a field set) or not
|
|
is_field_set = False
|
|
|
|
def __init__(self, parent, name, size=None, description=None):
|
|
"""
|
|
Set default class attributes, set right address if None address is
|
|
given.
|
|
|
|
@param parent: Parent field of this field
|
|
@type parent: L{Field}|None
|
|
@param name: Name of the field, have to be unique in parent. If it ends
|
|
with "[]", end will be replaced with "[new_id]" (eg. "raw[]"
|
|
becomes "raw[0]", next will be "raw[1]", and then "raw[2]", etc.)
|
|
@type name: str
|
|
@param size: Size of the field in bit (can be None, so it
|
|
will be computed later)
|
|
@type size: int|None
|
|
@param address: Address in bit relative to the parent absolute address
|
|
@type address: int|None
|
|
@param description: Optional string description
|
|
@type description: str|None
|
|
"""
|
|
assert issubclass(parent.__class__, Field)
|
|
assert (size is None) or (0 <= size)
|
|
self._parent = parent
|
|
self._name = name
|
|
self._address = parent.nextFieldAddress()
|
|
self._size = size
|
|
self._description = description
|
|
|
|
def _logger(self):
|
|
return self.path
|
|
|
|
def createDescription(self):
|
|
return ""
|
|
def _getDescription(self):
|
|
if self._description is None:
|
|
try:
|
|
self._description = self.createDescription()
|
|
if isinstance(self._description, str):
|
|
self._description = makePrintable(
|
|
self._description, "ISO-8859-1", to_unicode=True)
|
|
except HACHOIR_ERRORS, err:
|
|
self.error("Error getting description: " + unicode(err))
|
|
self._description = ""
|
|
return self._description
|
|
description = property(_getDescription,
|
|
doc="Description of the field (string)")
|
|
|
|
def __str__(self):
|
|
return self.display
|
|
def __unicode__(self):
|
|
return self.display
|
|
def __repr__(self):
|
|
return "<%s path=%r, address=%s, size=%s>" % (
|
|
self.__class__.__name__, self.path, self._address, self._size)
|
|
|
|
def hasValue(self):
|
|
return self._getValue() is not None
|
|
def createValue(self):
|
|
raise NotImplementedError()
|
|
def _getValue(self):
|
|
try:
|
|
value = self.createValue()
|
|
except HACHOIR_ERRORS, err:
|
|
self.error(_("Unable to create value: %s") % unicode(err))
|
|
value = None
|
|
self._getValue = lambda: value
|
|
return value
|
|
value = property(lambda self: self._getValue(), doc="Value of field")
|
|
|
|
def _getParent(self):
|
|
return self._parent
|
|
parent = property(_getParent, doc="Parent of this field")
|
|
|
|
def createDisplay(self):
|
|
return unicode(self.value)
|
|
def _getDisplay(self):
|
|
if not hasattr(self, "_Field__display"):
|
|
try:
|
|
self.__display = self.createDisplay()
|
|
except HACHOIR_ERRORS, err:
|
|
self.error("Unable to create display: %s" % err)
|
|
self.__display = u""
|
|
return self.__display
|
|
display = property(lambda self: self._getDisplay(),
|
|
doc="Short (unicode) string which represents field content")
|
|
|
|
def createRawDisplay(self):
|
|
value = self.value
|
|
if isinstance(value, str):
|
|
return makePrintable(value, "ASCII", to_unicode=True)
|
|
else:
|
|
return unicode(value)
|
|
def _getRawDisplay(self):
|
|
if not hasattr(self, "_Field__raw_display"):
|
|
try:
|
|
self.__raw_display = self.createRawDisplay()
|
|
except HACHOIR_ERRORS, err:
|
|
self.error("Unable to create raw display: %s" % err)
|
|
self.__raw_display = u""
|
|
return self.__raw_display
|
|
raw_display = property(lambda self: self._getRawDisplay(),
|
|
doc="(Unicode) string which represents raw field content")
|
|
|
|
def _getName(self):
|
|
return self._name
|
|
name = property(_getName,
|
|
doc="Field name (unique in its parent field set list)")
|
|
|
|
def _getIndex(self):
|
|
if not self._parent:
|
|
return None
|
|
return self._parent.getFieldIndex(self)
|
|
index = property(_getIndex)
|
|
|
|
def _getPath(self):
|
|
if not self._parent:
|
|
return '/'
|
|
names = []
|
|
field = self
|
|
while field:
|
|
names.append(field._name)
|
|
field = field._parent
|
|
names[-1] = ''
|
|
return '/'.join(reversed(names))
|
|
path = property(_getPath,
|
|
doc="Full path of the field starting at root field")
|
|
|
|
def _getAddress(self):
|
|
return self._address
|
|
address = property(_getAddress,
|
|
doc="Relative address in bit to parent address")
|
|
|
|
def _getAbsoluteAddress(self):
|
|
address = self._address
|
|
current = self._parent
|
|
while current:
|
|
address += current._address
|
|
current = current._parent
|
|
return address
|
|
absolute_address = property(_getAbsoluteAddress,
|
|
doc="Absolute address (from stream beginning) in bit")
|
|
|
|
def _getSize(self):
|
|
return self._size
|
|
size = property(_getSize, doc="Content size in bit")
|
|
|
|
def _getField(self, name, const):
|
|
if name.strip("."):
|
|
return None
|
|
field = self
|
|
for index in xrange(1, len(name)):
|
|
field = field._parent
|
|
if field is None:
|
|
break
|
|
return field
|
|
|
|
def getField(self, key, const=True):
|
|
if key:
|
|
if key[0] == "/":
|
|
if self._parent:
|
|
current = self._parent.root
|
|
else:
|
|
current = self
|
|
if len(key) == 1:
|
|
return current
|
|
key = key[1:]
|
|
else:
|
|
current = self
|
|
for part in key.split("/"):
|
|
field = current._getField(part, const)
|
|
if field is None:
|
|
raise MissingField(current, part)
|
|
current = field
|
|
return current
|
|
raise KeyError("Key must not be an empty string!")
|
|
|
|
def __getitem__(self, key):
|
|
return self.getField(key, False)
|
|
|
|
def __contains__(self, key):
|
|
try:
|
|
return self.getField(key, False) is not None
|
|
except FieldError:
|
|
return False
|
|
|
|
def _createInputStream(self, **args):
|
|
assert self._parent
|
|
return InputFieldStream(self, **args)
|
|
def getSubIStream(self):
|
|
if hasattr(self, "_sub_istream"):
|
|
stream = self._sub_istream()
|
|
else:
|
|
stream = None
|
|
if stream is None:
|
|
stream = self._createInputStream()
|
|
self._sub_istream = weakref_ref(stream)
|
|
return stream
|
|
def setSubIStream(self, createInputStream):
|
|
cis = self._createInputStream
|
|
self._createInputStream = lambda **args: createInputStream(cis, **args)
|
|
|
|
def __nonzero__(self):
|
|
"""
|
|
Method called by code like "if field: (...)".
|
|
Always returns True
|
|
"""
|
|
return True
|
|
|
|
def getFieldType(self):
|
|
return self.__class__.__name__
|
|
|