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
|
#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""
hierarchical prompt usage example
"""
from __future__ import print_function, unicode_literals
from PyInquirer import style_from_dict, Token, prompt, Separator
import aftl
def get_storage_prompt(session, s):
info = session.get_storage_info(s)
name = info.StorageDescription
if not name:
name = repr(s)
return { 'name': name, 'value': s }
def get_storage(session):
storage_prompt = {
'type': 'list',
'name': 'storage',
'message': 'Select storage',
'choices': list(map(lambda s: get_storage_prompt(session, s), session.get_storage_ids())) + [{'name': 'Quit', 'value': None}]
}
answers = prompt(storage_prompt)
return answers.get('storage')
def get_object_prompt(session, object):
info = session.get_object_info(object)
name = info.Filename
if info.ObjectFormat == aftl.ObjectFormat.Association:
name += '/'
return {'name': name, 'value': (object, info.ObjectFormat) }
def get_object(session, storage, parent):
objects = session.get_object_handles(storage, aftl.ObjectFormat.Any, parent)
object_prompt = {
'type': 'list',
'name': 'object',
'message': 'Browse objects',
'choices': [{'name': '..', 'value': (None, None) }] + \
list(map(lambda o: get_object_prompt(session, o), objects))
}
answers = prompt(object_prompt)
return answers.get('object')
def get_object_action(session):
action_prompt = {
'type': 'expand',
'name': 'action',
'message': 'Object Action (press h for help)',
'choices': [
{ 'key': 'b', 'name': 'Back', 'value': 'back' },
Separator(),
{ 'key': 'd', 'name': 'Download', 'value': 'download' },
]
}
return prompt(action_prompt).get('action')
def main():
session = aftl.Device.find_first().open_session()
while True:
storage = get_storage(session)
if storage is None:
break
path = [aftl.Session.Root]
while path:
object, type = get_object(session, storage, path[-1])
if type == aftl.ObjectFormat.Association:
path.append(object)
continue
if object is None:
path.pop()
continue
while True:
action = get_object_action(session)
if action == 'back':
break
if action == 'download':
name = session.get_object_string_property(object, aftl.ObjectProperty.ObjectFilename)
with open(name, "wb") as f:
session.get_object(object, f)
if __name__ == '__main__':
main()
|
