[PATCH 2/4] bbremote: Convert to python3
Jan Lübbe
jlu at pengutronix.de
Tue Sep 6 06:02:34 PDT 2022
On Tue, 2022-09-06 at 12:20 +0200, Sascha Hauer wrote:
> This is the long overdue conversion from python2 to python3.
>
> Signed-off-by: Sascha Hauer <s.hauer at pengutronix.de>
> ---
> scripts/bbremote | 2 +-
> scripts/remote/controller.py | 21 +++++++++++----------
> scripts/remote/main.py | 8 ++++----
> scripts/remote/messages.py | 15 +++++++++++++--
> scripts/remote/ratp.py | 4 ++--
> scripts/remote/ratpfs.py | 4 ++--
> scripts/remote/threadstdio.py | 4 ++--
> 7 files changed, 35 insertions(+), 23 deletions(-)
>
> diff --git a/scripts/bbremote b/scripts/bbremote
> index bc5351dbae..1eeabd08d1 100755
> --- a/scripts/bbremote
> +++ b/scripts/bbremote
> @@ -1,3 +1,3 @@
> -#!/usr/bin/env python2
> +#!/usr/bin/env python3
>
> import remote.main
> diff --git a/scripts/remote/controller.py b/scripts/remote/controller.py
> index b4493591dd..a3ae260558 100644
> --- a/scripts/remote/controller.py
> +++ b/scripts/remote/controller.py
> @@ -1,4 +1,4 @@
> -#!/usr/bin/env python2
> +#!/usr/bin/env python3
> # -*- coding: utf-8 -*-
>
> from __future__ import absolute_import, division, print_function
> @@ -8,7 +8,7 @@ import logging
> import sys
> import os
> from threading import Thread
> -from Queue import Queue, Empty
> +from queue import Queue, Empty
> from .ratpfs import RatpFSServer
> from .messages import *
> from .ratp import RatpError
> @@ -105,7 +105,7 @@ class Controller(Thread):
> self.rxq = None
> self.conn.connect(timeout=5.0)
> self._txq = Queue()
> - self._stop = False
> + self._stopit = False
Is this rename needed?
> self.fsserver = RatpFSServer()
>
> def _send(self, bbpkt):
> @@ -147,24 +147,24 @@ class Controller(Thread):
> return 0
>
> def command(self, cmd):
> - self._send(BBPacketCommand(cmd=cmd))
> + self._send(BBPacketCommand(cmd=cmd.encode()))
> r = self._expect(BBPacketCommandReturn, timeout=None)
> logging.info("Command: %r", r)
> return r.exit_code
>
> def getenv(self, varname):
> - self._send(BBPacketGetenv(varname=varname))
> + self._send(BBPacketGetenv(varname=varname.encode()))
> r = self._expect(BBPacketGetenvReturn)
> return r.text
>
> def md(self, path, addr, size):
> - self._send(BBPacketMd(path=path, addr=addr, size=size))
> + self._send(BBPacketMd(path=path.encode(), addr=addr, size=size))
> r = self._expect(BBPacketMdReturn)
> logging.info("Md return: %r", r)
> return (r.exit_code,r.data)
>
> def mw(self, path, addr, data):
> - self._send(BBPacketMw(path=path, addr=addr, data=data))
> + self._send(BBPacketMw(path=path.encode(), addr=addr, data=data))
> r = self._expect(BBPacketMwReturn)
> logging.info("Mw return: %r", r)
> return (r.exit_code,r.written)
> @@ -208,7 +208,7 @@ class Controller(Thread):
> def run(self):
> assert self.rxq is not None
> try:
> - while not self._stop:
> + while not self._stopit:
> # receive
> pkt = self.conn.recv()
> if pkt:
> @@ -235,15 +235,16 @@ class Controller(Thread):
> Thread.start(self)
>
> def stop(self):
> - self._stop = True
> + self._stopit = True
> self.join()
> - self._stop = False
> + self._stopit = False
> self.rxq = None
>
> def send_async(self, pkt):
> self._txq.put(pkt)
>
> def send_async_console(self, text):
> + assert isinstance(text, bytes)
> self._txq.put(BBPacketConsoleMsg(text=text))
>
> def send_async_ping(self):
> diff --git a/scripts/remote/main.py b/scripts/remote/main.py
> index cef5d92ee2..b99dba180a 100644
> --- a/scripts/remote/main.py
> +++ b/scripts/remote/main.py
> @@ -1,4 +1,4 @@
> -#!/usr/bin/env python2
> +#!/usr/bin/env python3
>
> from __future__ import absolute_import, division, print_function
>
> @@ -7,7 +7,7 @@ import os
> import argparse
> import binascii
> import logging
> -from Queue import Queue
> +from queue import Queue
> from .ratp import RatpError
>
> try:
> @@ -162,8 +162,8 @@ def handle_console(args):
> ctrl = get_controller(args)
> ctrl.export(args.export)
> ctrl.start(queue)
> - ctrl.send_async_console('\r')
> - cons = ConsoleInput(queue, exit='\x14') # CTRL-T
> + ctrl.send_async_console(b'\r')
> + cons = ConsoleInput(queue, exit=b'\x14') # CTRL-T
> cons.start()
> try:
> while True:
> diff --git a/scripts/remote/messages.py b/scripts/remote/messages.py
> index abd331c8b6..ae36189d4c 100644
> --- a/scripts/remote/messages.py
> +++ b/scripts/remote/messages.py
> @@ -1,4 +1,4 @@
> -#!/usr/bin/env python2
> +#!/usr/bin/env python3
> # -*- coding: utf-8 -*-
>
> from __future__ import absolute_import, division, print_function
> @@ -41,12 +41,13 @@ class BBPacket(object):
> if raw is not None:
> self.unpack(raw)
> else:
> - self.payload = payload
> + self.payload = payload.encode()
>
> def __repr__(self):
> return "BBPacket(%i, %i)" % (self.p_type, self.p_flags)
>
> def _unpack_payload(self, data):
> + assert isinstance(data, bytes)
> self.payload = data
>
> def _pack_payload(self):
> @@ -63,6 +64,7 @@ class BBPacket(object):
>
> class BBPacketCommand(BBPacket):
> def __init__(self, raw=None, cmd=None):
> + assert isinstance(cmd, bytes)
> self.cmd = cmd
> super(BBPacketCommand, self).__init__(BBType.command, raw=raw)
>
> @@ -70,6 +72,7 @@ class BBPacketCommand(BBPacket):
> return "BBPacketCommand(cmd=%r)" % self.cmd
>
> def _unpack_payload(self, payload):
> + assert isinstance(payload, bytes)
> self.cmd = payload
>
> def _pack_payload(self):
> @@ -94,6 +97,8 @@ class BBPacketCommandReturn(BBPacket):
>
> class BBPacketConsoleMsg(BBPacket):
> def __init__(self, raw=None, text=None):
> + if text != None:
Use 'if text is not None:'.
> + assert isinstance(text, bytes)
> self.text = text
> super(BBPacketConsoleMsg, self).__init__(BBType.consolemsg, raw=raw)
>
> @@ -101,6 +106,7 @@ class BBPacketConsoleMsg(BBPacket):
> return "BBPacketConsoleMsg(text=%r)" % self.text
>
> def _unpack_payload(self, payload):
> + assert isinstance(payload, bytes)
> self.text = payload
>
> def _pack_payload(self):
> @@ -125,6 +131,7 @@ class BBPacketPong(BBPacket):
>
> class BBPacketGetenv(BBPacket):
> def __init__(self, raw=None, varname=None):
> + assert isinstance(varname, bytes)
> self.varname = varname
> super(BBPacketGetenv, self).__init__(BBType.getenv, raw=raw)
>
> @@ -132,6 +139,7 @@ class BBPacketGetenv(BBPacket):
> return "BBPacketGetenv(varname=%r)" % self.varname
>
> def _unpack_payload(self, payload):
> + assert isinstance(payload, bytes)
> self.varname = payload
>
> def _pack_payload(self):
> @@ -148,6 +156,7 @@ class BBPacketGetenvReturn(BBPacket):
> return "BBPacketGetenvReturn(varvalue=%s)" % self.text
>
> def _unpack_payload(self, payload):
> + assert isinstance(payload, bytes)
> self.text = payload
>
> def _pack_payload(self):
> @@ -172,6 +181,7 @@ class BBPacketFSReturn(BBPacket):
>
> class BBPacketMd(BBPacket):
> def __init__(self, raw=None, path=None, addr=None, size=None):
> + assert isinstance(path, bytes)
> self.path = path
> self.addr = addr
> self.size = size
> @@ -214,6 +224,7 @@ class BBPacketMdReturn(BBPacket):
>
> class BBPacketMw(BBPacket):
> def __init__(self, raw=None, path=None, addr=None, data=None):
> + assert isinstance(path, bytes)
> self.path = path
> self.addr = addr
> self.data = data
> diff --git a/scripts/remote/ratp.py b/scripts/remote/ratp.py
> index 44f3e2f40a..25ca442d15 100644
> --- a/scripts/remote/ratp.py
> +++ b/scripts/remote/ratp.py
> @@ -1,4 +1,4 @@
> -#!/usr/bin/env python2
> +#!/usr/bin/env python3
> # -*- coding: utf-8 -*-
>
> from __future__ import absolute_import, division, print_function
> @@ -593,7 +593,7 @@ class RatpConnection(object):
> # reassemble
> if r.c_eor:
> logging.info("Reassembling %i frames", len(self._rx_buf))
> - self._rx_queue.append(''.join(self._rx_buf))
> + self._rx_queue.append(b''.join(self._rx_buf))
> self._rx_buf = []
>
> s = RatpPacket(flags='A')
> diff --git a/scripts/remote/ratpfs.py b/scripts/remote/ratpfs.py
> index 91ca044540..957628b648 100644
> --- a/scripts/remote/ratpfs.py
> +++ b/scripts/remote/ratpfs.py
> @@ -1,4 +1,4 @@
> -#!/usr/bin/env python2
> +#!/usr/bin/env python3
> # -*- coding: utf-8 -*-
>
> from __future__ import absolute_import, division, print_function
> @@ -97,7 +97,7 @@ class RatpFSServer(object):
> os.O_TRUNC)
> path = params[4:]
> try:
> - f = os.open(self._resolve(path), flags, 0666)
> + f = os.open(self._resolve(path), flags, 0o666)
This looks like an unrelated fix.
> except OSError as e:
> return struct.pack('!II', 0, e.errno)
> h = self._alloc_handle()
> diff --git a/scripts/remote/threadstdio.py b/scripts/remote/threadstdio.py
> index db249892ac..d8ad71413c 100644
> --- a/scripts/remote/threadstdio.py
> +++ b/scripts/remote/threadstdio.py
> @@ -1,11 +1,11 @@
> -#!/usr/bin/python2
> +#!/usr/bin/env python3
>
> import os
> import sys
> import termios
> import atexit
> from threading import Thread
> -from Queue import Queue, Empty
> +from queue import Queue, Empty
>
> class ConsoleInput(Thread):
> def __init__(self, queue, exit='\x14'):
--
Pengutronix e.K. | |
Industrial Linux Solutions | http://www.pengutronix.de/ |
Peiner Str. 6-8, 31137 Hildesheim, Germany | Phone: +49-5121-206917-0 |
Amtsgericht Hildesheim, HRA 2686 | Fax: +49-5121-206917-5555 |
More information about the barebox
mailing list