def list_system_commands(self):
api_key = self.plugin.get_settings().global_get(['api', 'key'])
port = self.plugin.get_settings().global_get(['server', 'port'])
header = {'X-Api-Key': api_key, 'Content-Type': 'application/json'}
response = requests.get('http://127.0.0.1:%s/api/system/commands' %
port,
headers=header)
if response.status_code != 200:
return None, error_embed(author=self.plugin.get_printer_name(),
title="Error code: %i" %
response.status_code,
description=response.content)
builder = EmbedBuilder()
builder.set_title('List of system commands')
builder.set_author(name=self.plugin.get_printer_name())
builder.set_description(
'To execute a system command, use /systemcommand {command}. '
'Where command is similar to "core/restart"')
data = json.loads(response.content)
for source in data:
for comm in data[source]:
if 'name' not in comm:
continue
comm_name = comm['name']
comm_description = "%s/%s" % (source, comm['action'])
if 'command' in comm:
comm_description = "%s - %s" % (comm_description,
comm['command'])
builder.add_field(title=comm_name, text=comm_description)
return None, builder.get_embeds()
def test_embedbuilder(self):
# Success
builder = EmbedBuilder()
builder.set_title("Test Title")
builder.set_description("This is a description")
builder.set_color(COLOR_INFO)
for i in range(0, 30):
builder.add_field("a" * MAX_TITLE, "b" * MAX_VALUE)
embeds = builder.get_embeds()
self.assertEqual(8, len(embeds))
first_embed = embeds[0].get_embed()
self.assertEqual("Test Title", first_embed['title'])
self.assertEqual("This is a description", first_embed['description'])
for embed in embeds:
embed_obj = embed.get_embed()
self.assertEqual(COLOR_INFO, embed_obj['color'])
self.assertIsNotNone(embed_obj['timestamp'])
self.assertLessEqual(len(embed_obj['fields']), MAX_NUM_FIELDS)
for field in embed_obj['fields']:
self.assertEqual("a" * MAX_TITLE, field['name'])
self.assertEqual("b" * MAX_VALUE, field['value'])
if "NET_TEST" in os.environ:
self.assertTrue(self.discord.send(embeds=embeds))
def test_unicode_embed(self):
teststr = "٩(-̮̮̃-̃)۶ ٩(●̮̮̃•̃)۶ ٩(͡๏̯͡๏)۶ ٩(-̮̮̃•̃)."
embed_builder = EmbedBuilder()
embed_builder.set_title(teststr)
embed_builder.set_description(teststr)
embed_builder.set_author(teststr)
embed_builder.add_field(teststr, teststr)
embeds = embed_builder.get_embeds()
self.assertIsNotNone(embeds)
def test_dispatch(self):
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
try:
s.connect(('10.255.255.255', 1))
except Exception as e:
self.fail("Can't be tested without an internet connection")
finally:
s.close()
# Should result in 3 messages, one embed, one embed with photo, and one photo.
builder = EmbedBuilder()
builder.set_title("Test title")
builder.set_description("No snapshot")
self.discord.send(messages=builder.get_embeds())
with open(self._get_path("test_pattern.png"), "rb") as f:
builder.set_description("With snapshot")
builder.set_image(("snapshot.png", f))
self.discord.send(messages=builder.get_embeds())
f.seek(0)
self.discord.send(messages=[(None,
File(fp=f, filename="snapshot.png"))])
def status(self):
builder = EmbedBuilder()
builder.set_title('Current Status')
builder.set_author(name=self.plugin.get_printer_name())
if self.plugin.get_settings().get(['show_local_ip'],
merged=True) != 'off':
ip_addr = self.plugin.get_ip_address()
if ip_addr != '127.0.0.1':
builder.add_field(title='Local IP', text=ip_addr, inline=True)
if self.plugin.get_settings().get(['show_external_ip'],
merged=True) != 'off':
builder.add_field(title='External IP',
text=self.plugin.get_external_ip_address(),
inline=True)
operational = self.plugin.get_printer().is_operational()
builder.add_field(title='Operational',
text='Yes' if operational else 'No',
inline=True)
current_data = self.plugin.get_printer().get_current_data()
if current_data.get('currentZ'):
builder.add_field(title='Current Z',
text=str(current_data['currentZ']),
inline=True)
if operational:
temperatures = self.plugin.get_printer().get_current_temperatures()
for heater in temperatures.keys():
if heater == 'bed':
continue
if temperatures[heater]['actual'] is None or len(
str(temperatures[heater]['actual'])) == 0:
continue
builder.add_field(title='Extruder Temp (%s)' % heater,
text=str(temperatures[heater]['actual']),
inline=True)
if temperatures['bed']['actual']:
builder.add_field(title='Bed Temp',
text=str(temperatures['bed']['actual']),
inline=True)
printing = self.plugin.get_printer().is_printing()
builder.add_field(title='Printing',
text='Yes' if printing else 'No',
inline=True)
if printing:
builder.add_field(title='File',
text=str(
current_data['job']['file']['name']),
inline=True)
completion = current_data['progress']['completion']
if completion:
builder.add_field(title='Progress',
text='%d%%' % completion,
inline=True)
builder.add_field(title='Time Spent',
text=self.plugin.get_print_time_spent(),
inline=True)
builder.add_field(title='Time Remaining',
text=self.plugin.get_print_time_remaining(),
inline=True)
builder.add_field(title='ETA',
text=self.plugin.get_print_eta(),
inline=True)
try:
cmd_response = subprocess.Popen(
['vcgencmd', 'get_throttled'],
stdout=subprocess.PIPE).communicate()
throttled_string = cmd_response[0].decode().split('=')[1].strip()
throttled_value = int(throttled_string, 0)
if throttled_value & (1 << 0):
builder.add_field(title='WARNING',
text="PI is under-voltage",
inline=True)
if throttled_value & (1 << 1):
builder.add_field(title='WARNING',
text="PI has capped it's ARM frequency",
inline=True)
if throttled_value & (1 << 2):
builder.add_field(title='WARNING',
text="PI is currently throttled",
inline=True)
if throttled_value & (1 << 3):
builder.add_field(title='WARNING',
text="PI has reached temperature limit",
inline=True)
if throttled_value & (1 << 16):
builder.add_field(title='WARNING',
text="PI Under-voltage has occurred",
inline=True)
if throttled_value & (1 << 17):
builder.add_field(title='WARNING',
text="PI ARM frequency capped has occurred",
inline=True)
if throttled_value & (1 << 18):
builder.add_field(title='WARNING',
text="PI Throttling has occurred",
inline=True)
if throttled_value & (1 << 19):
builder.add_field(title='WARNING',
text="PI temperature limit has occurred",
inline=True)
except OSError as e:
pass
snapshot = self.plugin.get_snapshot()
if snapshot:
builder.set_image(snapshot)
return builder.get_embeds()
def timelapse(self):
path = os.path.join(os.getcwd(), self.plugin._data_folder, '..', '..',
'timelapse')
path = os.path.abspath(path)
builder = EmbedBuilder()
builder.set_title('Files and Details')
builder.set_description('Download with /gettimelapse {filename}')
builder.set_author(name=self.plugin.get_printer_name())
baseurl = self.plugin.get_settings().get(["baseurl"])
port = self.plugin.get_port()
if baseurl is None or baseurl == "":
baseurl = "%s:%s" % (self.plugin.get_ip_address(), port)
for root, dirs, files in os.walk(path):
for name in files:
try:
file_path = os.path.join(root, name)
title = os.path.basename(file_path)
description = ''
description += 'Size: %s\n' % os.path.getsize(file_path)
description += 'Date of Creation: %s\n' % time.ctime(
os.path.getctime(file_path))
description += 'Download Path: %s\n' % \
("http://" + baseurl + "/downloads/timelapse/" + urllib.parse.quote(title))
builder.add_field(title=title, text=description)
except Exception as e:
pass
return builder.get_embeds()
