617 lines
22 KiB
Python
617 lines
22 KiB
Python
#
|
|
# torrent.py
|
|
#
|
|
# Copyright (C) 2007, 2008 Andrew Resch <andrewresch@gmail.com>
|
|
#
|
|
# Deluge is free software.
|
|
#
|
|
# You may 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.
|
|
#
|
|
# deluge 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 deluge. If not, write to:
|
|
# The Free Software Foundation, Inc.,
|
|
# 51 Franklin Street, Fifth Floor
|
|
# Boston, MA 02110-1301, USA.
|
|
#
|
|
|
|
|
|
"""Internal Torrent class"""
|
|
|
|
import os
|
|
|
|
try:
|
|
import deluge.libtorrent as lt
|
|
except ImportError:
|
|
import libtorrent as lt
|
|
if not (lt.version_major == 0 and lt.version_minor == 14):
|
|
raise ImportError("This version of Deluge requires libtorrent 0.14!")
|
|
|
|
import deluge.common
|
|
import deluge.component as component
|
|
from deluge.configmanager import ConfigManager
|
|
from deluge.log import LOG as log
|
|
import deluge.xmlrpclib
|
|
|
|
TORRENT_STATE = deluge.common.TORRENT_STATE
|
|
|
|
class Torrent:
|
|
"""Torrent holds information about torrents added to the libtorrent session.
|
|
"""
|
|
def __init__(self, handle, options, state=None, filename=None):
|
|
log.debug("Creating torrent object %s", str(handle.info_hash()))
|
|
# Get the core config
|
|
self.config = ConfigManager("core.conf")
|
|
|
|
self.signals = component.get("SignalManager")
|
|
|
|
# Set the libtorrent handle
|
|
self.handle = handle
|
|
# Set the torrent_id for this torrent
|
|
self.torrent_id = str(handle.info_hash())
|
|
|
|
# We store the filename just in case we need to make a copy of the torrentfile
|
|
self.filename = filename
|
|
|
|
# Holds status info so that we don't need to keep getting it from lt
|
|
self.status = self.handle.status()
|
|
self.torrent_info = self.handle.get_torrent_info()
|
|
|
|
# Files dictionary
|
|
self.files = self.get_files()
|
|
# Set the default file priorities to normal
|
|
self.file_priorities = [1]* len(self.files)
|
|
|
|
# Default total_uploaded to 0, this may be changed by the state
|
|
self.total_uploaded = 0
|
|
|
|
# Set default auto_managed value
|
|
self.auto_managed = options["auto_managed"]
|
|
if not handle.is_paused():
|
|
handle.auto_managed(self.auto_managed)
|
|
|
|
# We need to keep track if the torrent is finished in the state to prevent
|
|
# some weird things on state load.
|
|
self.is_finished = False
|
|
|
|
# Queueing options
|
|
self.stop_at_ratio = False
|
|
self.stop_ratio = 2.00
|
|
self.remove_at_ratio = False
|
|
|
|
# Load values from state if we have it
|
|
if state is not None:
|
|
# This is for saving the total uploaded between sessions
|
|
self.total_uploaded = state.total_uploaded
|
|
# Set the trackers
|
|
self.set_trackers(state.trackers)
|
|
# Set the filename
|
|
self.filename = state.filename
|
|
self.is_finished = state.is_finished
|
|
# Set the per-torrent queue options
|
|
self.set_stop_at_ratio(state.stop_at_ratio)
|
|
self.set_stop_ratio(state.stop_ratio)
|
|
self.set_remove_at_ratio(state.remove_at_ratio)
|
|
else:
|
|
# Tracker list
|
|
self.trackers = []
|
|
# Create a list of trackers
|
|
for value in self.handle.trackers():
|
|
tracker = {}
|
|
tracker["url"] = value.url
|
|
tracker["tier"] = value.tier
|
|
self.trackers.append(tracker)
|
|
|
|
# Set the allocation mode
|
|
self.compact = options["compact_allocation"]
|
|
|
|
# Various torrent options
|
|
self.set_max_connections(options["max_connections_per_torrent"])
|
|
self.set_max_upload_slots(options["max_upload_slots_per_torrent"])
|
|
self.set_max_upload_speed(options["max_upload_speed_per_torrent"])
|
|
self.set_max_download_speed(options["max_download_speed_per_torrent"])
|
|
self.set_prioritize_first_last(options["prioritize_first_last_pieces"])
|
|
self.handle.resolve_countries(True)
|
|
if options.has_key("file_priorities"):
|
|
self.set_file_priorities(options["file_priorities"])
|
|
|
|
# Where the torrent is being saved to
|
|
self.save_path = options["download_location"]
|
|
# Status message holds error info about the torrent
|
|
self.statusmsg = "OK"
|
|
|
|
# The torrents state
|
|
self.update_state()
|
|
|
|
# The tracker status
|
|
self.tracker_status = ""
|
|
|
|
log.debug("Torrent object created.")
|
|
|
|
def set_tracker_status(self, status):
|
|
"""Sets the tracker status"""
|
|
self.tracker_status = status
|
|
|
|
def set_max_connections(self, max_connections):
|
|
self.max_connections = int(max_connections)
|
|
self.handle.set_max_connections(self.max_connections)
|
|
|
|
def set_max_upload_slots(self, max_slots):
|
|
self.max_upload_slots = int(max_slots)
|
|
self.handle.set_max_uploads(self.max_upload_slots)
|
|
|
|
def set_max_upload_speed(self, m_up_speed):
|
|
self.max_upload_speed = m_up_speed
|
|
self.handle.set_upload_limit(int(m_up_speed * 1024))
|
|
|
|
def set_max_download_speed(self, m_down_speed):
|
|
self.max_download_speed = m_down_speed
|
|
self.handle.set_download_limit(int(m_down_speed * 1024))
|
|
|
|
def set_prioritize_first_last(self, prioritize):
|
|
self.prioritize_first_last = prioritize
|
|
if self.prioritize_first_last:
|
|
if self.handle.get_torrent_info().num_files() == 1:
|
|
# We only do this if one file is in the torrent
|
|
priorities = [1] * self.handle.get_torrent_info().num_pieces()
|
|
priorities[0] = 7
|
|
priorities[-1] = 7
|
|
self.handle.prioritize_pieces(priorities)
|
|
|
|
def set_save_path(self, save_path):
|
|
self.save_path = save_path
|
|
|
|
def set_auto_managed(self, auto_managed):
|
|
self.auto_managed = auto_managed
|
|
self.handle.auto_managed(auto_managed)
|
|
self.update_state()
|
|
|
|
def set_stop_ratio(self, stop_ratio):
|
|
self.stop_ratio = stop_ratio
|
|
|
|
def set_stop_at_ratio(self, stop_at_ratio):
|
|
self.stop_at_ratio = stop_at_ratio
|
|
|
|
def set_remove_at_ratio(self, remove_at_ratio):
|
|
self.remove_at_ratio = remove_at_ratio
|
|
|
|
def set_file_priorities(self, file_priorities):
|
|
if len(file_priorities) != len(self.files):
|
|
log.debug("file_priorities len != num_files")
|
|
self.options["file_priorities"] = self.handle.file_priorities()
|
|
return
|
|
|
|
if self.compact:
|
|
log.debug("setting file priority with compact allocation does not work!")
|
|
self.file_priorities = self.handle.file_priorities()
|
|
return
|
|
|
|
log.debug("setting %s's file priorities: %s", self.torrent_id, file_priorities)
|
|
|
|
self.handle.prioritize_files(file_priorities)
|
|
|
|
if 0 in self.file_priorities:
|
|
# We have previously marked a file 'Do Not Download'
|
|
# Check to see if we have changed any 0's to >0 and change state accordingly
|
|
for index, priority in enumerate(self.file_priorities):
|
|
if priority == 0 and file_priorities[index] > 0:
|
|
# We have a changed 'Do Not Download' to a download priority
|
|
self.is_finished = False
|
|
self.update_state()
|
|
break
|
|
|
|
self.file_priorities = file_priorities
|
|
|
|
# Set the first/last priorities if needed
|
|
self.set_prioritize_first_last(self.prioritize_first_last)
|
|
|
|
def set_trackers(self, trackers):
|
|
"""Sets trackers"""
|
|
if trackers == None:
|
|
trackers = []
|
|
for value in self.handle.trackers():
|
|
tracker = {}
|
|
tracker["url"] = value.url
|
|
tracker["tier"] = value.tier
|
|
trackers.append(tracker)
|
|
self.trackers = trackers
|
|
return
|
|
|
|
log.debug("Setting trackers for %s: %s", self.torrent_id, trackers)
|
|
tracker_list = []
|
|
|
|
for tracker in trackers:
|
|
new_entry = lt.announce_entry(tracker["url"])
|
|
new_entry.tier = tracker["tier"]
|
|
tracker_list.append(new_entry)
|
|
|
|
self.handle.replace_trackers(tracker_list)
|
|
|
|
# Print out the trackers
|
|
for t in self.handle.trackers():
|
|
log.debug("tier: %s tracker: %s", t.tier, t.url)
|
|
# Set the tracker list in the torrent object
|
|
self.trackers = trackers
|
|
if len(trackers) > 0:
|
|
# Force a reannounce if there is at least 1 tracker
|
|
self.force_reannounce()
|
|
|
|
def update_state(self):
|
|
"""Updates the state based on what libtorrent's state for the torrent is"""
|
|
# Set the initial state based on the lt state
|
|
LTSTATE = deluge.common.LT_TORRENT_STATE
|
|
ltstate = int(self.handle.status().state)
|
|
|
|
log.debug("set_state_based_on_ltstate: %s", ltstate)
|
|
|
|
if ltstate == LTSTATE["Queued"] or ltstate == LTSTATE["Checking"]:
|
|
self.state = "Checking"
|
|
return
|
|
elif ltstate == LTSTATE["Downloading"] or ltstate == LTSTATE["Downloading Metadata"]:
|
|
self.state = "Downloading"
|
|
elif ltstate == LTSTATE["Finished"] or ltstate == LTSTATE["Seeding"]:
|
|
self.state = "Seeding"
|
|
elif ltstate == LTSTATE["Allocating"]:
|
|
self.state = "Allocating"
|
|
|
|
if self.handle.is_paused() and len(self.handle.status().error) > 0:
|
|
# This is an error'd torrent
|
|
self.state = "Error"
|
|
self.set_status_message(self.handle.status().error)
|
|
self.handle.auto_managed(False)
|
|
elif self.handle.is_paused() and self.handle.is_auto_managed():
|
|
self.state = "Queued"
|
|
elif self.handle.is_paused() and not self.handle.is_auto_managed():
|
|
self.state = "Paused"
|
|
|
|
def set_state(self, state):
|
|
"""Accepts state strings, ie, "Paused", "Seeding", etc."""
|
|
if state not in TORRENT_STATE:
|
|
log.debug("Trying to set an invalid state %s", state)
|
|
return
|
|
|
|
self.state = state
|
|
return
|
|
|
|
def set_status_message(self, message):
|
|
self.statusmsg = message
|
|
|
|
def get_eta(self):
|
|
"""Returns the ETA in seconds for this torrent"""
|
|
if self.status == None:
|
|
status = self.handle.status()
|
|
else:
|
|
status = self.status
|
|
|
|
left = status.total_wanted - status.total_done
|
|
|
|
if left <= 0 or status.download_payload_rate == 0:
|
|
return 0
|
|
|
|
try:
|
|
eta = left / status.download_payload_rate
|
|
except ZeroDivisionError:
|
|
eta = 0
|
|
|
|
return eta
|
|
|
|
def get_ratio(self):
|
|
"""Returns the ratio for this torrent"""
|
|
if self.status == None:
|
|
status = self.handle.status()
|
|
else:
|
|
status = self.status
|
|
|
|
# Return -1.0 if the downloaded bytes is 0, this is to represent infinity
|
|
if status.all_time_download == 0:
|
|
return -1.0
|
|
|
|
return float(status.all_time_upload) / float(status.all_time_download)
|
|
|
|
def get_files(self):
|
|
"""Returns a list of files this torrent contains"""
|
|
if self.torrent_info == None:
|
|
torrent_info = self.handle.get_torrent_info()
|
|
else:
|
|
torrent_info = self.torrent_info
|
|
|
|
ret = []
|
|
files = torrent_info.files()
|
|
for index, file in enumerate(files):
|
|
ret.append({
|
|
'index': index,
|
|
'path': file.path.decode("utf8", "ignore"),
|
|
'size': file.size,
|
|
'offset': file.offset
|
|
})
|
|
return ret
|
|
|
|
def get_peers(self):
|
|
"""Returns a list of peers and various information about them"""
|
|
ret = []
|
|
peers = self.handle.get_peer_info()
|
|
|
|
for peer in peers:
|
|
# We do not want to report peers that are half-connected
|
|
if peer.flags & peer.connecting or peer.flags & peer.handshake:
|
|
continue
|
|
try:
|
|
client = str(peer.client).decode("utf-8")
|
|
except UnicodeDecodeError:
|
|
client = str(peer.client).decode("latin-1")
|
|
|
|
# Make country a proper string
|
|
country = str()
|
|
for c in peer.country:
|
|
if not c.isalpha():
|
|
country += " "
|
|
else:
|
|
country += c
|
|
|
|
ret.append({
|
|
"ip": "%s:%s" % (peer.ip[0], peer.ip[1]),
|
|
"up_speed": peer.up_speed,
|
|
"down_speed": peer.down_speed,
|
|
"country": country,
|
|
"client": client,
|
|
"seed": peer.flags & peer.seed
|
|
})
|
|
|
|
return ret
|
|
|
|
def get_queue_position(self):
|
|
"""Returns the torrents queue position"""
|
|
return self.handle.queue_position()
|
|
|
|
def get_file_progress(self):
|
|
"""Returns the file progress as a list of floats.. 0.0 -> 1.0"""
|
|
file_progress = self.handle.file_progress()
|
|
ret = []
|
|
for i,f in enumerate(self.files):
|
|
try:
|
|
ret.append(float(file_progress[i]) / float(f["size"]))
|
|
except ZeroDivisionError:
|
|
ret.append(0.0)
|
|
|
|
return ret
|
|
|
|
def get_status(self, keys):
|
|
"""Returns the status of the torrent based on the keys provided"""
|
|
# Create the full dictionary
|
|
self.status = self.handle.status()
|
|
self.torrent_info = self.handle.get_torrent_info()
|
|
|
|
# Adjust progress to be 0-100 value
|
|
progress = self.status.progress * 100
|
|
|
|
# Adjust status.distributed_copies to return a non-negative value
|
|
distributed_copies = self.status.distributed_copies
|
|
if distributed_copies < 0:
|
|
distributed_copies = 0.0
|
|
|
|
full_status = {
|
|
"distributed_copies": distributed_copies,
|
|
"total_done": self.status.total_done,
|
|
"total_uploaded": self.status.all_time_upload,
|
|
"all_time_download": self.status.all_time_download,
|
|
"state": self.state,
|
|
"paused": self.status.paused,
|
|
"progress": progress,
|
|
"next_announce": self.status.next_announce.seconds,
|
|
"total_payload_download": self.status.total_payload_download,
|
|
"total_payload_upload": self.status.total_payload_upload,
|
|
"download_payload_rate": self.status.download_payload_rate,
|
|
"upload_payload_rate": self.status.upload_payload_rate,
|
|
"num_peers": self.status.num_peers - self.status.num_seeds,
|
|
"num_seeds": self.status.num_seeds,
|
|
"total_peers": self.status.num_incomplete,
|
|
"total_seeds": self.status.num_complete,
|
|
"total_wanted": self.status.total_wanted,
|
|
"tracker": self.status.current_tracker,
|
|
"trackers": self.trackers,
|
|
"tracker_status": self.tracker_status,
|
|
"save_path": self.save_path,
|
|
"files": self.files,
|
|
"file_priorities": self.file_priorities,
|
|
"compact": self.compact,
|
|
"max_connections": self.max_connections,
|
|
"max_upload_slots": self.max_upload_slots,
|
|
"max_upload_speed": self.max_upload_speed,
|
|
"max_download_speed": self.max_download_speed,
|
|
"prioritize_first_last": self.prioritize_first_last,
|
|
"message": self.statusmsg,
|
|
"hash": self.torrent_id,
|
|
"active_time": self.status.active_time,
|
|
"seeding_time": self.status.seeding_time,
|
|
"seed_rank": self.status.seed_rank,
|
|
"is_auto_managed": self.auto_managed,
|
|
"stop_ratio": self.stop_ratio,
|
|
"stop_at_ratio": self.stop_at_ratio,
|
|
"remove_at_ratio": self.remove_at_ratio
|
|
}
|
|
def ti_name():
|
|
if self.handle.has_metadata():
|
|
try:
|
|
return self.torrent_info.name().decode("utf8", "ignore")
|
|
except UnicodeDecodeError:
|
|
return self.torrent_info.name()
|
|
|
|
return self.torrent_id
|
|
fns = {
|
|
"name": ti_name,
|
|
"private": self.torrent_info.priv,
|
|
"total_size": self.torrent_info.total_size,
|
|
"num_files": self.torrent_info.num_files,
|
|
"num_pieces": self.torrent_info.num_pieces,
|
|
"piece_length": self.torrent_info.piece_length,
|
|
"eta": self.get_eta,
|
|
"ratio": self.get_ratio,
|
|
"file_progress": self.get_file_progress,
|
|
"queue": self.handle.queue_position,
|
|
"is_seed": self.handle.is_seed,
|
|
"peers": self.get_peers,
|
|
}
|
|
|
|
# Create the desired status dictionary and return it
|
|
status_dict = {}
|
|
|
|
if len(keys) == 0:
|
|
status_dict = full_status
|
|
for key in fns:
|
|
status_dict[key] = fns[key]()
|
|
else:
|
|
for key in keys:
|
|
if key in full_status:
|
|
status_dict[key] = full_status[key]
|
|
elif key in fns:
|
|
status_dict[key] = fns[key]()
|
|
|
|
self.status = None
|
|
self.torrent_info = None
|
|
|
|
return status_dict
|
|
|
|
def apply_options(self):
|
|
"""Applies the per-torrent options that are set."""
|
|
self.handle.set_max_connections(self.max_connections)
|
|
self.handle.set_max_uploads(self.max_upload_slots)
|
|
self.handle.set_upload_limit(int(self.max_upload_speed * 1024))
|
|
self.handle.set_download_limit(int(self.max_download_speed * 1024))
|
|
self.handle.prioritize_files(self.file_priorities)
|
|
self.handle.resolve_countries(True)
|
|
|
|
def pause(self):
|
|
"""Pause this torrent"""
|
|
# Turn off auto-management so the torrent will not be unpaused by lt queueing
|
|
self.handle.auto_managed(False)
|
|
if self.handle.is_paused():
|
|
# This torrent was probably paused due to being auto managed by lt
|
|
# Since we turned auto_managed off, we should update the state which should
|
|
# show it as 'Paused'. We need to emit a torrent_paused signal because
|
|
# the torrent_paused alert from libtorrent will not be generated.
|
|
self.update_state()
|
|
self.signals.emit("torrent_paused", self.torrent_id)
|
|
else:
|
|
try:
|
|
self.handle.pause()
|
|
except Exception, e:
|
|
log.debug("Unable to pause torrent: %s", e)
|
|
return False
|
|
|
|
return True
|
|
|
|
def resume(self):
|
|
"""Resumes this torrent"""
|
|
|
|
if self.handle.is_paused() and self.handle.is_auto_managed():
|
|
log.debug("Torrent is being auto-managed, cannot resume!")
|
|
return
|
|
else:
|
|
# Reset the status message just in case of resuming an Error'd torrent
|
|
self.set_status_message("OK")
|
|
|
|
if self.handle.is_finished():
|
|
# If the torrent has already reached it's 'stop_seed_ratio' then do not do anything
|
|
if self.config["stop_seed_at_ratio"] or self.stop_at_ratio:
|
|
if self.stop_at_ratio:
|
|
ratio = self.stop_ratio
|
|
else:
|
|
ratio = self.config["stop_seed_ratio"]
|
|
|
|
if self.get_ratio() >= ratio:
|
|
self.signals.emit("torrent_resume_at_stop_ratio")
|
|
return
|
|
|
|
if self.auto_managed:
|
|
# This torrent is to be auto-managed by lt queueing
|
|
self.handle.auto_managed(True)
|
|
|
|
try:
|
|
self.handle.resume()
|
|
except:
|
|
pass
|
|
|
|
return True
|
|
|
|
def move_storage(self, dest):
|
|
"""Move a torrent's storage location"""
|
|
try:
|
|
self.handle.move_storage(dest.encode("utf8"))
|
|
except:
|
|
return False
|
|
|
|
return True
|
|
|
|
def write_fastresume(self):
|
|
"""Writes the .fastresume file for the torrent"""
|
|
resume_data = lt.bencode(self.handle.write_resume_data())
|
|
path = "%s/%s.fastresume" % (
|
|
self.config["state_location"],
|
|
self.torrent_id)
|
|
log.debug("Saving fastresume file: %s", path)
|
|
try:
|
|
self.delete_fastresume()
|
|
fastresume = open(path, "wb")
|
|
fastresume.write(resume_data)
|
|
fastresume.close()
|
|
except IOError:
|
|
log.warning("Error trying to save fastresume file")
|
|
|
|
def delete_fastresume(self):
|
|
"""Deletes the .fastresume file"""
|
|
path = "%s/%s.fastresume" % (
|
|
self.config["state_location"],
|
|
self.torrent_id)
|
|
log.debug("Deleting fastresume file: %s", path)
|
|
try:
|
|
os.remove(path)
|
|
except Exception, e:
|
|
log.warning("Unable to delete the fastresume file: %s", e)
|
|
|
|
def delete_torrentfile(self):
|
|
"""Deletes the .torrent file in the state"""
|
|
path = "%s/%s.torrent" % (
|
|
self.config["state_location"],
|
|
self.torrent_id)
|
|
log.debug("Deleting torrent file: %s", path)
|
|
try:
|
|
os.remove(path)
|
|
except Exception, e:
|
|
log.warning("Unable to delete the torrent file: %s", e)
|
|
|
|
def force_reannounce(self):
|
|
"""Force a tracker reannounce"""
|
|
try:
|
|
self.handle.force_reannounce()
|
|
except Exception, e:
|
|
log.debug("Unable to force reannounce: %s", e)
|
|
return False
|
|
|
|
return True
|
|
|
|
def scrape_tracker(self):
|
|
"""Scrape the tracker"""
|
|
try:
|
|
self.handle.scrape_tracker()
|
|
except Exception, e:
|
|
log.debug("Unable to scrape tracker: %s", e)
|
|
return False
|
|
|
|
return True
|
|
|
|
def force_recheck(self):
|
|
"""Forces a recheck of the torrents pieces"""
|
|
try:
|
|
self.handle.force_recheck()
|
|
except Exception, e:
|
|
log.debug("Unable to force recheck: %s", e)
|
|
return False
|
|
return True
|