From 1a0718b4ac1519eb9ca82ac016cb49ce1f8e5284 Mon Sep 17 00:00:00 2001
From: Andrew Resch <andrewresch@gmail.com>
Date: Mon, 21 Jan 2008 22:24:31 +0000
Subject: [PATCH] Update 'save_path' on storage_moved_alert.

---
 deluge/core/core.py           |  2 +-
 deluge/core/torrent.py        |  3 +++
 deluge/core/torrentmanager.py | 12 ++++++++++++
 3 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/deluge/core/core.py b/deluge/core/core.py
index eee01ca8b..24ff7ddd2 100644
--- a/deluge/core/core.py
+++ b/deluge/core/core.py
@@ -633,4 +633,4 @@ class Core(
         # Get the torrent_id
         torrent_id = str(alert.handle.info_hash())
         # Emit torrent_paused signal
-        self.torrent_paused(torrent_id)  
+        self.torrent_paused(torrent_id)
diff --git a/deluge/core/torrent.py b/deluge/core/torrent.py
index 751da9d77..126f5c61e 100644
--- a/deluge/core/torrent.py
+++ b/deluge/core/torrent.py
@@ -108,6 +108,9 @@ class Torrent:
     def set_prioritize_first_last(self, prioritize):
         pass
             
+    def set_save_path(self, save_path):
+        self.save_path = save_path
+        
     def get_state(self):
         """Returns the state of this torrent for saving to the session state"""
         status = self.handle.status()
diff --git a/deluge/core/torrentmanager.py b/deluge/core/torrentmanager.py
index edbed0eb4..67710aa2f 100644
--- a/deluge/core/torrentmanager.py
+++ b/deluge/core/torrentmanager.py
@@ -100,6 +100,8 @@ class TorrentManager(component.Component):
         self.alerts.register_handler("tracker_alert", self.on_alert_tracker)
         self.alerts.register_handler("tracker_warning_alert",
             self.on_alert_tracker_warning)
+        self.alerts.register_handler("storage_moved_alert",
+            self.on_alert_storage_moved)
 
     def start(self):
         # Try to load the state from file
@@ -611,3 +613,13 @@ class TorrentManager(component.Component):
             self.torrents[torrent_id].set_tracker_status(tracker_status)
         except KeyError:
             log.debug("torrent_id doesn't exist.")
+            
+    def on_alert_storage_moved(self, alert):
+        log.debug("on_alert_storage_moved")
+        # Get the torrent_id
+        torrent_id = str(alert.handle.info_hash())
+        try:
+            self.torrents[torrent_id].set_save_path(alert.handle.save_path())
+        except KeyError:
+            log.debug("torrent_id doesn't exist.")
+