mirror of https://github.com/evilhero/mylar
110 lines
4.3 KiB
Python
110 lines
4.3 KiB
Python
# This file is part of Mylar.
|
|
#
|
|
# Mylar 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.
|
|
#
|
|
# Mylar 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 Mylar. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import os
|
|
import sys
|
|
import re
|
|
import time
|
|
import shutil
|
|
import traceback
|
|
from base64 import b16encode, b32decode
|
|
|
|
import hashlib, StringIO
|
|
import bencode
|
|
from torrent.helpers.variable import link, symlink, is_rarfile
|
|
|
|
import requests
|
|
#from lib.unrar2 import RarFile
|
|
|
|
import torrent.clients.rtorrent as TorClient
|
|
|
|
import mylar
|
|
from mylar import logger, helpers
|
|
|
|
class RTorrent(object):
|
|
def __init__(self):
|
|
self.client = TorClient.TorrentClient()
|
|
if not self.client.connect(mylar.CONFIG.RTORRENT_HOST,
|
|
mylar.CONFIG.RTORRENT_USERNAME,
|
|
mylar.CONFIG.RTORRENT_PASSWORD,
|
|
mylar.CONFIG.RTORRENT_AUTHENTICATION,
|
|
mylar.CONFIG.RTORRENT_VERIFY,
|
|
mylar.CONFIG.RTORRENT_RPC_URL,
|
|
mylar.CONFIG.RTORRENT_CA_BUNDLE):
|
|
logger.error('[ERROR] Could not connect to %s - exiting' % mylar.CONFIG.RTORRENT_HOST)
|
|
sys.exit(-1)
|
|
|
|
def main(self, torrent_hash=None, filepath=None, check=False):
|
|
|
|
torrent = self.client.find_torrent(torrent_hash)
|
|
if torrent:
|
|
if check:
|
|
logger.fdebug('Successfully located torrent %s by hash on client. Detailed statistics to follow' % torrent_hash)
|
|
else:
|
|
logger.warn("%s Torrent already exists. Not downloading at this time." % torrent_hash)
|
|
return
|
|
else:
|
|
if check:
|
|
logger.warn('Unable to locate torrent with a hash value of %s' % torrent_hash)
|
|
return
|
|
|
|
if filepath:
|
|
loadit = self.client.load_torrent(filepath)
|
|
if loadit:
|
|
if filepath.startswith('magnet'):
|
|
torrent_hash = re.findall("urn:btih:([\w]{32,40})", filepath)[0]
|
|
if len(torrent_hash) == 32:
|
|
torrent_hash = b16encode(b32decode(torrent_hash)).lower()
|
|
torrent_hash = torrent_hash.upper()
|
|
else:
|
|
torrent_hash = self.get_the_hash(filepath)
|
|
else:
|
|
return
|
|
|
|
torrent = self.client.find_torrent(torrent_hash)
|
|
if torrent is None:
|
|
logger.warn('Couldn\'t find torrent with hash: %s' % torrent_hash)
|
|
sys.exit(-1)
|
|
|
|
torrent_info = self.client.get_torrent(torrent)
|
|
if check:
|
|
return torrent_info
|
|
|
|
if torrent_info['completed']:
|
|
logger.fdebug('Directory: %s' % torrent_info['folder'])
|
|
logger.fdebug('Name: %s' % torrent_info['name'])
|
|
logger.fdebug('FileSize: %s' % helpers.human_size(torrent_info['total_filesize']))
|
|
logger.fdebug('Completed: %s' % torrent_info['completed'])
|
|
logger.fdebug('Downloaded: %s' % helpers.human_size(torrent_info['download_total']))
|
|
logger.fdebug('Uploaded: %s' % helpers.human_size(torrent_info['upload_total']))
|
|
logger.fdebug('Ratio: %s' % torrent_info['ratio'])
|
|
#logger.info('Time Started: %s' % torrent_info['time_started'])
|
|
logger.fdebug('Seeding Time: %s' % helpers.humanize_time(int(time.time()) - torrent_info['time_started']))
|
|
|
|
if torrent_info['label']:
|
|
logger.fdebug('Torrent Label: %s' % torrent_info['label'])
|
|
|
|
#logger.info(torrent_info)
|
|
return torrent_info
|
|
|
|
def get_the_hash(self, filepath):
|
|
# Open torrent file
|
|
torrent_file = open(filepath, "rb")
|
|
metainfo = bencode.decode(torrent_file.read())
|
|
info = metainfo['info']
|
|
thehash = hashlib.sha1(bencode.encode(info)).hexdigest().upper()
|
|
logger.fdebug('Hash: %s' % thehash)
|
|
return thehash
|