codex-storage/deluge
1
0
Fork 0
mirror of https://github.com/codex-storage/deluge.git synced 2025-02-05 08:03:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Core] Split create part of save_state into create_state method

Browse Source
This commit is contained in:
Calum Lind 2015-10-01 18:19:37 +01:00
parent d34705860a
commit 40c1597c67
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
deluge/core/torrentmanager.py
View File

@ -569,8 +569,13 @@ class TorrentManager(component.Component):
log.info("Finished loading %d torrents.", len(state.torrents)) log.info("Finished loading %d torrents.", len(state.torrents))
component.get("EventManager").emit(SessionStartedEvent()) component.get("EventManager").emit(SessionStartedEvent())
def save_state(self): def create_state(self):
"""Save the state of the TorrentManager to the torrents.state file""" """Create a state of all the torrents in TorrentManager.
Returns:
TorrentManagerState: The TorrentManager state.
"""
state = TorrentManagerState() state = TorrentManagerState()
# Create the state for each Torrent and append to the list # Create the state for each Torrent and append to the list
for torrent in self.torrents.values(): for torrent in self.torrents.values():
@ -612,7 +617,16 @@ class TorrentManager(component.Component):
torrent.options["name"] torrent.options["name"]
) )
state.torrents.append(torrent_state) state.torrents.append(torrent_state)
return state
def save_state(self):
"""Save the state of the TorrentManager to the torrents.state file.
Returns:
bool: Always returns True so that the timer thread will continue.
"""
state = self.create_state()
filename = "torrents.state" filename = "torrents.state"
filepath = os.path.join(self.state_dir, filename) filepath = os.path.join(self.state_dir, filename)
filepath_bak = filepath + ".bak" filepath_bak = filepath + ".bak"
@ -641,7 +655,6 @@ class TorrentManager(component.Component):
if os.path.isfile(filepath_bak): if os.path.isfile(filepath_bak):
log.info("Restoring backup of %s from: %s", filename, filepath_bak) log.info("Restoring backup of %s from: %s", filename, filepath_bak)
os.rename(filepath_bak, filepath) os.rename(filepath_bak, filepath)
# We return True so that the timer thread will continue
return True return True
def save_resume_data(self, torrent_ids=None, flush_disk_cache=False): def save_resume_data(self, torrent_ids=None, flush_disk_cache=False):