mirror of
https://github.com/wting/autojump
synced 2024-10-27 20:34:07 +00:00
minor refactor
This commit is contained in:
parent
d692bc644a
commit
72b8c0a660
137
bin/autojump
137
bin/autojump
@ -41,7 +41,7 @@ except ImportError:
|
|||||||
|
|
||||||
class Database:
|
class Database:
|
||||||
"""
|
"""
|
||||||
Object for interfacing with autojump database file.
|
Abstraction for interfacing with with autojump database file.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, config):
|
def __init__(self, config):
|
||||||
@ -53,42 +53,6 @@ class Database:
|
|||||||
def __len__(self):
|
def __len__(self):
|
||||||
return len(self.data)
|
return len(self.data)
|
||||||
|
|
||||||
def add(self, path, increment=10):
|
|
||||||
"""
|
|
||||||
Increase weight of existing paths or initialize new ones to 10.
|
|
||||||
"""
|
|
||||||
if path == self.config['home']:
|
|
||||||
return
|
|
||||||
|
|
||||||
path = path.rstrip(os.sep)
|
|
||||||
|
|
||||||
if self.data[path]:
|
|
||||||
self.data[path] = math.sqrt((self.data[path]**2) + (increment**2))
|
|
||||||
else:
|
|
||||||
self.data[path] = increment
|
|
||||||
|
|
||||||
self.save()
|
|
||||||
|
|
||||||
def decrease(self, path, increment=15):
|
|
||||||
"""
|
|
||||||
Decrease weight of existing path. Unknown ones are ignored.
|
|
||||||
"""
|
|
||||||
if path == self.config['home']:
|
|
||||||
return
|
|
||||||
|
|
||||||
if self.data[path] < increment:
|
|
||||||
self.data[path] = 0
|
|
||||||
else:
|
|
||||||
self.data[path] -= increment
|
|
||||||
|
|
||||||
self.save()
|
|
||||||
|
|
||||||
def get_weight(self, path):
|
|
||||||
"""
|
|
||||||
Return path weight.
|
|
||||||
"""
|
|
||||||
return self.data[path]
|
|
||||||
|
|
||||||
def load(self, error_recovery = False):
|
def load(self, error_recovery = False):
|
||||||
"""
|
"""
|
||||||
Open database file, recovering from backup if needed.
|
Open database file, recovering from backup if needed.
|
||||||
@ -123,39 +87,6 @@ class Database:
|
|||||||
shutil.copy(self.filename + '.bak', self.filename)
|
shutil.copy(self.filename + '.bak', self.filename)
|
||||||
return self.load(True)
|
return self.load(True)
|
||||||
|
|
||||||
def maintenance(self):
|
|
||||||
"""
|
|
||||||
Decay weights by 10%, periodically remove bottom 10% entries.
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
items = self.data.iteritems()
|
|
||||||
except AttributeError:
|
|
||||||
items = self.data.items()
|
|
||||||
|
|
||||||
for path, _ in items:
|
|
||||||
self.data[path] *= 0.9
|
|
||||||
|
|
||||||
if len(self.data) > self.config['max_paths']:
|
|
||||||
remove_cnt = int(0.1 * len(self.data))
|
|
||||||
for path in sorted(self.data, key=self.data.get)[:remove_cnt]:
|
|
||||||
del self.data[path]
|
|
||||||
|
|
||||||
self.save()
|
|
||||||
|
|
||||||
def purge(self):
|
|
||||||
"""
|
|
||||||
Deletes all entries that no longer exist on system.
|
|
||||||
"""
|
|
||||||
removed = []
|
|
||||||
|
|
||||||
for path in list(self.data.keys()):
|
|
||||||
if not os.path.exists(path):
|
|
||||||
removed.append(path)
|
|
||||||
del self.data[path]
|
|
||||||
|
|
||||||
self.save()
|
|
||||||
return removed
|
|
||||||
|
|
||||||
def save(self):
|
def save(self):
|
||||||
"""
|
"""
|
||||||
Save database atomically and preserve backup, creating new database if
|
Save database atomically and preserve backup, creating new database if
|
||||||
@ -196,6 +127,72 @@ class Database:
|
|||||||
print("Error while creating backup autojump file. (%s)" %
|
print("Error while creating backup autojump file. (%s)" %
|
||||||
ex, file=sys.stderr)
|
ex, file=sys.stderr)
|
||||||
|
|
||||||
|
def add(self, path, increment=10):
|
||||||
|
"""
|
||||||
|
Increase weight of existing paths or initialize new ones to 10.
|
||||||
|
"""
|
||||||
|
if path == self.config['home']:
|
||||||
|
return
|
||||||
|
|
||||||
|
path = path.rstrip(os.sep)
|
||||||
|
|
||||||
|
if self.data[path]:
|
||||||
|
self.data[path] = math.sqrt((self.data[path]**2) + (increment**2))
|
||||||
|
else:
|
||||||
|
self.data[path] = increment
|
||||||
|
|
||||||
|
self.save()
|
||||||
|
|
||||||
|
def decrease(self, path, increment=15):
|
||||||
|
"""
|
||||||
|
Decrease weight of existing path. Unknown paths are ignored.
|
||||||
|
"""
|
||||||
|
if path == self.config['home']:
|
||||||
|
return
|
||||||
|
|
||||||
|
if self.data[path] < increment:
|
||||||
|
self.data[path] = 0
|
||||||
|
else:
|
||||||
|
self.data[path] -= increment
|
||||||
|
|
||||||
|
self.save()
|
||||||
|
|
||||||
|
def get_weight(self, path):
|
||||||
|
return self.data[path]
|
||||||
|
|
||||||
|
def maintenance(self):
|
||||||
|
"""
|
||||||
|
Decay weights by 10%, periodically remove bottom 10% entries.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
items = self.data.iteritems()
|
||||||
|
except AttributeError:
|
||||||
|
items = self.data.items()
|
||||||
|
|
||||||
|
for path, _ in items:
|
||||||
|
self.data[path] *= 0.9
|
||||||
|
|
||||||
|
if len(self.data) > self.config['max_paths']:
|
||||||
|
remove_cnt = int(0.1 * len(self.data))
|
||||||
|
for path in sorted(self.data, key=self.data.get)[:remove_cnt]:
|
||||||
|
del self.data[path]
|
||||||
|
|
||||||
|
self.save()
|
||||||
|
|
||||||
|
def purge(self):
|
||||||
|
"""
|
||||||
|
Remove non-existent paths.
|
||||||
|
"""
|
||||||
|
removed = []
|
||||||
|
|
||||||
|
for path in list(self.data.keys()):
|
||||||
|
if not os.path.exists(path):
|
||||||
|
removed.append(path)
|
||||||
|
del self.data[path]
|
||||||
|
|
||||||
|
self.save()
|
||||||
|
return removed
|
||||||
|
|
||||||
def set_defaults():
|
def set_defaults():
|
||||||
config = {}
|
config = {}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user