Ejemplo n.º 1
0
def main():
    with open(FILENAME, 'w+') as f:
        pass
    pyb = Pyboard('/dev/ttyACM0')
    pyb.enter_raw_repl()
    t0 = datetime.today()
    calibration_tracker.setT0(t0)
    print('Syncing RTC and PC clock')
    pyb.exec_raw("rtc = pyb.RTC()\n"
                 "t = ({0},{1},{2},{3},{4},{5},{6},{7})\n"
                 "rtc.datetime(t)\n"
                 "".format(t0.year, t0.month, t0.day,
                           t0.date().weekday() + 1, t0.hour, t0.minute,
                           t0.second + 1, 0),
                 data_consumer=write_to_console)
    while True:
        now = datetime.today()
        pyb.exec_raw("rtc_time = pyb.RTC().datetime()\n"
                     "t_expected = ({0},{1},{2},{3},{4},{5},{6},{7})\n"
                     "print(str(rtc_time) + ':' + str(t_expected))".format(
                         now.year, now.month, now.day,
                         now.date().weekday() + 1, now.hour, now.minute,
                         now.second + 1, 0),
                     data_consumer=write_to_buffer)
        time.sleep(SLEEP_PERIOD)
    pyb.exit_raw_repl()
    print('Done!')
Ejemplo n.º 2
0
def main():
    pyb = Pyboard('/dev/ttyACM0')
    pyb.enter_raw_repl()
    now = datetime.utcnow()
    print('Setting UTC time')
    pyb.exec_raw(""
        "import sys\n"
        "rtc = pyb.RTC()\n"
        "rtc.init()\n"
        "t_py = ({0},{1},{2},{3},{4},{5},{6},{7})\n"
        "t = '({0}, {1}, {2}, {4}, {5}, {6})'\n"
        "rtc.datetime(t_py)\n"
        "print('UTC Time now is:')\n"
        "print(rtc.datetime())\n"
        "try:\n"
        "   with open('datetime.correction', 'a+') as f:\n"
        "       f.write('Initialized UTC RTC with:' + t + '\\n')\n"
        "       f.flush()\n"
        "       f.close()\n"
        "       print('Done writing to file')\n"
        "except Exception as exc:\n"
        "   sys.print_exception(exc)\n"
        "print('PyBoard execution finished.')\n"
        "".format(now.year, now.month, now.day, now.date().weekday() + 1, now.hour, now.minute, now.second + 1, 0),
        data_consumer=write_to_console
    )
    pyb.exit_raw_repl()
    print('Done!')
Ejemplo n.º 3
0
def main(args: List[str]) -> None:
    opts = docopt(__doc__, argv=args)
    port = opts['PORT']
    force = opts['--force']

    print('Connecting to {}'.format(port), file=sys.stderr)
    board = Pyboard(port)
    files = Files(board)

    # Specifying subdirectories DOES NOT work as they will be deleted when the
    # parent directory is deleted. Specifying top level directories DOES work.
    exclude_files = ['boot.py']

    print('Removing the contents of the file system')
    wait_for_board()
    for name in files.ls(long_format=False):
        if force or name not in exclude_files:
            try:
                files.rm(name)
            except (RuntimeError, PyboardError):
                try:
                    files.rmdir(name)
                except (RuntimeError, PyboardError):
                    print('Unknown Error removing file {}'.format(name),
                          file=sys.stderr)

    print('Done')
Ejemplo n.º 4
0
def upload_file(local_path: str, remote_path: str) -> None:
    esp_port = __get_upload_port()
    print(f"Found esp device at port {esp_port}")
    pyb = Pyboard(esp_port, rawdelay=1)
    f = Files(pyb)
    print(f.ls())
    with open(local_path, 'rb') as local_file:
        f.put(remote_path, local_file.read())
def launch_pyb_script():
    pyb = Pyboard('/dev/ttyACM0', wait=30)
    pyb.enter_raw_repl()
    print('Before exec')
    with open('endless-accel-loop.py', 'rb') as f:
        pyfile = f.read()
        pyb.exec_raw(
            pyfile,
            data_consumer=data_consumer_console_printer)
    print('After exec')
    pyb.exit_raw_repl()
    print('Exited raw REPL')
Ejemplo n.º 6
0
    def connect(self, port):

        """Connect to the selected port and fill window infos"""

        for button in self.buttons.winfo_children():
            button.config(state=NORMAL)

        self.board_bar.entryconfig(0, state=NORMAL)
        self.board_bar.entryconfig(1, label="Disconnect", command=self.disconnect)

        # Shortcuts
        self.bind("<Control-S>", lambda e: self.disconnect())
        self.bind("<Control-s>", lambda e: self.disconnect())

        self.port = port
        self.board = Pyboard(self.port)
        self.files = Files(self.board)

        self.connected = True

        self.refresh()
def main(args: List[str]) -> None:
    opts = docopt(__doc__, argv=args)
    port = opts['PORT']

    print('Connecting to {}'.format(port), file=sys.stderr)
    board = Pyboard(port)
    files = Files(board)

    print('Removing the contents of the file system')
    wait_for_board()
    for name in files.ls():
        try:
            files.rm(name)
        except (RuntimeError, PyboardError):
            files.rmdir(name)
    print('Done')
Ejemplo n.º 8
0
def put():
    """
    Uploads all necessary files to the pyboard.
    """
    if len(argv) < 2:
        print(
            "Pyboard COM port not specified. Usage: led_panel_client_put /dev/tty.wchusbserial1410"
        )
        exit(1)

    pyboard_pyboard = Pyboard(argv[1])
    pyboard_files = Files(pyboard_pyboard)

    files_to_put = led_panel_client() | max7219()
    for file_path in files_to_put:
        name = basename(file_path)
        with open(file_path) as file_object:
            data = file_object.read()
        pyboard_files.put(name, data)
def main():
    pyb = Pyboard('/dev/ttyACM0')
    pyb.enter_raw_repl()
    now = datetime.utcnow()
    print('Writing RTC correction checkpoint')
    pyb.exec_raw(
        "rtc_time = pyb.RTC().datetime()\n"
        "rtc_time_str = '({{0}}, {{1}}, {{2}}, {{3}}, {{4}}, {{5}})'.format(rtc_time[0], rtc_time[1], rtc_time[2], rtc_time[4], rtc_time[5], rtc_time[6])\n"
        "t_real = ({0},{1},{2},{3},{4},{5})\n"
        "with open('datetime.correction', 'a+') as f:\n"
        "    f.write('Current UTC RTC is:' + rtc_time_str + '\\n')\n"
        "    print('Current UTC RTC on Pyboard is:' + rtc_time_str)\n"
        "    f.write('Real UTC time is:' + str(t_real) + '\\n')\n"
        "    print('Real UTC time is:' + str(t_real))\n"
        "    f.flush()\n"
        "    f.close()\n"
        "".format(now.year, now.month, now.day, now.hour, now.minute,
                  now.second + 1),
        data_consumer=write_to_console)
    pyb.exit_raw_repl()
    print('Done!')
Ejemplo n.º 10
0
def main(args: List[str]) -> None:
    global verbose
    opts = docopt(__doc__, argv=args)
    verbose = opts['--verbose']
    root = opts['PATH']

    chdir = opts['--chdir']
    if chdir:
        os.chdir(chdir)

    port = opts['PORT']
    print('Connecting to {}'.format(port), file=sys.stderr)
    board = Pyboard(port)
    files = Files(board)
    rel_root = os.path.relpath(root, os.getcwd())

    wait_for_board()

    if os.path.isdir(root):
        to_upload = [
            os.path.join(rel_root, x)
            for x in list_files(root, opts['--exclude'])
        ]
    else:
        to_upload = [rel_root]

    created_cache = set()
    for path in progress('Uploading files', to_upload):
        local_path = os.path.abspath(path)
        remote_path = os.path.normpath(path).replace(os.path.sep, '/')
        if verbose:
            print('\n{} -> {}'.format(local_path, remote_path),
                  file=sys.stderr,
                  flush=True)
        remote_dir = os.path.dirname(path)
        if remote_dir:
            make_dirs(files, remote_dir, created_cache)
        with open(local_path, 'rb') as fd:
            files.put(remote_path, fd.read())

    print('Soft reboot', file=sys.stderr, flush=True)
    soft_reset(board)
Ejemplo n.º 11
0
class AmpyGUI(Tk):

    def __init__(self):

        # Report all exceptions to a MessageBox.
        Tk.report_callback_exception = self.show_error
        
        super(AmpyGUI, self).__init__()

        # region GUI.
        self.title("AmpyGUI - Version 1.1.0")
        self.geometry("650x250")
        self.minsize(650, 250)

        if sys.platform == "win32":
            self.iconbitmap("data/AmpyGUI_icon.ico")
        elif sys.platform == "linux":
            self.icon = Image("photo", file="data/AmpyGUI_icon.png")
            self.tk.call("wm", "iconphoto", self._w, self.icon)

        menu_bar = Menu(self)

        self.board_bar = Menu(menu_bar, tearoff=0)
        self.board_bar.add_command(label="Put MPY", command=lambda: PutFiles(self, mpy=True), state=DISABLED, accelerator="   Ctrl+M")
        self.board_bar.add_command(label="Connect", command=lambda: SelectPort(self), accelerator="   Ctrl+S")
        self.board_bar.add_command(label="WebREPL", command=lambda: webbrowser.open("http://micropython.org/webrepl/"))
        self.board_bar.add_separator()
        self.board_bar.add_command(label="Close", command=self.quit, accelerator="   Alt+F4")

        help_bar = Menu(menu_bar, tearoff=0)
        help_bar.add_command(label="GitHub page", command=lambda: webbrowser.open("https://github.com/FlorianPoot/AmpyGUI"))
        help_bar.add_command(label="About", command=lambda: About(self))

        menu_bar.add_cascade(label="Board", menu=self.board_bar)
        menu_bar.add_cascade(label="Help", menu=help_bar)
        self.config(menu=menu_bar)

        self.tree_view = ttk.Treeview(self, selectmode=BROWSE)

        vsb = ttk.Scrollbar(self, orient="vertical", command=self.tree_view.yview)
        vsb.grid(column=1, row=0, sticky=N+S)

        self.tree_view.configure(yscrollcommand=vsb.set)

        self.tree_view["columns"] = ("type", "size")

        self.tree_view.column("type", width=50, minwidth=50, stretch=NO)
        self.tree_view.column("size", width=150, minwidth=150, stretch=NO)

        self.tree_view.heading("#0", text="Name", anchor=W)
        self.tree_view.heading("type", text="Type", anchor=W)
        self.tree_view.heading("size", text="Size", anchor=W)

        self.tree_view.grid(column=0, row=0, sticky=N+S+E+W)
        self.tree_view.bind("<ButtonRelease-1>", self.select_item)

        ttk.Separator(orient=HORIZONTAL).grid(column=0, row=1, columnspan=2, sticky=N+S+E+W)

        memory = Frame(self, bd=1, relief=GROOVE)

        self.total_label = Label(memory, text="Total: N/A MB")
        self.total_label.pack(expand=YES, fill=BOTH, side="left", padx=10)

        self.free_label = Label(memory, text="Free: N/A MB")
        self.free_label.pack(expand=YES, fill=BOTH, side="left", padx=10)

        self.used_label = Label(memory, text="Used: N/A %")
        self.used_label.pack(expand=YES, fill=BOTH, side="left", padx=10)

        memory.grid(column=0, row=2, columnspan=2, pady=5)

        self.buttons = Frame(self)

        self.get_button = ttk.Button(self.buttons, text="Get", takefocus=False, command=self.get, state=DISABLED)
        self.get_button.grid(column=0, row=0, sticky=E+W, padx=5)

        ttk.Button(self.buttons, text="Put", takefocus=False, command=lambda: PutFiles(self), state=DISABLED).grid(column=1, row=0, sticky=E+W, padx=5)
        ttk.Button(self.buttons, text="MkDir", takefocus=False, command=lambda: MkDir(self), state=DISABLED).grid(column=2, row=0, sticky=E+W, padx=5)
        ttk.Button(self.buttons, text="Reset", takefocus=False, command=self.reset, state=DISABLED).grid(column=3, row=0, sticky=E+W, padx=5)

        self.remove_button = ttk.Button(self.buttons, text="Remove", takefocus=False, command=self.remove, state=DISABLED)
        self.remove_button.grid(column=4, row=0, sticky=E+W, padx=5)

        ttk.Button(self.buttons, text="Format", takefocus=False, command=self.format, state=DISABLED).grid(column=5, row=0, sticky=E+W, padx=5)

        for i in range(6):
            self.buttons.columnconfigure(i, weight=1)

        self.buttons.grid(column=0, row=3, columnspan=2, sticky=N+S+E+W, padx=10, pady=5)

        self.columnconfigure(0, weight=1)
        self.rowconfigure(0, weight=1)
        # endregion

        # region Attributes.
        self.loboris_port = False

        self.port = None
        self.board = None
        self.files = None

        self.connected = False

        self.real_paths = dict()  # ID: Path
        # endregion

        # region Shortcuts

        self.bind("<Control-S>", lambda e: SelectPort(self))
        self.bind("<Control-s>", lambda e: SelectPort(self))
        self.bind("<Control-M>", lambda e: PutFiles(self, mpy=True) if self.connected else None)
        self.bind("<Control-m>", lambda e: PutFiles(self, mpy=True) if self.connected else None)

        # endregion

        self.update()
        # Start port popup
        SelectPort(self)

        self.protocol("WM_DELETE_WINDOW", self.close)

        self.mainloop()

    def select_item(self, event=None):

        """Enable buttons if an item is selected in tree view."""

        item = self.tree_view.item(self.tree_view.focus())["text"]

        if self.connected and item != "":
            self.remove_button.config(state=NORMAL)

            # Folders are determined by the lack of extension.
            if "." in item:
                self.get_button.config(state=NORMAL)
            else:
                self.get_button.config(state=DISABLED)
        else:
            self.remove_button.config(state=DISABLED)
            self.get_button.config(state=DISABLED)

    def connect(self, port):

        """Connect to the selected port and fill window infos"""

        for button in self.buttons.winfo_children():
            button.config(state=NORMAL)

        self.board_bar.entryconfig(0, state=NORMAL)
        self.board_bar.entryconfig(1, label="Disconnect", command=self.disconnect)

        # Shortcuts
        self.bind("<Control-S>", lambda e: self.disconnect())
        self.bind("<Control-s>", lambda e: self.disconnect())

        self.port = port
        self.board = Pyboard(self.port)
        self.files = Files(self.board)

        self.connected = True

        self.refresh()

    def disconnect(self):

        """Disable buttons and release COM port"""

        for button in self.buttons.winfo_children():
            button.config(state=DISABLED)

        self.board_bar.entryconfig(0, state=DISABLED)
        self.board_bar.entryconfig(1, label="Connect", command=lambda: SelectPort(self))

        # Shortcuts
        self.bind("<Control-S>", lambda e: SelectPort(self))
        self.bind("<Control-s>", lambda e: SelectPort(self))

        self.board.close()

        self.connected = False

    def get_path(self):

        """Get path of selected item."""

        name = self.tree_view.item(self.tree_view.focus())["text"]
        item = self.tree_view.focus()

        if self.loboris_port and (name == "" or "." in name):
            name = "/flash"
            for key, values in self.real_paths.items():
                if name == values:
                    item = key

        # Folders are determined by the lack of extension.
        if item != "" and "." not in name:
            return self.real_paths[item] + "/"
        else:
            return "/"

    def list_dir(self):

        """Listdir board and insert result into tree view"""

        directories = {"": ""}
        files = self.files.ls(recursive=True)

        for file in files:
            name, size = file.split(" - ")
            directory = name.split("/")[:-1]

            for index in range(1, len(directory)):
                if "/".join(directory[:index + 1])[1:] not in directories:
                    _id = self.tree_view.insert(directories["/".join(directory[:index])[1:]], "end", text=directory[index])
                    self.real_paths[_id] = "/".join(directory[:index + 1])
                    directories["/".join(directory[:index + 1])[1:]] = _id

        for file in files:
            name, size = file.split(" - ")
            directory = name.split("/")[:-1]
            directory = "/".join(directory)[1:]

            # Folders are determined by the lack of extension.
            if "." in name:
                _id = self.tree_view.insert(directories[directory], "end", text=name.split("/")[-1], values=(name.split("/")[-1].split(".")[1].upper(), size))
            else:
                _id = self.tree_view.insert(directories[directory], "end", text=name.split("/")[-1])

            self.real_paths[_id] = name

    def clear(self):

        """Clear tree view"""

        self.real_paths = dict()
        self.tree_view.delete(*self.tree_view.get_children())

    def time_out(self):

        """Close board and spread an error popup if delay is over."""

        self.board.close()
        self.disconnect()

    def get_space_info(self):

        """Get total space, free space and used space."""

        command = """
                  import uos
                  print(uos.statvfs("{}"))
                  """.format("/" if not self.loboris_port else "/flash")

        self.board.enter_raw_repl()
        out = self.board.exec_(textwrap.dedent(command)).decode("utf-8")
        self.board.exit_raw_repl()

        out = ast.literal_eval(out)

        total = round((out[0] * out[2]) / 1024**2, 2)

        if total == 0:
            # Loboris port.
            self.loboris_port = True
            return self.get_space_info()

        self.total_label.config(text=f"Total: {total} MB")

        free = round((out[0] * out[3]) / 1024**2, 2)
        self.free_label.config(text=f"Free: {free} MB")

        used = round(((total - free) / total) * 100, 2)
        self.used_label.config(text=f"Used: {used} %")

    def refresh(self):

        """Clear treeview, Listdir and get space info."""

        # Start timer
        timer = threading.Timer(10, self.time_out)
        timer.start()

        self.clear()
        self.list_dir()
        self.select_item()
        self.get_space_info()

        timer.cancel()

    def get(self):

        """Download selected item to selected directory."""

        def get_thread():
            try:
                with open(path + "/" + name, "wb") as file:
                    file.write(self.files.get(self.real_paths[item]))
            except Exception as e:
                self.show_error(e)

            loading.close()

        name = self.tree_view.item(self.tree_view.focus())["text"]
        item = self.tree_view.focus()

        path = filedialog.askdirectory()

        loading = Loading(self, title="Downloading")
        threading.Thread(target=get_thread).start()

    def remove(self):

        """Remove selected file or folder."""

        if not messagebox.askyesno("Remove", "Are you sure you want to remove the selected file or folder ?"):
            return

        item = self.tree_view.focus()

        # Folders are determined by the lack of extension.
        if "." not in self.real_paths[item]:
            self.files.rmdir(self.real_paths[item])
        else:
            self.files.rm(self.real_paths[item])

        del self.real_paths[item]

        self.tree_view.delete(self.tree_view.focus())
        self.select_item()
        self.get_space_info()

    def reset(self):

        """Perform a soft reset."""

        self.board.serial.write(b"\x03\x04")

    def format(self):

        """Remove all files except boot.py."""

        def format_thread():
            if not self.loboris_port:
                try:
                    for value in self.real_paths.values():
                        if value != "/boot.py" and "/" not in value[1:]:

                            # Folders are determined by the lack of extension.
                            if "." not in value:
                                self.files.rmdir(value)
                            else:
                                self.files.rm(value)

                    self.refresh()
                except Exception as e:
                    self.show_error(e)
            else:
                try:
                    self.files.rmdir("/flash")
                    self.refresh()
                except (Exception, RuntimeError):
                    pass

            loading.close()

        if not messagebox.askyesno("Format", "Are you sure you want to remove all files except boot.py ?"):
            return

        loading = Loading(self, title="Formatting")
        threading.Thread(target=format_thread).start()

    def close(self):

        """Close board and destroy window."""

        try:
            self.board.close()
        except AttributeError:
            pass

        self.destroy()

    @staticmethod
    def show_error(*args):
        messagebox.showerror("Exception", args)