bar = pyqtSignal()

def __init__(self, controller, webview = None):

super(JavaScriptBridge, self).__init__()

self.controller = controller

self.webview = webview

self.last_dropped_urls = []

@pyqtSlot(QString, result="QString")

def getFullPath(self, file_name):

for url in self.last_dropped_urls:

if url.path().split('/')[-1] == file_name:

return url.path()

return QString("")

@pyqtSlot(result="QString")

def getSettings(self):

out = []

s = self.controller.settings()

out.append(u"{")

out.append(u'"upload_rate_limit" : {},'.format(s.upload_rate_limit))

out.append(u'"download_rate_limit" : {},'.format(s.download_rate_limit))

out.append(u'"local_upload_rate_limit" : {},'.format(s.local_upload_rate_limit))

out.append(u'"local_download_rate_limit" : {}'.format(s.local_download_rate_limit))

out.append(u"}")

return QString(u''.join(out))

'''

@pyqtSlot(QString, result="QString")

def getStatus(self):

out = []

s = d.status()

state_str = ['queued', 'checking', 'downloading metadata', 'downloading', 'finished', 'seeding', 'allocating', 'checking fastresume']

out.append(u"{")

if s.state != libtorrent.torrent_status.seeding:

out.append(u'"state" : {},'.format(state_str[s.state]))

out.append(u'"progress" : {},'.format(s.progress))

out.append(u'"total_done" : {},'.format(s.total_done))

out.append(u'"num_peers" : {},'.format(s.num_peers))

out.append(u'"distributed_copies" : {},'.format(s.distributed_copies))

out.append(u'"download_rate" : {},'.format(s.download_rate))

out.append(u'"total_download" : {},'.format(s.total_download))

out.append(u'"upload_rate" : {},'.format(s.upload_rate))

out.append(u'"total_upload" : {},'.format(s.total_upload))

out.append(u'"next_announce" : {},'.format(s.next_announce))

out.append(u'"tracker" : {},'.format(s.tracker))

out.append(u"}")

return QString(u''.join(out))

'''

'''

@pyqtSlot(QString, result="QString")

def getPeers(self, download_name):

out = []

peers = d.get_peer_info()

out.append(u"{")

for p in peers:

out.append(u'"down_speed" : {},'.format(p.down_speed))

out.append(u'"total_download" : {},'.format(p.total_download))

out.append(u'"up_speed" : {},'.format(p.up_speed))

out.append(u'"total_upload" : {},'.format(p.total_upload))

out.append(u'"download_queue_length" : {},'.format(p.download_queue_length))

out.append(u'"upload_queue_length" : {}'.format(p.upload_queue_length))

out.append(u"}")

return QString(u''.join(out))

'''

@pyqtSlot(QString, QString)

def torrentAction(self, action, info_hash):

d = self.controller.findTorrentByHash(str(info_hash))

if d:

if action == "remove":

d.remove()

elif action == "resume":

d.resume()

elif action == "pause":

d.pause()

else:

print("Unknown action: {}".format(action))

else:

print("Download not found: {}".format(info_hash))

@pyqtSlot(result="QString")

def getAllTorrents(self):

out = []

out.append(u'{"torrents" : [')

for i, d in enumerate(self.controller.downloads):

if i != 0:

out.append(u',')

out.append(u'{')

out.append(u'"hash":"{}",'.format(d.info_hash()))

out.append(u'"name":"{}",'.format(d.name()))

out.append(u'"progress":{},'.format(d.progress()))

out.append(u'"num_peers":{},'.format(d.num_peers()))

out.append(u'"num_seeds":{},'.format(d.num_seeds()))

out.append(u'"status":"{}",'.format(d.status()))

if d.has_metadata():

out.append(u'"total_size":"{}",'.format(d.total_size()))

out.append(u'"num_files":"{}",'.format(d.num_files()))

out.append(u'"upload_rate":{},'.format(d.upload_rate()))

out.append(u'"download_rate":{}'.format(d.download_rate()))

out.append(u'}')

out.append(u']}')

return QString(u''.join(out))

@pyqtSlot(QString, QString, result="QString")

def getTorrentFiles(self, info_hash, abs_base_path):

out = []

print("getTorrentFiles: path {}".format(abs_base_path))

out.append(u'{"torrent_files" : [')

if len(abs_base_path) == 0:

abs_base_path = "/"

if abs_base_path[-1] != "/":

abs_base_path += "/"

if abs_base_path[0] != '/':

abs_base_path = "/"+ abs_base_path

subfolders = {}

d = self.controller.findTorrentByHash(info_hash)

if d and d.has_metadata():

files = d.files()

priorities = d.file_priorities()

progress = d.file_progress()

if len(files) != len(priorities) or len(files) != len(progress):

print("Element mismatch")

return

first = True

for i, file in enumerate(files):

if file.pad_file:

continue

abs_file_path = file.path

if len(abs_file_path) == 0 or abs_file_path[0] != '/':

abs_file_path = "/"+ abs_file_path

if not abs_file_path.startswith(abs_base_path):

continue

rel_file_path = abs_file_path[len(abs_base_path):]

pos = rel_file_path.find("/")

if pos < 0:

#file in current path

file_name = rel_file_path

if first:

first = False

else:

out.append(u',')

out.append(u'{')

out.append(u'"name":"{}",'.format(file_name))

out.append(u'"path":"{}",'.format(file.path))

out.append(u'"size":"{}",'.format(file.size))

out.append(u'"priority":"{}",'.format(priorities[i]))

out.append(u'"progress":"{}"'.format(progress[i]))

out.append(u'}')

else:

#file in subfolder

folder_name = rel_file_path[0:pos]

if folder_name in subfolders:

folder = subfolders[folder_name]

folder["size"] += file.size

folder["priority"].append(priorities[i])

folder["progress"].append(progress[i])

else:

folder = {}

subfolders[folder_name] = folder

folder["size"] = 0

folder["priority"] = []

folder["progress"] = []

for folder_name, folder in subfolders.iteritems():

if first:

first = False

else:

out.append(u',')

out.append(u'{')

out.append(u'"name":"{}",'.format(folder_name))

out.append(u'"size":"{}",'.format(folder["size"]))

out.append(u'"priority":"{}",'.format(0))

out.append(u'"progress":"{}"'.format(0))

out.append(u'}')

out.append(u']}')

print(u''.join(out))

return QString(u''.join(out))

@pyqtSlot(int, result="QString")

def getFolderSize(start_path):

#some security

if self.getFullPath(start_path).length() == 0:

return -1

total_size = 0

for dirpath, dirnames, filenames in os.walk(start_path):

for f in filenames:

fp = os.path.join(dirpath, f)

total_size += os.path.getsize(fp)

return total_size

@pyqtSlot(int, result="QString")

def getFolderCount(start_path):

#some security

if self.getFullPath(start_path).length() == 0:

return -1

total_count = 0

for dirpath, dirnames, filenames in os.walk(start_path):

for f in filenames:

total_count += 1

return total_count

def callFunction(self, javascript):

frame = self.webview.page().mainFrame()

frame.evaluateJavaScript(javascript)

def showProgress(self, progress):

self.callFunction('showProgress({})'.format(progress))

def showMessage(self, msg):

self.callFunction('showMessage("{}")'.format(msg))

@pyqtSlot(QString, QStringList)

def createTorrent(self, title, paths):

#convert QStringList into a normal python list

title = str(title)

paths = [str(p) for p in paths]

torrent_root = os.path.join(self.controller.cache_path, title)

if os.path.exists(torrent_root):

self.showMessage("Folder already exists: {}".format(title))

return

''' Create a temp torrent file. the one in the cache folder will be written at program shutdown '''

fd, torrent_file = tempfile.mkstemp(suffix=".torrent")

os.close(fd)

try:

os.mkdir(torrent_root)

''' Copy all file to one location '''

for path in paths:

print(u"jsbridge: Copy {} to {}".format(path, torrent_root))

shutil.copy(path, torrent_root)

TorrentController.createTorrent(torrent_file, torrent_root, self.showProgress)

self.controller.addTorrentFile(torrent_file)

except:

e = sys.exc_info()[0]

print(e)

self.showMessage(e)

#some cleanup