Python TVEpisode.saveToDB方法代码示例

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
 def test_init_empty_db(self):
     show = TVShow(1, 1, "en")
     ep = TVEpisode(show, 1, 1)
     ep.name = "asdasdasdajkaj"
     ep.saveToDB()
     ep.loadFromDB(1, 1)
     self.assertEqual(ep.name, "asdasdasdajkaj")

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
 def test_init_empty_db(self):
     """
     test init empty db
     """
     show = TVShow(1, 1, "en")
     episode = TVEpisode(show, 1, 1)
     episode.name = "asdasdasdajkaj"
     episode.saveToDB()
     episode.loadFromDB(1, 1)
     self.assertEqual(episode.name, "asdasdasdajkaj")

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
    def test_process(self):
        show = TVShow(3)
        show.name = test.SHOWNAME
        show.location = test.SHOWDIR
        show.saveToDB()

        sickbeard.showList = [show]
        ep = TVEpisode(show, test.SEASON, test.EPISODE)
        ep.name = "some ep name"
        ep.saveToDB()

        pp = PostProcessor(test.FILEPATH)
        self.assertTrue(pp.process())

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
    def test_process(self):
        show = TVShow(1, 3)
        show.name = SHOWNAME
        show.location = SHOWDIR
        show.saveToDB()

        sickbeard.showList = [show]
        ep = TVEpisode(show, SEASON, EPISODE)
        ep.name = "some ep name"
        ep.saveToDB()

        addNameToCache('show name', 3)
        self.pp = PostProcessor(FILEPATH, process_method='move')
        self.assertTrue(self.pp.process())

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
    def test_process(self):
        show = TVShow(1,3)
        show.name = test.SHOWNAME
        show.location = test.SHOWDIR
        show.saveToDB()

        sickbeard.showList = [show]
        ep = TVEpisode(show, test.SEASON, test.EPISODE)
        ep.name = "some ep name"
        ep.saveToDB()

        addNameToCache('show name', 3)
        sickbeard.PROCESS_METHOD = 'move'

        pp = PostProcessor(test.FILEPATH)
        self.assertTrue(pp.process())

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
    def test(self):
        global searchItems
        searchItems = curData["i"]
        show = TVShow(tvdbdid)
        show.name = show_name
        show.quality = curData["q"]
        show.saveToDB()
        sickbeard.showList.append(show)

        for epNumber in curData["e"]:
            episode = TVEpisode(show, curData["s"], epNumber)
            episode.status = c.WANTED
            episode.saveToDB()

        bestResult = search.findEpisode(episode, forceSearch)
        if not bestResult:
            self.assertEqual(curData["b"], bestResult)
        self.assertEqual(curData["b"], bestResult.name) #first is expected, second is choosen one

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
    def test_process(self):
        """
        Test process
        """
        show = TVShow(1, 3)
        show.name = test.SHOW_NAME
        show.location = test.SHOW_DIR
        show.saveToDB()

        sickbeard.showList = [show]
        episode = TVEpisode(show, test.SEASON, test.EPISODE)
        episode.name = "some episode name"
        episode.saveToDB()

        addNameToCache('show name', 3)
        sickbeard.PROCESS_METHOD = 'move'

        post_processor = PostProcessor(test.FILE_PATH)
        self.assertTrue(post_processor.process())

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
    def test(self):
        global searchItems
        searchItems = curData['i']
        show = TVShow(1, tvdbdid)
        show.name = show_name
        show.quality = curData['q']
        show.saveToDB()
        sickbeard.showList.append(show)
        episode = None

        for epNumber in curData['e']:
            episode = TVEpisode(show, curData['s'], epNumber)
            episode.status = c.WANTED
            episode.saveToDB()

        bestResult = search.search_providers(show, episode.season, episode.episode, forceSearch)
        if not bestResult:
            self.assertEqual(curData['b'], bestResult)
        self.assertEqual(curData['b'], bestResult.name) #first is expected, second is choosen one

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
    def do_test():
        """
        Test to perform
        """
        global search_items  # pylint: disable=global-statement
        search_items = cur_data["i"]
        show = TVShow(1, tvdb_id)
        show.name = show_name
        show.quality = cur_data["q"]
        show.saveToDB()
        sickbeard.showList.append(show)
        episode = None

        for epNumber in cur_data["e"]:
            episode = TVEpisode(show, cur_data["s"], epNumber)
            episode.status = common.WANTED
            episode.saveToDB()

        best_result = search.searchProviders(show, episode.episode, force_search)
        if not best_result:
            assert cur_data["b"] == best_result
        # pylint: disable=no-member
        assert cur_data["b"] == best_result.name  # first is expected, second is chosen one

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
    def setUp(self):
        sickbeard.showList = []
        setup_test_db()
        setup_test_episode_file()
        setup_test_show_dir()
        setup_test_processing_dir()

        show = TVShow(1, 0001, 'en')
        show.name = SHOW_NAME
        show.location = FILE_DIR

        show.episodes = {}
        for season in range(1, NUM_SEASONS):
            show.episodes[season] = {}
            for episode in range(1, EPISODES_PER_SEASON):
                if season == SEASON and episode == EPISODE:
                    episode = TVEpisode(show, season, episode, ep_file=FILE_PATH)
                else:
                    episode = TVEpisode(show, season, episode)
                show.episodes[season][episode] = episode
                episode.saveToDB()

        show.saveToDB()
        sickbeard.showList = [show]

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
def _on_failed_torrent(key, removeFromRunningTorrents=True, markEpisodesWanted=False):
    rTorr = _find_running_torrent_by_field('key', key)
    if not rTorr:
        logger.log(u'Failed to locate torrent with key "%s"' % (key), logger.MESSAGE)
        return False

    if rTorr['blacklistOrigUrlOnFailure'] and rTorr['originalTorrentUrl']:
        _blacklist_torrent_url(rTorr['originalTorrentUrl'])

    if markEpisodesWanted:
        if rTorr['episodes']:
            for ep in rTorr['episodes']:
                # fucked up no?  We need to do this b/c there's no way to *refresh* from the db without
                # actually creating a new TVEpisode object!
                epTemp = TVEpisode(show=ep.show, season=ep.season, episode=ep.episode)
                if epTemp.status in Quality.SNATCHED + Quality.SNATCHED_PROPER:
                    logger.log(u'Changing episode %s status from SNATCHED to WANTED' % (epTemp.prettyName()),
                               logger.MESSAGE)
                    epTemp.status = WANTED
                    epTemp.saveToDB()
                else:
                    logger.log(u'NOT Changing episode %s status to WANTED b/c current '
                               'status is not SNATCHED (actual status is %s)' % (
                                        epTemp.prettyName(), str(epTemp.status)),
                               logger.MESSAGE)
        else:
            logger.log(u'Cannot markEpisodesWanted b/c entry has no episodes',
                   logger.DEBUG)
    else:
        logger.log(u'Not marking episodes as wanted b/c markEpisodesWanted was False',
                   logger.DEBUG)

    if removeFromRunningTorrents:
        _remove_torrent_by_handle(rTorr['handle'], deleteFilesToo=True)

    return True

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
    def do_test():
        """
        Test to perform
        """
        show = TVShow(1, int(cur_data["tvdbid"]))
        show.name = cur_name
        show.quality = common.ANY | common.Quality.UNKNOWN | common.Quality.RAWHDTV
        show.saveToDB()
        sickbeard.showList.append(show)

        for ep_number in cur_data["e"]:
            episode = TVEpisode(show, cur_data["s"], ep_number)
            episode.status = common.WANTED

            # We aren't updating scene numbers, so fake it here
            episode.scene_season = cur_data["s"]
            episode.scene_episode = ep_number

            episode.saveToDB()

            cur_provider.show = show
            season_strings = cur_provider._get_season_search_strings(episode)  # pylint: disable=protected-access
            episode_strings = cur_provider._get_episode_search_strings(episode)  # pylint: disable=protected-access

            fail = False
            cur_string = ''
            for cur_string in season_strings, episode_strings:
                if not all([isinstance(cur_string, list), isinstance(cur_string[0], dict)]):
                    print " %s is using a wrong string format!" % cur_provider.name
                    print cur_string
                    fail = True
                    continue

            if fail:
                continue

            try:
                assert season_strings == cur_data["s_strings"]
                assert episode_strings == cur_data["e_strings"]
            except AssertionError:
                print " %s is using a wrong string format!" % cur_provider.name
                print cur_string
                continue

            search_strings = episode_strings[0]
            # search_strings.update(season_strings[0])
            # search_strings.update({"RSS":['']})

            # print search_strings

            if not cur_provider.public:
                continue

            items = cur_provider._doSearch(search_strings)  # pylint: disable=protected-access
            if not items:
                print "No results from cur_provider?"
                continue

            title, url = cur_provider._get_title_and_url(items[0])  # pylint: disable=protected-access
            for word in show.name.split(" "):
                if not word.lower() in title.lower():
                    print "Show cur_name not in title: %s. URL: %s" % (title, url)
                    continue

            if not url:
                print "url is empty"
                continue

            quality = cur_provider.getQuality(items[0])
            size = cur_provider._get_size(items[0])  # pylint: disable=protected-access

            if not show.quality & quality:
                print "Quality not in common.ANY, %r %s" % (quality, size)
                continue

# 需要导入模块: from sickbeard.tv import TVEpisode [as 别名]
# 或者: from sickbeard.tv.TVEpisode import saveToDB [as 别名]
    def test(self):
        show = TVShow(1, int(curData["tvdbid"]))
        show.name = name
        show.quality = c.ANY | c.Quality.UNKNOWN | c.Quality.RAWHDTV
        show.saveToDB()
        sickbeard.showList.append(show)

        for epNumber in curData["e"]:
            episode = TVEpisode(show, curData["s"], epNumber)
            episode.status = c.WANTED

            # We arent updating scene numbers, so fake it here
            episode.scene_season = curData["s"]
            episode.scene_episode = epNumber

            episode.saveToDB()

            provider.show = show
            season_strings = provider._get_season_search_strings(episode)
            episode_strings = provider._get_episode_search_strings(episode)

            fail = False
            for cur_string in season_strings, episode_strings:
                if not all([isinstance(cur_string, list), isinstance(cur_string[0], dict)]):
                    print " %s is using a wrong string format!" % provider.name
                    print cur_string
                    fail = True
                    continue

            if fail:
                continue

            try:
                assert(season_strings == curData["s_strings"])
                assert(episode_strings == curData["e_strings"])
            except AssertionError:
                print " %s is using a wrong string format!" % provider.name
                print cur_string
                continue

            search_strings = episode_strings[0]
            #search_strings.update(season_strings[0])
            #search_strings.update({"RSS":['']})

            #print search_strings

            if not provider.public:
                continue

            items = provider._doSearch(search_strings)
            if not items:
                print "No results from provider?"
                continue

            title, url = provider._get_title_and_url(items[0])
            for word in show.name.split(" "):
                if not word.lower() in title.lower():
                    print "Show name not in title: %s. URL: %s" % (title, url)
                    continue

            if not url:
                print "url is empty"
                continue

            quality = provider.getQuality(items[0])
            size = provider._get_size(items[0])
            if not show.quality & quality:
                print "Quality not in common.ANY, %r" % quality
                continue