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