1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
# Copyright (C) 2009, 2010 Roman Zimbelmann <romanz@lavabit.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import os
from curses import *
from curses.ascii import *
from inspect import getargspec, ismethod
from ranger import RANGERDIR
from ranger.api import *
from ranger.container.bookmarks import ALLOWED_KEYS as ALLOWED_BOOKMARK_KEYS
from ranger.container.keymap import KeyMap, Direction, KeyMapWithDirections
# A dummy that allows the generation of docstrings in ranger.defaults.keys
class DummyKeyManager(object):
def get_context(self, _):
class Dummy(object):
def __getattr__(self, *_, **__):
return Dummy()
__call__ = __getattr__
return Dummy()
keymanager = DummyKeyManager()
class Wrapper(object):
def __init__(self, firstattr):
self.__firstattr__ = firstattr
def __getattr__(self, attr):
if attr.startswith('_'):
raise AttributeError
def wrapper(*real_args, **real_keywords):
def function(command_argument):
args, kws = real_args, real_keywords
number = command_argument.n
direction = command_argument.direction
obj = getattr(command_argument, self.__firstattr__)
fnc = getattr(obj, attr)
if number is not None or direction is not None:
args, kws = replace_narg(number, direction, fnc, args, kws)
return fnc(*args, **kws)
return function
return wrapper
# fm.enter_dir('~') is translated into lambda arg: arg.fm.enter_dir('~')
# this makes things like this possible:
# bind('gh', fm.enter_dir('~'))
#
# but NOT: (note the 2 dots)
# bind('H', fm.history.go(-1))
#
# for something like that, use the long version:
# bind('H', lambda arg: arg.fm.history.go(-1))
#
# If the method has an argument named "narg", pressing a number before
# the key will pass that number as the narg argument. If you want the
# same behaviour in a custom lambda function, you can write:
# bind('gg', fm.move(to=0))
# as:
# bind('gg', lambda arg: narg(arg.n, arg.fm.move, to=0))
fm = Wrapper('fm')
wdg = Wrapper('wdg')
DIRARG_KEYWORD = 'dirarg'
NARG_KEYWORD = 'narg'
def narg(number_, function_, *args_, **keywords_):
"""
This applies the replace_narg function to the arguments and keywords
and directly runs this function.
Example:
def foo(xyz, narg): return hash((xyz, narg))
narg(50, foo, 123) == foo(123, narg=50)
"""
args, keywords = replace_narg(number_, function_, args_, keywords_)
return function_(*args, **keywords)
def replace_narg(number, direction, function, args, keywords):
"""
This function returns (args, keywords) with one little change:
if <function> has a named argument called "narg", args and keywords
will be modified so that the value of "narg" will be <number>.
def foo(xyz, narg): pass
replace_narg(666, foo, (), {'narg': 10, 'xyz': 5})
=> (), {'narg': 666, 'xyz': 5}
replace_narg(666, foo, (1, 2), {})
=> (1, 666), {}
"""
argspec = getargspec(function).args
args = list(args)
if number is not None and NARG_KEYWORD in argspec:
try:
# is narg in args?
index = argspec.index(NARG_KEYWORD)
if ismethod(function):
index -= 1 # because of 'self'
args[index] = number
except (ValueError, IndexError):
# is narg in keywords?
keywords = dict(keywords)
keywords[NARG_KEYWORD] = number
if direction is not None and DIRARG_KEYWORD in argspec:
try:
index = argspec.index(DIRARG_KEYWORD)
if ismethod(function):
index -= 1 # because of 'self'
args[index] = direction
except (ValueError, IndexError):
# is narg in keywords?
keywords = dict(keywords)
keywords[DIRARG_KEYWORD] = direction
return args, keywords
|