diff options
| -rw-r--r-- | ranger/api/commands.py | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/ranger/api/commands.py b/ranger/api/commands.py index 8a9df1ec..f66be7ee 100644 --- a/ranger/api/commands.py +++ b/ranger/api/commands.py @@ -164,6 +164,38 @@ class Command(FileManagerAware): self._setting_line = result return result + def parse_flags(self): + """Finds and returns flags in the command + + >>> Command("").parse_flags() + ('', '') + >>> Command("foo").parse_flags() + ('', '') + >>> Command("shell test").parse_flags() + ('', 'test') + >>> Command("shell -t ls -l").parse_flags() + ('t', 'ls -l') + >>> Command("shell -f -- -q test").parse_flags() + ('f', '-q test') + >>> Command("shell -foo -bar rest of the command").parse_flags() + ('foobar', 'rest of the command') + """ + flags = "" + args = self.line.split() + rest = "" + if len(args) > 0: + rest = self.line[len(args[0]):].lstrip() + for arg in args[1:]: + if arg == "--": + rest = rest[2:].lstrip() + break + elif len(arg) > 1 and arg[0] == "-": + rest = rest[len(arg):].lstrip() + flags += arg[1:] + else: + break + return flags, rest + # XXX: Lazy properties? Not so smart? self.line can change after all! @lazy_property def _tabinsert_left(self): @@ -343,3 +375,8 @@ class AliasCommand(Command): _line = "" def execute(self): self.fm.execute_console(self._line + ' ' + self.rest(1)) + + +if __name__ == '__main__': + import doctest + doctest.testmod() |