mirror of
https://github.com/moparisthebest/SickRage
synced 2024-08-13 16:53:54 -04: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!
142 lines
4.7 KiB
Python
142 lines
4.7 KiB
Python
import gettext
|
|
_ = gettext.gettext
|
|
|
|
try:
|
|
from types import ModuleType
|
|
except:
|
|
from new import module as ModuleType
|
|
import copy
|
|
import re
|
|
|
|
import _base
|
|
from html5lib.constants import voidElements
|
|
|
|
tag_regexp = re.compile("{([^}]*)}(.*)")
|
|
|
|
moduleCache = {}
|
|
|
|
def getETreeModule(ElementTreeImplementation):
|
|
name = "_" + ElementTreeImplementation.__name__+"builder"
|
|
if name in moduleCache:
|
|
return moduleCache[name]
|
|
else:
|
|
mod = ModuleType("_" + ElementTreeImplementation.__name__+"builder")
|
|
objs = getETreeBuilder(ElementTreeImplementation)
|
|
mod.__dict__.update(objs)
|
|
moduleCache[name] = mod
|
|
return mod
|
|
|
|
def getETreeBuilder(ElementTreeImplementation):
|
|
ElementTree = ElementTreeImplementation
|
|
|
|
class TreeWalker(_base.NonRecursiveTreeWalker):
|
|
"""Given the particular ElementTree representation, this implementation,
|
|
to avoid using recursion, returns "nodes" as tuples with the following
|
|
content:
|
|
|
|
1. The current element
|
|
|
|
2. The index of the element relative to its parent
|
|
|
|
3. A stack of ancestor elements
|
|
|
|
4. A flag "text", "tail" or None to indicate if the current node is a
|
|
text node; either the text or tail of the current element (1)
|
|
"""
|
|
def getNodeDetails(self, node):
|
|
if isinstance(node, tuple): # It might be the root Element
|
|
elt, key, parents, flag = node
|
|
if flag in ("text", "tail"):
|
|
return _base.TEXT, getattr(elt, flag)
|
|
else:
|
|
node = elt
|
|
|
|
if not(hasattr(node, "tag")):
|
|
node = node.getroot()
|
|
|
|
if node.tag in ("<DOCUMENT_ROOT>", "<DOCUMENT_FRAGMENT>"):
|
|
return (_base.DOCUMENT,)
|
|
|
|
elif node.tag == "<!DOCTYPE>":
|
|
return (_base.DOCTYPE, node.text,
|
|
node.get("publicId"), node.get("systemId"))
|
|
|
|
elif node.tag == ElementTree.Comment:
|
|
return _base.COMMENT, node.text
|
|
|
|
else:
|
|
assert type(node.tag) in (str, unicode), type(node.tag)
|
|
#This is assumed to be an ordinary element
|
|
match = tag_regexp.match(node.tag)
|
|
if match:
|
|
namespace, tag = match.groups()
|
|
else:
|
|
namespace = None
|
|
tag = node.tag
|
|
attrs = {}
|
|
for name, value in node.attrib.items():
|
|
match = tag_regexp.match(name)
|
|
if match:
|
|
attrs[(match.group(1),match.group(2))] = value
|
|
else:
|
|
attrs[(None,name)] = value
|
|
return (_base.ELEMENT, namespace, tag,
|
|
attrs, len(node) or node.text)
|
|
|
|
def getFirstChild(self, node):
|
|
if isinstance(node, tuple):
|
|
element, key, parents, flag = node
|
|
else:
|
|
element, key, parents, flag = node, None, [], None
|
|
|
|
if flag in ("text", "tail"):
|
|
return None
|
|
else:
|
|
if element.text:
|
|
return element, key, parents, "text"
|
|
elif len(element):
|
|
parents.append(element)
|
|
return element[0], 0, parents, None
|
|
else:
|
|
return None
|
|
|
|
def getNextSibling(self, node):
|
|
if isinstance(node, tuple):
|
|
element, key, parents, flag = node
|
|
else:
|
|
return None
|
|
|
|
if flag == "text":
|
|
if len(element):
|
|
parents.append(element)
|
|
return element[0], 0, parents, None
|
|
else:
|
|
return None
|
|
else:
|
|
if element.tail and flag != "tail":
|
|
return element, key, parents, "tail"
|
|
elif key < len(parents[-1]) - 1:
|
|
return parents[-1][key+1], key+1, parents, None
|
|
else:
|
|
return None
|
|
|
|
def getParentNode(self, node):
|
|
if isinstance(node, tuple):
|
|
element, key, parents, flag = node
|
|
else:
|
|
return None
|
|
|
|
if flag == "text":
|
|
if not parents:
|
|
return element
|
|
else:
|
|
return element, key, parents, None
|
|
else:
|
|
parent = parents.pop()
|
|
if not parents:
|
|
return parent
|
|
else:
|
|
return parent, list(parents[-1]).index(parent), parents, None
|
|
|
|
return locals()
|