def getQueuedTorrentsCount(self):
dbObject = Databases(
dbType = "SQLITE3",
databaseSettings = {
'databaseLocation': self.testDatabaseFile
}
)
return dbObject.getQueuedTorrentsCount()
def __init__(self):
# Setup the database object
self.database = Databases(dbType=self.defaultDatabaseType)
self.logger.info('TransmissionClient INIT')
if settings['client']['type'] == 'transmission':
self.logger.debug('TorrentClient Setup')
self.client = Transmission.Client()
else:
self.logger.info('TorrentClient Not Defined')
raise ValueError('Torrent client type not defined!')
def test_addTorrentsToQueue_torrentExists_deleteTorrent(self):
self.removeDatabase()
testTorrent = Torrents.TV(
torrentTitle='some.show.s01e01.720p.junk.here',
url='http://testurl.com/test')
testTorrent2 = Torrents.TV(
torrentTitle='some.show.s01e02.720p.junk.here',
url='http://testurl.com/test2')
torrentQueue = Queue()
torrentQueue.append(testTorrent)
dbObject = Databases(
dbType="SQLITE3",
databaseSettings={'databaseLocation': self.testDatabaseFile})
dbObject.addTorrentsToQueue(torrentQueue)
self.assertTrue(dbObject.torrentExists(testTorrent))
dbObject.updateHashString({'hashString': 'abc123'},
{'url': 'http://testurl.com/test'})
self.assertTrue(
len(
dbObject.getTorrentInfo(hashString='abc123',
selectors=['torrentTitle', 'url'])) ==
1)
torrentQueue = Queue()
torrentQueue.append(testTorrent2)
dbObject.addTorrentsToQueue(torrentQueue)
self.assertTrue(dbObject.getQueuedTorrentsCount() == 2)
self.assertTrue(len(dbObject.getQueuedTorrents()) == 2)
dbObject.deleteTorrent(url='http://testurl.com/test')
self.assertFalse(dbObject.torrentExists(testTorrent))
self.assertTrue(dbObject.getQueuedTorrentsCount() == 1)
dbObject.deleteTorrent(url='http://testurl.com/test2')
self.assertFalse(dbObject.torrentExists(testTorrent))
self.assertTrue(dbObject.getQueuedTorrentsCount() == 0)
self.removeDatabase()
def test_AddBlacklistedTorrent(self):
self.removeDatabase()
dbObject = Databases(
dbType="SQLITE3",
databaseSettings={'databaseLocation': self.testDatabaseFile})
dbObject.addBlacklistedTorrent(url='http://testurl.com/test')
self.assertTrue(
dbObject.torrentBlacklisted(url='http://testurl.com/test'))
self.removeDatabase()
#-------------------------------------------------------------------------------
# Name: Torrent
# Purpose: This module is a generic torrent module that clients should use
# when trying to describe torrents. Status can be added as needed.
#
#
#-------------------------------------------------------------------------------
# -*- coding: utf-8 -*-
# System Includes
import time
from flannelfox.settings import settings
from flannelfox.databases import Databases
# Setup the database object
TorrentDB = Databases(settings['database']['defaultDatabaseEngine'])
class Status(object):
Paused = 0
QueuedForVerification = 1
Verifying = 2
QueuedForDownloading = 3
Downloading = 4
QueuedForSeeding = 5
Seeding = 6
class Torrent(object):
def __init__(self,
hashString=None,
id=None,
def __init__(self, *args): self.database = Databases( dbType = self.defaultDatabaseType ) self.torrentClient = self.setupTorrentClient() self.torrentClient.updateQueue()
def __init__(self, *args):
self.elements = list(*args)
self.database = Databases(dbType=self.defaultDatabaseType)
