def make_interface_choice() -> list:
int_list = []
for _, value in interfaces.items():
int_string = "{:<10}".format(value.name[:8]) + " "
int_string += "{:<10}".format(value.provider[:8]) + " "
int_string += "{:<10}".format(value.itype[:8]) + " "
int_string += "{:<19}".format(value.mac) + " "
int_string += "{:<15}".format(value.status[:15]) + " "
int_list.append(int_string)
cli = Check(
prompt=" Select from the interfaces below: (Using Spacebar)",
choices=int_list,
check_on_switch=colors.foreground["green"],
word_on_switch=colors.foreground["green"],
check_color=colors.foreground["green"],
word_color=colors.foreground["white"],
check="√",
shift=1,
margin=1,
)
temp_list = []
for result in cli.launch():
temp_list.append(result.split()[0])
print("")
return temp_list
def run_cli() -> Options:
"""
Run CLI and return options
"""
cli = VerticalPrompt(
[
Input("Project Name? ", default="latex_project", strip=True),
Bullet(
"Which template to use?",
choices=[
t
for t in os.listdir("./latex_up/templates") if "." not in t
],
),
Check(
"Include lists? (Press Space to check/uncheck)",
["Enumerate", "Itemize", "Description"],
),
Check(
"Choose other latex structures to include. (Press Space to check/uncheck)",
["Graphic", "Bibtex", "Equation"],
),
],
spacing=1,
)
result = cli.launch()
return Options.from_args(result[0][1], result[1][1], *result[2][1],
*result[3][1])
def ask_interfaces(_interfaces):
"""
Parameters
----------
_interfaces
Returns
-------
"""
click.echo('\n') # terminal new line
choices = [f'dns[{itf["dns"]}], ip[{itf["ip"]}]' for itf in _interfaces]
select_interfaces_cli = Check(
prompt=f"Choose interfaces: input <space> to choose, then input <enter> to finish",
choices=choices,
align=4,
margin=1,
)
selected_interfaces = select_interfaces_cli.launch()
selected_interfaces = list(
filter(
lambda itf: f'dns[{itf["dns"]}], ip[{itf["ip"]}]' in selected_interfaces,
_interfaces,
),
)
return selected_interfaces
def start_instance(self):
ec2 = boto3.resource('ec2', region_name=self.region)
instance_iterator = ec2.instances.all()
instance_list = []
for i in instance_iterator:
if not i.tags:
name = ""
else:
name_tag = [x['Value'] for x in i.tags if x['Key'] == 'Name']
name = name_tag[0] if len(name_tag) else ''
tmp = i.id + self.splitter + i.state['Name'] + self.splitter + name
instance_list.append(tmp)
if not instance_list:
print('No EC2 instance in ' + self.region + '.')
exit(127)
cli = Check(prompt="Hit <Space Bar> to check instances:",
choices=instance_list,
indent=0,
align=3,
margin=2,
shift=1,
pad_right=3,
check='✅')
selected = cli.launch()
for s in selected:
instance_id = s.split(self.splitter)[0]
print('Starting ' + instance_id + '\n')
res = ec2.Instance(instance_id).start()
def file_types_prompt(prompt, valid_file_types=VigFile.ALL):
if not prompt:
prompt = "Select one or multiple file types from the list below:"
choices = {f"{f}": f for f in VigFile if int(f) & valid_file_types == f}
instructions = "(use SPACE BAR to select each file type, ENTER to confirm your selections)"
file_types_prompt = Check(
prompt=instructions,
choices=list(choices.keys()),
check=EMOJIS.get("CHECK", ""),
shift=1,
indent=0,
margin=2,
check_color=colors.foreground["default"],
check_on_switch=colors.foreground["default"],
word_color=colors.foreground["default"],
word_on_switch=colors.bright(colors.foreground["cyan"]),
background_color=colors.background["default"],
background_on_switch=colors.background["default"],
)
file_types = []
while not file_types:
subprocess.run(["clear"])
print_message(prompt, fg="bright_yellow", bold=True, underline=True)
result = file_types_prompt.launch()
if not result:
print_error("\nYou must select at least one file type!")
pause(message="Press any key to continue...")
continue
file_types = [choices[sel] for sel in result]
return file_types
def multi_season_prompt(db_session, prompt=None, heading=None):
if not prompt:
prompt = "Select one or multiple seasons from the list below:"
all_seasons = db.Season.get_all_regular_seasons(db_session)
choices = {f"{season.year}": season.year for season in all_seasons}
instructions = "(use SPACE BAR to select each file type, ENTER to confirm your selections)"
seasons_prompt = Check(
prompt=instructions,
choices=list(choices.keys()),
check=EMOJIS.get("CHECK", ""),
shift=1,
indent=0,
margin=2,
check_color=colors.foreground["default"],
check_on_switch=colors.foreground["default"],
word_color=colors.foreground["default"],
word_on_switch=colors.bright(colors.foreground["cyan"]),
background_color=colors.background["default"],
background_on_switch=colors.background["default"],
)
selected_years = []
while not selected_years:
subprocess.run(["clear"])
if heading:
print_heading(heading, fg="bright_yellow")
print_message(prompt, wrap=True)
result = seasons_prompt.launch()
if not result:
print_error("\nYou must select at least one season!")
pause(message="Press any key to continue...")
continue
selected_years = [choices[sel] for sel in result]
return selected_years
def ask_graphs(graphs):
"""
1つのホストに対して行う
return graphs: [dicts]
"""
click.echo('\n') # terminal new line
choices = ['all']
choices.extend([f'{graph["host"]}: {graph["name"]}' for graph in graphs])
# 入力のバリデーションするのでwhile回す
while True:
select_graphs_cli = Check(
prompt=
f"Choose graph: input <space> to choose, then input <enter> to finish",
choices=choices,
align=4,
margin=1,
)
selected_graphs = select_graphs_cli.launch()
if 'all' in selected_graphs:
if len(selected_graphs) != 1:
# When select all, be able to select only all.
click.echo('Select only all.')
continue
else:
return graphs
selected_graphs = list(
filter(
lambda graph: f'{graph["host"]}: {graph["name"]}' in
selected_graphs,
graphs,
))
return selected_graphs
def mark_deletion(dir):
if os.path.exists(dir):
filenames = sorted(os.listdir(dir), reverse=True)
if filenames:
not_dotfiles = [f for f in filenames if not f.startswith(".")]
cprint(f"Mark listened with [Space]. [Enter] to continue.", "red")
cli = Check(choices=not_dotfiles,
check="X ",
check_color=colors.foreground["red"])
return cli.launch()
def list_room(client: AsyncClient):
selection=[]
user_id=ARGS.user
with open(
f"{OUTPUT_DIR}/.env/rooms_list.{user_id}.txt","w"
) as rlist:
for room_id, room in client.rooms.items():
selection.append( f"{room_id} -> {room.display_name}")
# print(selection)
cli = Check(choices=selection)
result=cli.launch()
#rlist.write(f"{room_id}, {room.display_name}\n")
final=""
for val in result:
val=re.sub(":smart4.*","",re.sub("!","",val))
final=final + val+"\n"
rlist.write(final)
return client
def ask_hosts(hosts):
"""
host を選択する
return selected_hosts: [dict]
"""
click.echo('\n') # ターミナルをリフレッシュ
select_hostnames_cli = Check(
prompt=
"Choose instance: input <space> to choose, then input <enter> to finish",
choices=[host['name'] for host in hosts],
align=4,
margin=1,
)
hostnames = select_hostnames_cli.launch()
selected_hosts = list(filter(lambda host: host['name'] in hostnames,
hosts))
return selected_hosts
def setup_experimental_flags(data):
print_wrapped(
_("EWS does not require any configuration, you only need to scan "
"a QR code when you start up EH Forwarder Bot. It’s as simple as "
"that.\n"
"\n"
"We has provided some experimental features that you can use. "
"They are not required to be enabled for EWS to work. If you do not "
"want to enable these feature, just press ENTER, and you are good to go."
))
widget = YesNo(prompt=_("Do you want to config experimental features? "),
prompt_prefix="[yN] ")
if not widget.launch(default="n"):
return
for key, value in flags_settings.items():
default, cat, params, desc = value
if data.data['flags'].get(key) is not None:
default = data.data['flags'].get(key)
print()
print(key)
print_wrapped(desc)
if cat == 'bool':
prompt_prefix = '[Yn] ' if default else '[yN] '
ans = YesNo(prompt=f"{key}? ",
prompt_prefix=prompt_prefix) \
.launch(default='y' if default else 'n')
data.data['flags'][key] = ans
elif cat == 'int':
ans = Numbers(prompt=f"{key} [{default}]? ") \
.launch(default=default)
data.data['flags'][key] = ans
elif cat == 'choices':
ans = Bullet(prompt=f"{key}?", choices=params) \
.launch(default=default)
data.data['flags'][key] = ans
elif cat == 'multiple':
ans = Check(prompt=f"{key}?",
choices=params).launch(default=default)
data.data['flags'][key] = ans
elif cat == 'str':
ans = input(f"{key} [{default}]: ")
data.data['flags'][key] = ans or default
else:
print(_("Skipped."))
print(_("Saving configurations..."), end="", flush=True)
data.save()
print(_("OK"))
def get_user_picked_domains_to_change():
# List Route 53 domains
response = r53domains.list_domains()
domain_list = []
for domain in response["Domains"]:
domain_list.append(domain["DomainName"])
# Prompt user to pick which domains they want
domains = Check(
"Which domains would you like to update? (Press space to (de)select a domain)",
choices=domain_list,
).launch()
print("\nYou selected:")
for d in domains:
print(d)
print("\n")
return domains
from bullet import Bullet
from bullet import Check
from bullet import styles
client = Check(prompt="Choose from a list: ", **styles.Example, **styles.Exam)
print('\n', end='')
result = client.launch()
print(result)
def rlaunch(self, key, depth):
results = {}
section_config = self.configuration[key]
if section_config["prompt_type"] == "Check":
ui = Check(section_config["prompt"],
choices=section_config["choices"],
check=" √",
margin=2,
check_color=colors.bright(colors.foreground["red"]),
check_on_switch=colors.bright(colors.foreground["red"]),
background_color=colors.background["black"],
background_on_switch=colors.background["white"],
word_color=colors.foreground["white"],
word_on_switch=colors.foreground["black"],
indent=depth * 2)
choices = ui.launch()
branching = section_config.get("branching")
if branching is not None:
for sub_key in choices:
branching_key = branching.get(sub_key)
if branching_key is not None:
if branching_key.startswith("."):
results[sub_key] = self.rlaunch("{}{}".format(key, branching_key), depth)
else:
results[sub_key] = self.rlaunch(branching_key, depth)
else:
raise ValueError("the key {} is not in branching {}".format(sub_key, branching.keys()))
return results
else:
return results
if section_config["prompt_type"] == "ListInput":
ui = ListInput(section_config["prompt"],
word_color=colors.foreground["yellow"],
indent=depth * 2)
results = ui.launch()
return results
if section_config["prompt_type"] == "Input":
ui = Input(section_config["prompt"],
word_color=colors.foreground["yellow"],
indent=depth * 2)
results = ui.launch()
return results
if section_config["prompt_type"] == "YesNo":
ui = YesNo(section_config["prompt"],
word_color=colors.foreground["yellow"],
default=section_config["default"] if "default" in section_config else 'y',
indent=depth * 2)
results = ui.launch()
return results
if section_config["prompt_type"] == "Bullet":
ui = Bullet(section_config["prompt"],
choices=section_config["choices"],
bullet=" >",
margin=2,
bullet_color=colors.bright(colors.foreground["cyan"]),
background_color=colors.background["black"],
background_on_switch=colors.background["black"],
word_color=colors.foreground["white"],
word_on_switch=colors.foreground["white"],
indent=depth * 2)
results = ui.launch()
return results
if section_config["prompt_type"] == "GoTo":
for sub_key in section_config["goto"]:
if sub_key.startswith("."):
sub_value = self.rlaunch("{}{}".format(key, sub_key), depth)
sub_key = sub_key[1:]
else:
sub_value = self.rlaunch(sub_key, depth)
if isinstance(sub_value, bool) or sub_value:
# If True/False or other non-empty data (! "", [], {})
results[sub_key] = sub_value
return results
def canvas_git_map_table_wizard(course : Course) -> Table:
"""Create a Canvas-Git map CSV file."""
students = course.students()
if len(students) <= 0:
warn((f"No students found for course '{course.name}'."))
return Table([])
number_of_students = len(students)
head = min(number_of_students, HEAD)
number_shown = head if head < number_of_students else "all"
table = tabulate([s.fields() for s in students[:head]], headers = "keys")
inform((f"Found {len(students)} students for this course. "
f"Showing available data for {number_shown} of them: \n\n"
f"{table}\n"))
canvas_id_key = CANVAS_LOGIN_ID
inform((f"The column '{canvas_id_key}' contains students' Canvas ID."))
cli = VerticalPrompt([
Bullet(
prompt = ASK_GIT_ID,
choices = PUBLIC_USER_FIELDS,
bullet = " >",
margin = 2,
align = 1,
shift = 1,
),
Check(
prompt = ASK_EXTRA_COLUMNS,
choices = PUBLIC_USER_FIELDS,
check = " >",
margin = 2,
align = 1,
shift = 1,
),
])
git_id_key, extra_columns = [r for (p, r) in cli.launch()]
data = []
for student in students:
row = {}
fields = student.fields()
if canvas_id_key in fields:
row[CANVAS_ID] = fields[canvas_id_key]
else:
row[CANVAS_ID] = ""
if git_id_key in fields:
row[GIT_ID] = fields[git_id_key]
else:
row[GIT_ID] = ""
for column in extra_columns:
if column in fields:
row[column] = fields[column]
else:
row[column] = ""
data.append(row)
return Table(data)
margin = 2,
bullet_color = colors.bright(colors.foreground["cyan"]),
background_color = colors.background["black"],
background_on_switch = colors.background["black"],
word_color = colors.foreground["white"],
word_on_switch = colors.foreground["white"]
),
Check("What food do you like? ",
choices = ["🍣 Sushi",
"🍜 Ramen",
"🌭 Hotdogs",
"🍔 Hamburgers",
"🍕 Pizza",
"🍝 Spaghetti",
"🍰 Cakes",
"🍩 Donuts"],
check = " √",
margin = 2,
check_color = colors.bright(colors.foreground["red"]),
check_on_switch = colors.bright(colors.foreground["red"]),
background_color = colors.background["black"],
background_on_switch = colors.background["white"],
word_color = colors.foreground["white"],
word_on_switch = colors.foreground["black"]
),
]
)
print('\n')
result = cli.launch()
cli.summarize()
# -*- coding: utf-8 -*-
"""
A module for adding shebang and encoding to python files
"""
from os import listdir
from bullet import Check
SHEBANG = "#!/usr/bin/env python3\n"
ENCODING = "# -*- coding: utf-8 -*-\n"
ALL_FILES = listdir(path=".")
PYTHON_FILES = [file for file in ALL_FILES if file.endswith(".py")]
CLI = Check(prompt="Choose an option:", choices=PYTHON_FILES)
RESULT = CLI.launch()
NEEDED = None
for file in RESULT:
with open(file=file, mode="r") as f:
FIRST_LINE = f.readline()
SECOND_LINE = f.readline()
if FIRST_LINE == SHEBANG and SECOND_LINE == ENCODING:
print(f"Shebang and encoding present in {file}.")
elif FIRST_LINE == SHEBANG and SECOND_LINE != ENCODING:
print(f"Shebang present in {file}, but not encoding.")
NEEDED = "encoding"
else:
print(f"Shebang and encoding not present in {file}.")
