mirror of https://github.com/tilde-team/botany
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
class LoginCompleter:
|
|
""" A loop-based completion system for logins """
|
|
def __init__(self, menu):
|
|
self.s = ""
|
|
self.logins = None
|
|
self.completions = []
|
|
# completion_id has a value of -1 for the base user input
|
|
# and between 0 and len(completions)-1 for completions
|
|
self.completion_id = -1
|
|
self.completion_base = ""
|
|
self.menu = menu
|
|
|
|
def initialize(self):
|
|
""" Initialise the list of completable logins """
|
|
garden = self.menu.user_data.retrieve_garden_from_db()
|
|
self.logins = set()
|
|
for plant_id in garden:
|
|
if not garden[plant_id]:
|
|
continue
|
|
entry = garden[plant_id]
|
|
if "owner" in entry:
|
|
self.logins.add(entry["owner"])
|
|
self.logins = sorted(list(self.logins))
|
|
|
|
def update_input(self, s):
|
|
""" Update the user input and reset completion base """
|
|
self.s = s
|
|
self.completion_base = self.s
|
|
self.completion_id = -1
|
|
|
|
def complete(self, direction = 1):
|
|
"""
|
|
Returns the completed string from the user input
|
|
Loops forward in the list of logins if direction is positive, and
|
|
backwards if direction is negative
|
|
"""
|
|
def loginFilter(x):
|
|
return x.startswith(self.s) & (x != self.s)
|
|
|
|
# Refresh possible completions after the user edits
|
|
if self.completion_id == -1:
|
|
if self.logins is None:
|
|
self.initialize()
|
|
self.completion_base = self.s
|
|
self.completions = list(filter(loginFilter, self.logins))
|
|
|
|
self.completion_id += direction
|
|
# Loop from the back
|
|
if self.completion_id == -2:
|
|
self.completion_id = len(self.completions) - 1
|
|
# If we are at the base input, return it
|
|
if self.completion_id == -1 or self.completion_id == len(self.completions):
|
|
self.completion_id = -1
|
|
return self.completion_base
|
|
return self.completions[self.completion_id]
|