Browse Source
Bugfix in metadata code for XBMC 12+ to display proper episodeguide urls. Misc tweaks here and there.tags/release_0.1.0
echel0n
11 years ago
12 changed files with 331 additions and 97 deletions
@ -1,29 +0,0 @@ |
|||
import sys |
|||
import logging |
|||
import traceback |
|||
|
|||
from indexer_api import indexerApi |
|||
from indexer_exceptions import indexer_exception |
|||
|
|||
# Set our common indexer_api options here |
|||
INDEXER_API_PARMS = {'apikey': '9DAF49C96CBF8DAC', |
|||
'language': 'en', |
|||
'useZip': True} |
|||
|
|||
|
|||
INDEXER_API_PARMS['indexer'] = 'Tvdb' |
|||
lindexer_api_parms = INDEXER_API_PARMS.copy() |
|||
|
|||
try: |
|||
imdbid = " " |
|||
|
|||
t = indexerApi().config['valid_languages'] |
|||
t = indexerApi(**lindexer_api_parms) |
|||
myEp = t[258171] |
|||
|
|||
if getattr(myEp, 'seriesname', None) is not None: |
|||
print "FOUND" |
|||
|
|||
except indexer_exception as e: |
|||
print e |
|||
pass |
|||
@ -0,0 +1 @@ |
|||
__author__ = 'Justin' |
|||
@ -0,0 +1,51 @@ |
|||
from __future__ import with_statement |
|||
|
|||
import unittest |
|||
|
|||
import sqlite3 |
|||
|
|||
import sys |
|||
import os.path |
|||
sys.path.append(os.path.abspath('..')) |
|||
sys.path.append(os.path.abspath('../lib')) |
|||
|
|||
import sickbeard |
|||
import shutil |
|||
|
|||
from sickbeard import encodingKludge as ek, providers, tvcache |
|||
from sickbeard import db |
|||
from sickbeard.databases import mainDB |
|||
from sickbeard.databases import cache_db |
|||
|
|||
|
|||
from indexer_api import indexerApi |
|||
from indexer_exceptions import indexer_exception |
|||
|
|||
class APICheck(unittest.TestCase): |
|||
indexer_id = 258171 |
|||
indexer = 'Tvdb' |
|||
# Set our common indexer_api options here |
|||
INDEXER_API_PARMS = {'apikey': '9DAF49C96CBF8DAC', |
|||
'language': 'en', |
|||
'useZip': True} |
|||
|
|||
|
|||
INDEXER_API_PARMS['indexer'] = indexer |
|||
lindexer_api_parms = INDEXER_API_PARMS.copy() |
|||
|
|||
try: |
|||
imdbid = " " |
|||
showurl = indexerApi(**lindexer_api_parms).config['base_url'] + indexer_id + '/all/en.zip' |
|||
t = indexerApi().config['valid_languages'] |
|||
t = indexerApi(**lindexer_api_parms) |
|||
myEp = t[258171] |
|||
|
|||
if getattr(myEp, 'seriesname', None) is not None: |
|||
print "FOUND" |
|||
|
|||
except indexer_exception as e: |
|||
print e |
|||
pass |
|||
|
|||
if __name__ == "__main__": |
|||
unittest.main() |
|||
@ -0,0 +1,227 @@ |
|||
# coding=UTF-8 |
|||
# Author: Dennis Lutter <lad1337@gmail.com> |
|||
# URL: http://code.google.com/p/sickbeard/ |
|||
# |
|||
# This file is part of Sick Beard. |
|||
# |
|||
# Sick Beard 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. |
|||
# |
|||
# Sick Beard 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 Sick Beard. If not, see <http://www.gnu.org/licenses/>. |
|||
|
|||
from __future__ import with_statement |
|||
|
|||
import unittest |
|||
|
|||
import sqlite3 |
|||
|
|||
import sys |
|||
import os.path |
|||
sys.path.append(os.path.abspath('..')) |
|||
sys.path.append(os.path.abspath('../lib')) |
|||
|
|||
import sickbeard |
|||
import shutil |
|||
|
|||
from sickbeard import encodingKludge as ek, providers, tvcache |
|||
from sickbeard import db |
|||
from sickbeard.databases import mainDB |
|||
from sickbeard.databases import cache_db |
|||
|
|||
#================= |
|||
# test globals |
|||
#================= |
|||
TESTDIR = os.path.abspath('.') |
|||
TESTDBNAME = "sickbeard.db" |
|||
TESTCACHEDBNAME = "cache.db" |
|||
|
|||
|
|||
SHOWNAME = u"show name" |
|||
SEASON = 4 |
|||
EPISODE = 2 |
|||
FILENAME = u"show name - s0" + str(SEASON) + "e0" + str(EPISODE) + ".mkv" |
|||
FILEDIR = os.path.join(TESTDIR, SHOWNAME) |
|||
FILEPATH = os.path.join(FILEDIR, FILENAME) |
|||
|
|||
SHOWDIR = os.path.join(TESTDIR, SHOWNAME + " final") |
|||
|
|||
#sickbeard.logger.sb_log_instance = sickbeard.logger.SBRotatingLogHandler(os.path.join(TESTDIR, 'sickbeard.log'), sickbeard.logger.NUM_LOGS, sickbeard.logger.LOG_SIZE) |
|||
sickbeard.logger.SBRotatingLogHandler.log_file = os.path.join(os.path.join(TESTDIR, 'Logs'), 'test_sickbeard.log') |
|||
|
|||
|
|||
#================= |
|||
# prepare env functions |
|||
#================= |
|||
def createTestLogFolder(): |
|||
if not os.path.isdir(sickbeard.LOG_DIR): |
|||
os.mkdir(sickbeard.LOG_DIR) |
|||
|
|||
# call env functions at appropriate time during sickbeard var setup |
|||
|
|||
#================= |
|||
# sickbeard globals |
|||
#================= |
|||
sickbeard.SYS_ENCODING = 'UTF-8' |
|||
sickbeard.showList = [] |
|||
sickbeard.QUALITY_DEFAULT = 4 # hdtv |
|||
sickbeard.FLATTEN_FOLDERS_DEFAULT = 0 |
|||
|
|||
sickbeard.NAMING_PATTERN = '' |
|||
sickbeard.NAMING_ABD_PATTERN = '' |
|||
sickbeard.NAMING_MULTI_EP = 1 |
|||
|
|||
|
|||
sickbeard.PROVIDER_ORDER = ["sick_beard_index"] |
|||
sickbeard.newznabProviderList = providers.getNewznabProviderList("Sick Beard Index|http://lolo.sickbeard.com/|0|5030,5040|0!!!NZBs.org|http://nzbs.org/||5030,5040,5070,5090|0!!!Usenet-Crawler|http://www.usenet-crawler.com/||5030,5040|0") |
|||
sickbeard.providerList = providers.makeProviderList() |
|||
|
|||
sickbeard.PROG_DIR = os.path.abspath('..') |
|||
sickbeard.DATA_DIR = sickbeard.PROG_DIR |
|||
sickbeard.LOG_DIR = os.path.join(TESTDIR, 'Logs') |
|||
createTestLogFolder() |
|||
sickbeard.logger.sb_log_instance.initLogging(False) |
|||
|
|||
|
|||
#================= |
|||
# dummy functions |
|||
#================= |
|||
def _dummy_saveConfig(): |
|||
return True |
|||
# this overrides the sickbeard save_config which gets called during a db upgrade |
|||
# this might be considered a hack |
|||
mainDB.sickbeard.save_config = _dummy_saveConfig |
|||
|
|||
|
|||
# the real one tries to contact tvdb just stop it from getting more info on the ep |
|||
def _fake_specifyEP(self, season, episode): |
|||
pass |
|||
|
|||
sickbeard.tv.TVEpisode.specifyEpisode = _fake_specifyEP |
|||
|
|||
|
|||
#================= |
|||
# test classes |
|||
#================= |
|||
class SickbeardTestDBCase(unittest.TestCase): |
|||
def setUp(self): |
|||
sickbeard.showList = [] |
|||
setUp_test_db() |
|||
setUp_test_episode_file() |
|||
setUp_test_show_dir() |
|||
|
|||
def tearDown(self): |
|||
sickbeard.showList = [] |
|||
tearDown_test_db() |
|||
tearDown_test_episode_file() |
|||
tearDown_test_show_dir() |
|||
|
|||
|
|||
class TestDBConnection(db.DBConnection, object): |
|||
|
|||
def __init__(self, dbFileName=TESTDBNAME): |
|||
dbFileName = os.path.join(TESTDIR, dbFileName) |
|||
super(TestDBConnection, self).__init__(dbFileName) |
|||
|
|||
|
|||
class TestCacheDBConnection(TestDBConnection, object): |
|||
|
|||
def __init__(self, providerName): |
|||
db.DBConnection.__init__(self, os.path.join(TESTDIR, TESTCACHEDBNAME)) |
|||
|
|||
# Create the table if it's not already there |
|||
try: |
|||
sql = "CREATE TABLE " + providerName + " (name TEXT, season NUMERIC, episodes TEXT, indexerid NUMERIC, url TEXT, time NUMERIC, quality TEXT);" |
|||
self.connection.execute(sql) |
|||
self.connection.commit() |
|||
except sqlite3.OperationalError, e: |
|||
if str(e) != "table " + providerName + " already exists": |
|||
raise |
|||
|
|||
# Create the table if it's not already there |
|||
try: |
|||
sql = "CREATE TABLE lastUpdate (provider TEXT, time NUMERIC);" |
|||
self.connection.execute(sql) |
|||
self.connection.commit() |
|||
except sqlite3.OperationalError, e: |
|||
if str(e) != "table lastUpdate already exists": |
|||
raise |
|||
|
|||
# this will override the normal db connection |
|||
sickbeard.db.DBConnection = TestDBConnection |
|||
sickbeard.tvcache.CacheDBConnection = TestCacheDBConnection |
|||
|
|||
|
|||
#================= |
|||
# test functions |
|||
#================= |
|||
def setUp_test_db(): |
|||
"""upgrades the db to the latest version |
|||
""" |
|||
# upgrading the db |
|||
db.upgradeDatabase(db.DBConnection(), mainDB.InitialSchema) |
|||
# fix up any db problems |
|||
db.sanityCheckDatabase(db.DBConnection(), mainDB.MainSanityCheck) |
|||
|
|||
#and for cache.b too |
|||
db.upgradeDatabase(db.DBConnection("cache.db"), cache_db.InitialSchema) |
|||
|
|||
|
|||
def tearDown_test_db(): |
|||
"""Deletes the test db |
|||
although this seams not to work on my system it leaves me with an zero kb file |
|||
""" |
|||
# uncomment next line so leave the db intact between test and at the end |
|||
#return False |
|||
if os.path.exists(os.path.join(TESTDIR, TESTDBNAME)): |
|||
os.remove(os.path.join(TESTDIR, TESTDBNAME)) |
|||
if os.path.exists(os.path.join(TESTDIR, TESTCACHEDBNAME)): |
|||
os.remove(os.path.join(TESTDIR, TESTCACHEDBNAME)) |
|||
|
|||
|
|||
def setUp_test_episode_file(): |
|||
if not os.path.exists(FILEDIR): |
|||
os.makedirs(FILEDIR) |
|||
|
|||
try: |
|||
with open(FILEPATH, 'w') as f: |
|||
f.write("foo bar") |
|||
except EnvironmentError: |
|||
print "Unable to set up test episode" |
|||
raise |
|||
|
|||
|
|||
def tearDown_test_episode_file(): |
|||
shutil.rmtree(FILEDIR) |
|||
|
|||
|
|||
def setUp_test_show_dir(): |
|||
if not os.path.exists(SHOWDIR): |
|||
os.makedirs(SHOWDIR) |
|||
|
|||
|
|||
def tearDown_test_show_dir(): |
|||
shutil.rmtree(SHOWDIR) |
|||
|
|||
tearDown_test_db() |
|||
|
|||
if __name__ == '__main__': |
|||
print "==================" |
|||
print "Dont call this directly" |
|||
print "==================" |
|||
print "you might want to call" |
|||
|
|||
dirList = os.listdir(TESTDIR) |
|||
for fname in dirList: |
|||
if (fname.find("_test") > 0) and (fname.find("pyc") < 0): |
|||
print "- " + fname |
|||
|
|||
print "==================" |
|||
print "or just call all_tests.py" |
|||
Loading…
Reference in new issue