def test_banner_alternative_stream(mocked_print, repl: Riposte):
repl.banner = "foobar"
repl._process = mock.Mock(side_effect=StopIteration)
repl._parse_args = mock.Mock(return_value=True)
repl.run()
mocked_print.assert_not_called()
def test_banner(mocked_print, repl: Riposte):
repl.banner = "foobar"
repl._process = mock.Mock(side_effect=StopIteration)
repl._parse_args = mock.Mock(return_value=False)
repl.run()
mocked_print.assert_called_once_with(repl.banner)
def test_banner_alternative_stream(mocked_print, repl: Riposte):
repl.banner = "foobar"
repl.print_banner = False
repl._process = mock.Mock(side_effect=StopIteration)
repl.parse_cli_arguments = mock.Mock()
repl.run()
mocked_print.assert_not_called()
def test_banner(mocked_print, repl: Riposte):
repl.banner = "foobar"
repl.print_banner = True
repl._process = mock.Mock(side_effect=StopIteration)
repl.parse_cli_arguments = mock.Mock()
repl.run()
mocked_print.assert_called_once_with(repl.banner)
emu.error("No image set")
return
global mount_path
if mount_path is None:
emu.error("Nothing mounted. Nothing to do")
image_helper.cleanup_image_and_device(image, device)
mount_path = None
@emu.command("add_network")
def add_network_device(dev_ip, host_ip, iface_dev):
if not have_image():
emu.error("No image set")
return
if not runner:
emu.error("No runner set")
return
if runner.add_net_device(dev_ip, host_ip, iface_dev):
emu.success("Successfully added network information")
else:
emu.error("Failed to add network device")
emu.run()
@repl.command('run')
def run_command():
try:
if app.target is not None:
repl.info('Fetching %s' % app.target)
response = requests.get(app.target, headers={'User-Agent': 'XY'})
if response.status_code == 200:
repl.info('Searching for links...')
soup = BeautifulSoup(response.content, features='html.parser')
# <a> http
for link in soup.findAll(
'a', attrs={'href': re.compile("^http://")}):
repl.success('Found: %s' % link['href'])
# <a> https
for link in soup.findAll(
'a', attrs={'href': re.compile("^https://")}):
repl.success('Found: %s' % link['href'])
else:
repl.error('Failed to fetch %s' % app.target)
else:
repl.error(
'TARGET is None. Please set the TARGET. Use: set TARGET <argument>'
)
except KeyboardInterrupt:
repl.info('')
repl.run()
async def on_message(message):
if message.content.startswith('!bad apple'):
oldTimestamp = time.time()
start = oldTimestamp
seconds = 0
minutes = 0
i = 0
while i < len(frames)-1:
disp = False
while not disp:
newTimestamp = time.time()
if (newTimestamp - oldTimestamp) >= TIMEOUT:
await message.channel.send(frames[int(i)])
newTimestamp = time.time()
i += (newTimestamp - oldTimestamp)/TIMEOUT
oldTimestamp = newTimestamp
disp = True
@bad_apple.command("exit")
def shutdown():
bad_apple.status("Exiting ...")
exit()
try:
client.run(Token)#<--- replace Token with your bot token here
except:
print("This is required, please refer to the readme for instructions.")
exit()
bad_apple.run()
)
return True
@pyshellrm.complete("test")
def test_completer(text, line, start_index, end_index):
return [
host
for host in CFG_HOSTS.keys()
if host.startswith(text)
]
def get_config(config_path):
global CFG_HOSTS
config = Path(config_path)
if not config.is_file():
print("Config file not found.")
quit()
with open(config_path, 'r') as cfg:
yaml = YAML(typ="safe")
CFG_HOSTS = yaml.load(cfg)
for key in CFG_HOSTS:
if not "server" in CFG_HOSTS[key]:
raise KeyError("server")
arguments = pyshellrm._parser.parse_args()
get_config(arguments.config)
pyshellrm.run()
@eupl.command('bin')
def bin_func(num: int):
eupl.success(bin(num))
@eupl.command('get')
def post_func(url: str, file_name: str):
r = requests.get(url=url)
with open('{}.txt'.format(file_name), 'w') as file:
json.dump(r.json(), file)
eupl.success('Data from GET request dumped in {}'.format(
pathlib.Path(__file__).parent.absolute()))
@eupl.command('man')
def man_func(cmd: str):
for command, desc in MANUAL.items():
if cmd in command:
eupl.success('{} - {}'.format(command, desc))
# ADD HERE
@eupl.command('cmd')
def cmd_func():
for command in COMMANDS:
eupl.success(command)
eupl.run()
@mainrepl.command("exit")
def exit():
if sys.platform == 'win32':
os.system('cls')
else:
os.system('clear')
sys.exit(0)
@mainrepl.complete("show")
def completer_start(text, line, start_index, end_index):
return [
subcommand
for subcommand in SUBCOMMANDS
if subcommand.startswith(text)
]
@mainrepl.complete("merge")
def completer_start(text, line, start_index, end_index):
return [
subcommand
for subcommand in SUBCOMMANDS[:-1]
if subcommand.startswith(text)
]
mainrepl.run()
from riposte import Riposte
calculator = Riposte(prompt="calc:~$ ")
MEMORY = []
@calculator.command("add")
def add(x: int, y: int):
result = f"{x} + {y} = {x + y}"
MEMORY.append(result)
calculator.success(result)
@calculator.command("multiply")
def multiply(x: int, y: int):
result = f"{x} * {y} = {x * y}"
MEMORY.append(result)
calculator.success(result)
@calculator.command("memory")
def memory():
for entry in MEMORY:
calculator.print(entry)
calculator.run()
@peek.command('publickey')
def generate(filename: str):
if os.path.isfile(filename):
key = RSA.import_key(open(filename).read())
peek.print(
Palette.CYAN.format(key.publickey().export_key().decode('utf-8')))
else:
peek.error("File doesn't exist!")
@peek.command('sign')
def sign(filename: str, message: str):
if os.path.isfile(filename):
key = RSA.import_key(open(filename).read())
h = SHA256.new(message.encode('utf-8'))
signature = pkcs1_15.new(key).sign(h)
peek.print(Palette.CYAN.format(b64encode(signature).decode('utf-8')))
else:
peek.error("File doesn't exists!")
@peek.command('exit')
def exit():
peek.status("Goobye!")
sys.exit()
peek.run()
public_key_file = open(filename, 'wb')
public_key_file.write(enc_key)
blast.success("Done")
else:
blast.error("File already exists!")
@blast.command('publickey')
def publickey(filename: str):
if os.path.isfile(filename):
key = RSA.import_key(open(filename).read())
blast.print(Palette.CYAN.format(key.publickey().export_key().decode('utf-8')))
else:
blast.error("File doesn't exist!")
@blast.command('sign')
def sign(filename: str, message: str):
if os.path.isfile(filename):
key = RSA.import_key(open(filename).read())
h = SHA256.new(message.encode('utf-8'))
signature = pkcs1_15.new(key).sign(h)
blast.print(Palette.CYAN.format(b64encode(signature).decode('utf-8')))
else:
blast.error("File doesn't exists!")
@blast.command('exit')
def exit():
blast.status("Goobye!")
sys.exit()
blast.run()