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
|
# NOTE: Following example requires boto3 package.
import os
import boto3
from InquirerPy import inquirer
from InquirerPy.exceptions import InvalidArgument
from InquirerPy.validator import PathValidator
client = boto3.client("s3")
os.environ["INQUIRERPY_VI_MODE"] = "true"
def get_bucket(_):
return [bucket["Name"] for bucket in client.list_buckets()["Buckets"]]
def walk_s3_bucket(bucket):
response = []
paginator = client.get_paginator("list_objects")
for result in paginator.paginate(Bucket=bucket):
for file in result["Contents"]:
response.append(file["Key"])
return response
try:
action = inquirer.select(
message="Select an S3 action:", choices=["Upload", "Download"]
).execute()
if action == "Upload":
file_to_upload = inquirer.filepath(
message="Enter the filepath to upload:",
validate=PathValidator(),
only_files=True,
).execute()
bucket = inquirer.fuzzy(
message="Select a bucket:", choices=get_bucket, spinner_enable=True
).execute()
else:
bucket = inquirer.fuzzy(
message="Select a bucket:", choices=get_bucket, spinner_enable=True
).execute()
file_to_download = inquirer.fuzzy(
message="Select files to download:",
choices=lambda _: walk_s3_bucket(bucket),
multiselect=True,
spinner_enable=True,
).execute()
destination = inquirer.filepath(
message="Enter destination folder:",
only_directories=True,
validate=PathValidator(),
).execute()
confirm = inquirer.confirm(message="Confirm?").execute()
except InvalidArgument:
print("No available choices")
# Download or Upload the file based on result ...
|
