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()