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)