2 import sys, os, stat, fnmatch
3 from bup import options, git, shquote, vfs
4 from bup.helpers import *
9 def node_name(text, n):
10 if stat.S_ISDIR(n.mode):
12 elif stat.S_ISLNK(n.mode):
18 class OptionError(Exception):
25 a,all include hidden files in the listing
27 ls_opt = options.Options(ls_optspec, onabort=OptionError)
31 (opt, flags, extra) = ls_opt.parse(cmd_args)
32 except OptionError, e:
37 for path in (extra or ['.']):
38 n = pwd.try_resolve(path)
40 if stat.S_ISDIR(n.mode):
43 if opt.all or not len(name)>1 or not name.startswith('.'):
44 L.append(node_name(name, sub))
46 L.append(node_name(path, n))
47 print columnate(L, '')
50 def write_to_file(inf, outf):
51 for blob in chunkyreader(inf):
56 if os.isatty(sys.stdin.fileno()):
59 yield raw_input('bup> ')
63 for line in sys.stdin:
67 def _completer_get_subs(line):
68 (qtype, lastword) = shquote.unfinished_word(line)
69 (dir,name) = os.path.split(lastword)
70 #log('\ncompleter: %r %r %r\n' % (qtype, lastword, text))
73 subs = list(filter(lambda x: x.name.startswith(name),
75 except vfs.NoSuchFile, e:
77 return (dir, name, qtype, lastword, subs)
80 def find_readline_lib():
81 """Return the name (and possibly the full path) of the readline library
82 linked to the given readline module.
85 f = open(readline.__file__, "rb")
91 m = re.search('\0([^\0]*libreadline[^\0]*)\0', data)
97 def init_readline_vars():
98 """Work around trailing space automatically inserted by readline.
99 See http://bugs.python.org/issue5833"""
103 # python before 2.5 didn't have the ctypes module; but those
104 # old systems probably also didn't have this readline bug, so
107 lib_name = find_readline_lib()
108 if lib_name is not None:
109 lib = ctypes.cdll.LoadLibrary(lib_name)
110 global rl_completion_suppress_append
111 rl_completion_suppress_append = ctypes.c_int.in_dll(lib,
112 "rl_completion_suppress_append")
115 rl_completion_suppress_append = None
118 def completer(text, state):
121 global rl_completion_suppress_append
122 if rl_completion_suppress_append is not None:
123 rl_completion_suppress_append.value = 1
125 line = readline.get_line_buffer()[:readline.get_endidx()]
126 if _last_line != line:
127 _last_res = _completer_get_subs(line)
129 (dir, name, qtype, lastword, subs) = _last_res
130 if state < len(subs):
132 sn1 = sn.try_resolve() # find the type of any symlink target
133 fullname = os.path.join(dir, sn.name)
134 if stat.S_ISDIR(sn1.mode):
135 ret = shquote.what_to_add(qtype, lastword, fullname+'/',
138 ret = shquote.what_to_add(qtype, lastword, fullname,
139 terminate=True) + ' '
145 traceback.print_tb(sys.exc_traceback)
146 except Exception, e2:
147 log('Error printing traceback: %s\n' % e2)
148 log('\nError in completion: %s\n' % e)
152 bup ftp [commands...]
154 o = options.Options(optspec)
155 (opt, flags, extra) = o.parse(sys.argv[1:])
157 git.check_repo_or_die()
159 top = vfs.RefList(None)
169 log('* readline module not available: line editing disabled.\n')
173 readline.set_completer_delims(' \t\n\r/')
174 readline.set_completer(completer)
175 readline.parse_and_bind("tab: complete")
182 words = [word for (wordstart,word) in shquote.quotesplit(line)]
183 cmd = words[0].lower()
184 #log('execute: %r %r\n' % (cmd, parm))
190 for parm in words[1:]:
191 np = np.resolve(parm)
192 if not stat.S_ISDIR(np.mode):
193 raise vfs.NotDir('%s is not a directory' % parm)
198 for parm in words[1:]:
199 write_to_file(pwd.resolve(parm).open(), sys.stdout)
201 if len(words) not in [2,3]:
203 raise Exception('Usage: get <filename> [localname]')
205 (dir,base) = os.path.split(rname)
206 lname = len(words)>2 and words[2] or base
207 inf = pwd.resolve(rname).open()
208 log('Saving %r\n' % lname)
209 write_to_file(inf, open(lname, 'wb'))
211 for parm in words[1:]:
212 (dir,base) = os.path.split(parm)
213 for n in pwd.resolve(dir).subs():
214 if fnmatch.fnmatch(n.name, base):
216 log('Saving %r\n' % n.name)
218 outf = open(n.name, 'wb')
219 write_to_file(inf, outf)
223 log(' error: %s\n' % e)
224 elif cmd == 'help' or cmd == '?':
225 log('Commands: ls cd pwd cat get mget help quit\n')
226 elif cmd == 'quit' or cmd == 'exit' or cmd == 'bye':
230 raise Exception('no such command %r' % cmd)
233 log('error: %s\n' % e)