Inception

2025-04-19 12:57:17 +00:00 · 2016-12-28 23:52:35 +01:00 · 2016-12-28 23:52:35 +01:00 · 1f71d3af2b
commit 1f71d3af2b
parent 64ded5aefc
16 changed files with 8580 additions and 4 deletions
--- a/.gitignore
+++ b/.gitignore
@ -1,3 +1,5 @@
+.idea/
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]
@ -28,7 +30,7 @@ var/
 #  Usually these files are written by a python script from a template
 #  before PyInstaller builds the exe, so as to inject date/other infos into it.
 *.manifest
-*.spec
+#*.spec

 # Installer logs
 pip-log.txt
@ -86,4 +88,4 @@ ENV/
 .spyderproject

 # Rope project settings
-.ropeproject
+.ropeproject
--- a/About.py
+++ b/About.py
@ -0,0 +1,74 @@
+import sys
+
+import wx                  # This module uses the new wx namespace
+import wx.html
+import wx.lib.wxpTag
+
+#---------------------------------------------------------------------------
+
+class MyAboutBox(wx.Dialog):
+    text = '''
+<html>
+<body bgcolor="#AC76DE">
+<center><table bgcolor="#458154" width="100%%" cellspacing="0"
+cellpadding="0" border="1">
+<tr>
+    <td align="center">
+    <h1>wxPython %s</h1>
+    (%s)<br>
+    Running on Python %s<br>
+    </td>
+</tr>
+</table>
+
+<p><b>wxPython</b> is a Python extension module that
+encapsulates the wxWindows GUI classes.</p>
+
+<p>This demo shows off some of the capabilities
+of <b>wxPython</b>.  Select items from the menu or tree control,
+sit back and enjoy.  Be sure to take a peek at the source code for each
+demo item so you can learn how to use the classes yourself.</p>
+
+<p><b>wxPython</b> is brought to you by <b>Robin Dunn</b> and<br>
+<b>Total Control Software,</b> Copyright (c) 1997-2011.</p>
+
+<p>
+<font size="-1">Please see <i>license.txt</i> for licensing information.</font>
+</p>
+
+<p><wxp module="wx" class="Button">
+    <param name="label" value="Okay">
+    <param name="id"    value="ID_OK">
+</wxp></p>
+</center>
+</body>
+</html>
+'''
+    def __init__(self, parent):
+        wx.Dialog.__init__(self, parent, -1, 'About the wxPython demo',)
+        html = wx.html.HtmlWindow(self, -1, size=(420, -1))
+        if "gtk2" in wx.PlatformInfo or "gtk3" in wx.PlatformInfo:
+            html.SetStandardFonts()
+        py_version = sys.version.split()[0]
+        txt = self.text % (wx.VERSION_STRING,
+                           ", ".join(wx.PlatformInfo[1:]),
+                           py_version
+                           )
+        html.SetPage(txt)
+        btn = html.FindWindowById(wx.ID_OK)
+        ir = html.GetInternalRepresentation()
+        html.SetSize( (ir.GetWidth()+25, ir.GetHeight()+25) )
+        self.SetClientSize(html.GetSize())
+        self.CentreOnParent(wx.BOTH)
+
+#---------------------------------------------------------------------------
+
+
+
+if __name__ == '__main__':
+    app = wx.App()
+    dlg = MyAboutBox(None)
+    dlg.ShowModal()
+    dlg.Destroy()
+    app.MainLoop()
+
--- a/Main.py
+++ b/Main.py
@ -0,0 +1,336 @@
+#!/usr/bin/env python
+
+import wx
+import wx.lib.inspection
+import wx.lib.mixins.inspection
+import sys
+import esptool
+import threading
+from serial.tools import list_ports
+from esptool import ESPROM
+from argparse import Namespace
+
+# stuff for debugging
+print "Python version:", sys.version
+print "wxPython version:", wx.version()
+print "esptool.py version:", esptool.version(None)
+
+__version__ = "0.1.0"
+
+# ---------------------------------------------------------------------------
+
+
+# See discussion at http://stackoverflow.com/q/41101897/131929
+class RedirectText:
+    def __init__(self, text_ctrl):
+        self.__out = text_ctrl
+        self.__pending_backspaces = 0
+
+    def write(self, string):
+        new_string = ""
+        number_of_backspaces = 0
+        for c in string:
+            if c == "\b":
+                number_of_backspaces += 1
+            else:
+                new_string += c
+
+        if self.__pending_backspaces > 0:
+            # current value minus pending backspaces plus new string
+            new_value = self.__out.GetValue()[:-1 * self.__pending_backspaces] + new_string
+            wx.CallAfter(self.__out.SetValue, new_value)
+        else:
+            wx.CallAfter(self.__out.AppendText, new_string)
+
+        self.__pending_backspaces = number_of_backspaces
+
+    def flush(self):
+        None
+
+# ---------------------------------------------------------------------------
+
+
+# ---------------------------------------------------------------------------
+class FlashingThread(threading.Thread):
+    def __init__(self, parent, config):
+        threading.Thread.__init__(self)
+        self.daemon = True
+        self.__parent = parent
+        self.__config = config
+
+    def run(self):
+        esp = ESPROM(port=self.__config.port)
+        args = Namespace()
+        args.flash_size = "detect"
+        args.flash_mode = self.__config.mode
+        args.flash_freq = "40m"
+        args.no_progress = False
+        args.verify = True
+        args.baud = self.__config.baud
+        args.addr_filename = [[int("0x00000", 0), open(self.__config.file, 'rb')]]
+        # needs connect() before each operation, see  https://github.com/espressif/esptool/issues/157
+        if self.__config.erase_before_flash:
+            esp.connect()
+            esptool.erase_flash(esp, args)
+        esp.connect()
+        esptool.write_flash(esp, args)
+
+# ---------------------------------------------------------------------------
+
+
+# ---------------------------------------------------------------------------
+# DTO between GUI and flashing thread
+class FlashConfig:
+    def __init__(self):
+        self.baud = 115200
+        self.erase_before_flash = False
+        self.mode = "qio"
+        self.file = None
+        self.port = None
+
+    def is_complete(self):
+        return self.file is not None and self.port is not None
+
+# ---------------------------------------------------------------------------
+
+
+# ---------------------------------------------------------------------------
+class NodeMcuFlasher(wx.Frame):
+
+    def __init__(self, parent, title):
+        wx.Frame.__init__(self, parent, -1, title, size=(700, 650),
+                          style=wx.DEFAULT_FRAME_STYLE | wx.NO_FULL_REPAINT_ON_RESIZE)
+        self.__config = FlashConfig()
+
+        self.__build_status_bar()
+        self.__set_icons()
+        self.__build_menu_bar()
+        self.__init_ui()
+
+        sys.stdout = RedirectText(self.console_ctrl)
+
+        self.SetMinSize((640, 480))
+        self.Centre(wx.BOTH)
+        self.Show(True)
+
+    def __init_ui(self):
+        panel = wx.Panel(self)
+
+        hbox = wx.BoxSizer(wx.HORIZONTAL)
+
+        fgs = wx.FlexGridSizer(7, 2, 10, 10)
+
+        port_label = wx.StaticText(panel, label="Serial port")
+        file_label = wx.StaticText(panel, label="NodeMCU firmware")
+        baud_label = wx.StaticText(panel, label="Baud rate")
+        flashmode_label = wx.StaticText(panel, label="Flash mode")
+        erase_label = wx.StaticText(panel, label="Erase flash")
+        console_label = wx.StaticText(panel, label="Console")
+
+        self.choice = wx.Choice(panel, choices=self.__get_serial_ports())
+        self.choice.Bind(wx.EVT_CHOICE, self.__on_select_port)
+        bmp = images.Reload.GetBitmap()
+        reload_button = wx.BitmapButton(panel, id=wx.ID_ANY, bitmap=bmp,
+                                        size=(bmp.GetWidth() + 7, bmp.GetHeight() + 7))
+        reload_button.Bind(wx.EVT_BUTTON, self.__on_reload)
+
+        file_picker = wx.FilePickerCtrl(panel, style=wx.FLP_USE_TEXTCTRL)
+        file_picker.Bind(wx.EVT_FILEPICKER_CHANGED, self.__on_pick_file)
+
+        button = wx.Button(panel, -1, "Flash NodeMCU")
+        button.Bind(wx.EVT_BUTTON, self.__on_clicked)
+
+        self.console_ctrl = wx.TextCtrl(panel, style=wx.TE_MULTILINE | wx.TE_READONLY | wx.HSCROLL)
+        self.console_ctrl.SetFont(wx.Font(13, wx.FONTFAMILY_TELETYPE, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_NORMAL))
+        self.console_ctrl.SetBackgroundColour(wx.BLACK)
+        self.console_ctrl.SetForegroundColour(wx.RED)
+        self.console_ctrl.SetDefaultStyle(wx.TextAttr(wx.RED))
+
+        serial_boxsizer = wx.BoxSizer(wx.HORIZONTAL)
+        serial_boxsizer.Add(self.choice, 1,  wx.EXPAND)
+        serial_boxsizer.AddStretchSpacer(0)
+        serial_boxsizer.Add(reload_button, 0, wx.ALIGN_RIGHT, 20)
+
+        baud_boxsizer = wx.BoxSizer(wx.HORIZONTAL)
+        baud_boxsizer.Add(wx.RadioButton(panel, name="baud-9600", label="9600", style=wx.RB_GROUP))
+        baud_boxsizer.AddSpacer(10)
+        baud_boxsizer.Add(wx.RadioButton(panel, name="baud-57600", label="57600"))
+        baud_boxsizer.AddSpacer(10)
+        baud_boxsizer.Add(wx.RadioButton(panel, name="baud-74880", label="74880"))
+        baud_boxsizer.AddSpacer(10)
+        radio_button_115200 = wx.RadioButton(panel, name="baud-115200", label="115200")
+        radio_button_115200.SetValue(True)  # checks/selects the control
+        baud_boxsizer.Add(radio_button_115200)
+        baud_boxsizer.AddSpacer(10)
+        baud_boxsizer.Add(wx.RadioButton(panel, name="baud-230400", label="230400"))
+        baud_boxsizer.AddSpacer(10)
+        baud_boxsizer.Add(wx.RadioButton(panel, name="baud-460800", label="460800"))
+        baud_boxsizer.AddSpacer(10)
+        baud_boxsizer.Add(wx.RadioButton(panel, name="baud-921600", label="921600"))
+
+        flashmode_boxsizer = wx.BoxSizer(wx.HORIZONTAL)
+        qio_button = wx.RadioButton(panel, name="mode-qio", label="Quad Flash I/O (qio)", style=wx.RB_GROUP)
+        qio_button.SetValue(True)
+        flashmode_boxsizer.Add(qio_button)
+        flashmode_boxsizer.AddSpacer(10)
+        flashmode_boxsizer.Add(wx.RadioButton(panel, name="mode-dio", label="Dual Flash I/O (dio), usually for >=4MB flash chips"))
+
+        erase_boxsizer = wx.BoxSizer(wx.HORIZONTAL)
+        qio_button = wx.RadioButton(panel, name="erase-no", label="no", style=wx.RB_GROUP)
+        qio_button.SetValue(True)
+        erase_boxsizer.Add(qio_button)
+        erase_boxsizer.AddSpacer(10)
+        erase_boxsizer.Add(wx.RadioButton(panel, name="erase-yes", label="yes, wipes all data"))
+
+        # handler for all radio button groups as these "light weight" groups can't accept dedicated handlers
+        self.Bind(wx.EVT_RADIOBUTTON, self.__on_radio_group)
+
+        fgs.AddMany([
+                    port_label, (serial_boxsizer, 1, wx.EXPAND),
+                    file_label, (file_picker, 1, wx.EXPAND),
+                    baud_label, baud_boxsizer,
+                    flashmode_label, flashmode_boxsizer,
+                    erase_label, erase_boxsizer,
+                    (wx.StaticText(panel, label="")), (button, 1, wx.EXPAND),
+                    (console_label, 1, wx.EXPAND), (self.console_ctrl, 1, wx.EXPAND)])
+        fgs.AddGrowableRow(6, 1)
+        fgs.AddGrowableCol(1, 1)
+        hbox.Add(fgs, proportion=2, flag=wx.ALL | wx.EXPAND, border=15)
+        panel.SetSizer(hbox)
+
+    def __get_serial_ports(self):
+        ports = [""]
+        for port, desc, hwid in sorted(list_ports.comports()):
+            ports.append(port)
+        return ports
+
+    def __on_reload(self, event):
+        self.choice.SetItems(self.__get_serial_ports())
+
+    def __on_radio_group(self, event):
+        rb = event.GetEventObject()
+        name = rb.GetName()
+        if name.startswith("baud-"):
+            self.__config.baud = int(name[5:])
+        elif name.startswith("mode-"):
+            self.__config.mode = name[5:]
+        elif name.startswith("erase-"):
+            self.__config.erase_before_flash = name == "erase-yes"
+
+    def __on_clicked(self, event):
+        self.console_ctrl.SetValue("")
+        worker = FlashingThread(self, self.__config)
+        worker.start()
+
+    def __on_select_port(self, event):
+        choice = event.GetEventObject()
+        self.__config.port = choice.GetString(choice.GetSelection())
+
+    def __on_pick_file(self, event):
+        self.__config.file = event.GetPath().replace("'", "")
+
+    def __set_icons(self):
+        self.SetIcon(images.Icon.GetIcon())
+
+    def __build_status_bar(self):
+        self.statusBar = self.CreateStatusBar(2, wx.ST_SIZEGRIP)
+        self.statusBar.SetStatusWidths([-2, -1])
+        status_text = "Welcome to NodeMCU PyFlasher %s" % __version__
+        self.statusBar.SetStatusText(status_text, 0)
+
+    def __build_menu_bar(self):
+
+        # File menu
+        self.menuBar = wx.MenuBar()
+        menu = wx.Menu()
+        wx.App.SetMacExitMenuItemId(wx.ID_EXIT)
+        exit_item = menu.Append(wx.ID_EXIT, "E&xit\tCtrl-Q", "Exit NodeMCU PyFlasher")
+        exit_item.SetBitmap(images.Exit.GetBitmap())
+        self.Bind(wx.EVT_MENU, self.__on_exit_app, exit_item)
+        self.menuBar.Append(menu, "&File")
+
+        # Help menu
+        menu = wx.Menu()
+        help_item = menu.Append(wx.ID_ABOUT, '&About NodeMCU PyFlasher', 'About')
+        self.Bind(wx.EVT_MENU, self.__on_help_about, help_item)
+        self.menuBar.Append(menu, '&Help')
+
+        self.SetMenuBar(self.menuBar)
+
+    # Menu methods
+    def __on_exit_app(self, event):
+        self.Close(True)
+
+    def __on_help_about(self, event):
+        from About import MyAboutBox
+        about = MyAboutBox(self)
+        about.ShowModal()
+        about.Destroy()
+
+# ---------------------------------------------------------------------------
+
+
+# ---------------------------------------------------------------------------
+class MySplashScreen(wx.SplashScreen):
+    def __init__(self):
+        wx.SplashScreen.__init__(self, images.Splash.GetBitmap(),
+                                 wx.SPLASH_CENTRE_ON_SCREEN | wx.SPLASH_TIMEOUT,
+                                 2500, None, -1)
+        self.Bind(wx.EVT_CLOSE, self.__on_close)
+        self.__fc = wx.FutureCall(2000, self.__show_main)
+
+    def __on_close(self, evt):
+        # Make sure the default handler runs too so this window gets
+        # destroyed
+        evt.Skip()
+        self.Hide()
+
+        # if the timer is still running then go ahead and show the
+        # main frame now
+        if self.__fc.IsRunning():
+            self.__fc.Stop()
+            self.__show_main()
+
+    def __show_main(self):
+        frame = NodeMcuFlasher(None, "NodeMCU PyFlasher")
+        frame.Show()
+        if self.__fc.IsRunning():
+            self.Raise()
+
+# ---------------------------------------------------------------------------
+
+
+# ----------------------------------------------------------------------------
+class App(wx.App, wx.lib.mixins.inspection.InspectionMixin):
+    def OnInit(self):
+
+        import images as i
+        global images
+        images = i
+
+        wx.SystemOptions.SetOptionInt("mac.window-plain-transition", 1)
+        self.SetAppName("NodeMCU PyFlasher")
+
+        # Create and show the splash screen.  It will then create and
+        # show the main frame when it is time to do so.  Normally when
+        # using a SplashScreen you would create it, show it and then
+        # continue on with the application's initialization, finally
+        # creating and showing the main application window(s).  In
+        # this case we have nothing else to do so we'll delay showing
+        # the main frame until later (see ShowMain above) so the users
+        # can see the SplashScreen effect.
+        splash = MySplashScreen()
+        splash.Show()
+
+        return True
+
+
+# ---------------------------------------------------------------------------
+def main():
+    app = App(False)
+    app.MainLoop()
+# ---------------------------------------------------------------------------
+
+if __name__ == '__main__':
+    __name__ = 'Main'
+    main()
--- a/README.md
+++ b/README.md
@ -1,2 +1,6 @@
-# nodemcu-pyflasher
-Self-contained NodeMCU flasher with GUI based on esptool.py
+# NodeMCU PyFlasher
+[![License](https://img.shields.io/badge/license-MIT-blue.svg?style=flat)](https://github.com/marcelstoer/nodemcu-pyflasher/blob/master/LICENSE)
+
+Self-contained NodeMCU flasher with GUI based on [esptool.py](https://github.com/espressif/esptool) and [wxPython](https://www.wxpython.org/).
+
+![Image of NodeMCU PyFlasher GUI](images/gui.png)
--- a/build.sh
+++ b/build.sh
@ -0,0 +1,7 @@
+pyinstaller --noconfirm --log-level=INFO \
+    --onefile --windowed \
+    --name="NodeMCU PyFlasher" \
+    --icon=./images/icon-256.icns \
+    --osx-bundle-identifier=com.frightanic.nodemcu-pyflasher \
+    --upx-dir=/usr/local/share/ \
+    nodemcu-pyflasher.spec
--- a/encode-bitmaps.py
+++ b/encode-bitmaps.py
@ -0,0 +1,20 @@
+
+"""
+This is a way to save the startup time when running img2py on lots of
+files...
+"""
+
+from wx.tools import img2py
+
+command_lines = [
+    "-F -n Exit images/exit.png images.py",
+    "-a -F -n Reload images/reload.png images.py",
+    "-a -F -n Splash images/splash.png images.py",
+    "-a -F -i -n Icon images/icon-256.png images.py",
+    ]
+
+if __name__ == "__main__":
+    for line in command_lines:
+        args = line.split()
+        img2py.main(args)
+
--- a/images.py
+++ b/images.py
--- a/images/NodeMCU-icon-org.png
+++ b/images/NodeMCU-icon-org.png
--- a/images/exit.png
+++ b/images/exit.png
--- a/images/gui.png
+++ b/images/gui.png
--- a/images/icon-256.icns
+++ b/images/icon-256.icns
--- a/images/icon-256.png
+++ b/images/icon-256.png
--- a/images/reload.png
+++ b/images/reload.png
--- a/images/splash.png
+++ b/images/splash.png
--- a/nodemcu-pyflasher.py
+++ b/nodemcu-pyflasher.py
@ -0,0 +1,4 @@
+#!/usr/bin/env python
+
+import Main
+Main.main()
--- a/nodemcu-pyflasher.spec
+++ b/nodemcu-pyflasher.spec
@ -0,0 +1,33 @@
+# -*- mode: python -*-
+
+block_cipher = None
+
+
+a = Analysis(['nodemcu-flasher.py', 'Main.py'],
+             pathex=['/Users/marcelstoer/Data/Python/NodeMCU-Flasher'],
+             binaries=None,
+             datas=None,
+             hiddenimports=[],
+             hookspath=[],
+             runtime_hooks=[],
+             excludes=[],
+             win_no_prefer_redirects=False,
+             win_private_assemblies=False,
+             cipher=block_cipher)
+pyz = PYZ(a.pure, a.zipped_data,
+             cipher=block_cipher)
+exe = EXE(pyz,
+          a.scripts,
+          exclude_binaries=True,
+          name='nodemcu-flasher',
+          debug=False,
+          strip=False,
+          upx=True,
+          console=True )
+coll = COLLECT(exe,
+               a.binaries,
+               a.zipfiles,
+               a.datas,
+               strip=False,
+               upx=True,
+               name='nodemcu-flasher')