[PATCH 2/4] bbremote: Convert to python3
Sascha Hauer
s.hauer at pengutronix.de
Tue Sep 6 03:20:47 PDT 2022
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
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:
+ 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)
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'):
--
2.30.2
More information about the barebox
mailing list