diff --git a/cme/protocols/rdp.py b/cme/protocols/rdp.py index cc3e9178..194a51ad 100644 --- a/cme/protocols/rdp.py +++ b/cme/protocols/rdp.py @@ -202,9 +202,9 @@ class rdp(connection): if self.conn is not None and self.conn.desktop_buffer_has_data is True: buffer = self.conn.get_desktop_buffer(VIDEO_FORMAT.PIL) - filename = os.path.expanduser('~/.cme/screenshots/{}_{}_{}'.format(self.hostname, self.host, datetime.now().strftime("%Y-%m-%d_%H%M%S"))) + filename = os.path.expanduser('~/.cme/screenshots/{}_{}_{}.png'.format(self.hostname, self.host, datetime.now().strftime("%Y-%m-%d_%H%M%S"))) buffer.save(filename,'png') - self.logger.highlight("Screenshot saved {}".format(filename + ".png")) + self.logger.highlight("Screenshot saved {}".format(filename)) def screenshot(self): asyncio.run(self.screen())