Add reset_ip_filter() to core.

This commit is contained in:
Andrew Resch 2008-02-21 04:19:13 +00:00
commit 361a276da4

View file

@ -187,10 +187,6 @@ class Core(
self.settings = lt.session_settings() self.settings = lt.session_settings()
self.settings.user_agent = "Deluge %s" % deluge.common.get_version() self.settings.user_agent = "Deluge %s" % deluge.common.get_version()
# Create the IP Filter
self.ip_filter = lt.ip_filter()
self.session.set_ip_filter(self.ip_filter)
# Set lazy bitfield # Set lazy bitfield
self.settings.lazy_bitfields = 1 self.settings.lazy_bitfields = 1
self.session.set_settings(self.settings) self.session.set_settings(self.settings)
@ -505,7 +501,16 @@ class Core(
def export_block_ip_range(self, range): def export_block_ip_range(self, range):
"""Block an ip range""" """Block an ip range"""
try:
self.ip_filter.add_rule(range[0], range[1], 1) self.ip_filter.add_rule(range[0], range[1], 1)
except AttributeError:
self.export_reset_ip_filter()
self.ip_filter.add_rule(range[0], range[1], 1)
def export_reset_ip_filter(self):
"""Clears the ip filter"""
self.ip_filter = lt.ip_filter()
self.session.set_ip_filter(self.ip_filter)
# Signals # Signals
def torrent_added(self, torrent_id): def torrent_added(self, torrent_id):