mirror of
https://github.com/evilhero/mylar
synced 2024-12-24 08:42:37 +00:00
Redo of #1495
This commit is contained in:
Undeadhunter
evilhero
parent
0961f45ed0
commit
8f4a6797fa
1 changed files with 126 additions and 49 deletions
|
|
@ -21,73 +21,150 @@ class TorrentClient(object):
|
|||
|
||||
|
||||
#if username and password:
|
||||
logger.info('Connecting to ' + host + ':' + portnr + ' Username: ' + username + ' Password: ' + password )
|
||||
self.client = DelugeRPCClient(host,int(portnr),username,password)
|
||||
self.client.connect()
|
||||
|
||||
logger.info('connected? ' + str(self.client.connected))
|
||||
|
||||
return self.client
|
||||
|
||||
# logger.info('Connecting to ' + host + ':' + portnr + ' Username: ' + username + ' Password: ' + password )
|
||||
try:
|
||||
self.client = DelugeRPCClient(host,int(portnr),username,password)
|
||||
except Exception as e:
|
||||
logger.error('Could not create DelugeRPCClient Object' + e)
|
||||
return False
|
||||
else:
|
||||
try:
|
||||
self.client.connect()
|
||||
except Exception as e:
|
||||
logger.error('Could not connect to Deluge ' + host)
|
||||
else:
|
||||
return self.client
|
||||
|
||||
def find_torrent(self, hash):
|
||||
#To be Added
|
||||
return False
|
||||
logger.debug('Finding Torrent hash: ' + hash)
|
||||
torrent_info = self.get_torrent(hash)
|
||||
if torrent_info:
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
def get_torrent(self, torrent):
|
||||
# To be Added
|
||||
return False
|
||||
def get_torrent(self, hash):
|
||||
logger.debug('Getting Torrent info hash: ' + hash)
|
||||
try:
|
||||
torrent_info = self.client.call('core.get_torrent_status', hash, '')
|
||||
except Exception as e:
|
||||
logger.error('Could not get torrent info for ' + hash)
|
||||
return False
|
||||
else:
|
||||
logger.info('Getting Torrent Info!')
|
||||
return torrent_info
|
||||
|
||||
def start_torrent(self, torrent):
|
||||
return torrent.start()
|
||||
|
||||
def stop_torrent(self, torrent):
|
||||
return torrent.stop()
|
||||
def start_torrent(self, hash):
|
||||
try:
|
||||
self.find_torrent(hash)
|
||||
except Exception as e:
|
||||
return False
|
||||
else:
|
||||
try:
|
||||
self.client.call('core.resume_torrent', hash)
|
||||
except Exception as e:
|
||||
logger.error('Torrent failed to start ' + e)
|
||||
else:
|
||||
logger.info('Torrent ' + hash + ' was started')
|
||||
return True
|
||||
|
||||
def stop_torrent(self, hash):
|
||||
try:
|
||||
self.client.find_torrent(hash)
|
||||
except Exception as e:
|
||||
logger.error('Torrent Not Found')
|
||||
return False
|
||||
else:
|
||||
try:
|
||||
self.client.call('core.pause_torrent', hash)
|
||||
except Exception as e:
|
||||
logger.error('Torrent failed to be stopped: ' + e)
|
||||
return false
|
||||
else:
|
||||
logger.info('Torrent ' + hash + ' was stopped')
|
||||
return True
|
||||
|
||||
|
||||
def load_torrent(self, filepath):
|
||||
|
||||
logger.info('filepath to torrent file set to : ' + filepath)
|
||||
torrent_id = False
|
||||
|
||||
if self.client.connected is True:
|
||||
logger.info('Trying to send to deluge now')
|
||||
# Open torrent to endcode
|
||||
logger.info('Checking if Torrent Exists!')
|
||||
|
||||
torrentfile = open(filepath, 'rb')
|
||||
torrentcontent = torrentfile.read()
|
||||
torrentencode = base64.encodestring(torrentcontent)
|
||||
torrentcontent = open(filepath, 'rb').read()
|
||||
hash = str.lower(self.get_the_hash(filepath)) # Deluge expects a lower case hash
|
||||
|
||||
# Send to Deluge and return torrent_id
|
||||
logger.info('FileName: ' + str(os.path.basename(filepath)))
|
||||
|
||||
|
||||
torrent_id = self.client.call('core.add_torrent_file', str(os.path.basename(filepath)), torrentencode, '')
|
||||
|
||||
if not torrent_id:
|
||||
logger.error('Torrent not added')
|
||||
return False
|
||||
logger.debug('Torrent Hash (load_torrent): "' + hash + '"')
|
||||
logger.debug('FileName (load_torrent): ' + str(os.path.basename(filepath)))
|
||||
|
||||
|
||||
#Check if torrent already added
|
||||
if self.find_torrent(str.lower(hash)):
|
||||
logger.info('load_torrent: Torrent already exists!')
|
||||
else:
|
||||
logger.info('TorrentID: ' + torrent_id)
|
||||
|
||||
# If label enabled put label on torrent in Deluge
|
||||
logger.info('Label: ' + mylar.DELUGE_LABEL)
|
||||
if torrent_id and mylar.DELUGE_LABEL:
|
||||
logger.info ('Setting label to ' + mylar.DELUGE_LABEL)
|
||||
logger.info('Torrent not added yet, trying to add it now!')
|
||||
try:
|
||||
self.client.call('label.set_torrent', torrent_id, mylar.DELUGE_LABEL)
|
||||
except:
|
||||
#if label isn't set, let's try and create one.
|
||||
torrent_id = self.client.call('core.add_torrent_file', str(os.path.basename(filepath)), base64.encodestring(torrentcontent), '')
|
||||
except Exception as e:
|
||||
logger.debug('Torrent not added')
|
||||
return False
|
||||
else:
|
||||
logger.debug('TorrentID: ' + torrent_id)
|
||||
return True
|
||||
|
||||
# If label enabled put label on torrent in Deluge
|
||||
if torrent_id and mylar.DELUGE_LABEL:
|
||||
logger.info ('Setting label to ' + mylar.DELUGE_LABEL)
|
||||
try:
|
||||
self.client.call('label.add', mylar.DELUGE_LABEL)
|
||||
self.client.call('label.set_torrent', torrent_id, mylar.DELUGE_LABEL)
|
||||
except:
|
||||
logger.warn('Unable to set label - Either try to create it manually within Deluge, and/or ensure there are no spaces, capitalization or special characters in label')
|
||||
return False
|
||||
logger.info('Succesfully set label to ' + mylar.DELUGE_LABEL)
|
||||
return True
|
||||
#if label isn't set, let's try and create one.
|
||||
try:
|
||||
self.client.call('label.add', mylar.DELUGE_LABEL)
|
||||
self.client.call('label.set_torrent', torrent_id, mylar.DELUGE_LABEL)
|
||||
except:
|
||||
logger.warn('Unable to set label - Either try to create it manually within Deluge, and/or ensure there are no spaces, capitalization or special characters in label')
|
||||
return False
|
||||
logger.info('Succesfully set label to ' + mylar.DELUGE_LABEL)
|
||||
try:
|
||||
self.find_torrent(torrent_id)
|
||||
logger.info('Double checking torrent was added.')
|
||||
except Exception as e:
|
||||
logger.warn('Torrent was not added! Please check logs')
|
||||
return False
|
||||
else:
|
||||
logger.info('Torrent successfully added!')
|
||||
return True
|
||||
|
||||
|
||||
def delete_torrent(self, torrent):
|
||||
# To be Added
|
||||
return False
|
||||
def delete_torrent(self, hash, removeData=False):
|
||||
try:
|
||||
self.client.find_torrent(hash)
|
||||
except Exception as e:
|
||||
logger.error('Torrent ' + hash + ' does not exist')
|
||||
return False
|
||||
else:
|
||||
try:
|
||||
self.client.call('core.remote_torrent', hash, removeData)
|
||||
except Exception as e:
|
||||
logger.error('Unable to delete torrent ' + hash)
|
||||
return False
|
||||
else:
|
||||
logger.info('Torrent deleted ' + hash)
|
||||
return True
|
||||
|
||||
|
||||
def get_the_hash(self, filepath):
|
||||
import hashlib, StringIO, bencode
|
||||
|
||||
# Open torrent file
|
||||
torrent_file = open(filepath, "rb")
|
||||
metainfo = bencode.bdecode(torrent_file.read())
|
||||
info = metainfo['info']
|
||||
thehash = hashlib.sha1(bencode.bencode(info)).hexdigest().upper()
|
||||
logger.debug('Hash: ' + thehash)
|
||||
return thehash
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue