pep/poezio
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix the file (/import & /export) completion on the roster tab

Browse source
This commit is contained in:
mathieui 2014-12-17 21:20:55 +01:00
parent 64ef02d586
commit 2bb4f45ed4
No known key found for this signature in database
GPG key ID: C59F84CEEFD616E3
1 changed files with 29 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
src/tabs/rostertab.py
View file

@ -16,6 +16,7 @@ import difflib
import os import os
import ssl import ssl
from os import getenv, path from os import getenv, path
from functools import partial
from . import Tab from . import Tab
@ -112,12 +113,12 @@ class RosterInfoTab(Tab):
usage=_('[/path/to/file]'), usage=_('[/path/to/file]'),
desc=_('Export your contacts into /path/to/file if specified, or $HOME/poezio_contacts if not.'), desc=_('Export your contacts into /path/to/file if specified, or $HOME/poezio_contacts if not.'),
shortdesc=_('Export your roster to a file.'), shortdesc=_('Export your roster to a file.'),
completion=self.completion_file) completion=partial(self.completion_file, 1))
self.register_command('import', self.command_import, self.register_command('import', self.command_import,
usage=_('[/path/to/file]'), usage=_('[/path/to/file]'),
desc=_('Import your contacts from /path/to/file if specified, or $HOME/poezio_contacts if not.'), desc=_('Import your contacts from /path/to/file if specified, or $HOME/poezio_contacts if not.'),
shortdesc=_('Import your roster from a file.'), shortdesc=_('Import your roster from a file.'),
completion=self.completion_file) completion=partial(self.completion_file, 1))
self.register_command('clear', self.command_clear, self.register_command('clear', self.command_clear,
shortdesc=_('Clear the info buffer.')) shortdesc=_('Clear the info buffer.'))
self.register_command('last_activity', self.command_last_activity, self.register_command('last_activity', self.command_last_activity,
@ -484,29 +485,42 @@ class RosterInfoTab(Tab):
not self.input.help_message: not self.input.help_message:
self.complete_commands(self.input) self.complete_commands(self.input)
def completion_file(self, the_input): def completion_file(self, complete_number, the_input):
""" """
Completion for /import and /export Generic quoted completion for files/paths
(use functools.partial to use directly as a completion
for a command)
""" """
text = the_input.get_text() text = the_input.get_text()
args = text.split() args = common.shell_split(text)
n = len(args) n = the_input.get_argument_position()
if n == 1: if n == complete_number:
if args[n-1] == '' or len(args) < n+1:
home = os.getenv('HOME') or '/' home = os.getenv('HOME') or '/'
return the_input.auto_completion([home, '/tmp'], '') return the_input.new_completion([home, '/tmp'], n, quotify=True)
path_ = args[n]
if path.isdir(path_):
dir_ = path_
base = ''
else: else:
the_path = text[text.index(' ')+1:] dir_ = path.dirname(path_)
base = path.basename(path_)
try: try:
names = os.listdir(the_path) names = os.listdir(dir_)
except: except OSError:
names = [] names = []
names_filtered = [name for name in names if name.startswith(base)]
if names_filtered:
names = names_filtered
if not names:
names = [path_]
end_list = [] end_list = []
for name in names: for name in names:
value = os.path.join(the_path, name) value = os.path.join(dir_, name)
if not name.startswith('.'): if not name.startswith('.'):
end_list.append(value) end_list.append(value)
return the_input.auto_completion(end_list, '') return the_input.new_completion(end_list, n, quotify=True)
@command_args_parser.ignored @command_args_parser.ignored
def command_clear(self): def command_clear(self):