brom-dump: spft-replay: refactor code (part 2 of 2)

Rename device.py and its Device class to brom.py and BromProtocol accordingly
arzam16 · Dec 1, 2023 · 9cb6847 · 9cb6847
1 parent bf6ee8f
commit 9cb6847
Show file tree

Hide file tree

Showing 4 changed files with 168 additions and 164 deletions.
diff --git a/brom-dump/spft-replay/spft-replay.py b/brom-dump/spft-replay/spft-replay.py
@@ -6,8 +6,8 @@
 import logging
 from functools import partial, partialmethod
 
+from src.brom import BromProtocol
 from src.common import as_0x
-from src.device import Device
 from src.manager import DeviceManager
 from src.transport import UsbTransport
 
@@ -99,13 +99,13 @@ def main():
     transport = UsbTransport()
     transport.start()
 
-    device = Device(transport)
+    brom = BromProtocol(transport)
     try:
-        device.handshake()
+        brom.handshake()
     except:
         logging.critical("Handshake error!", exc_info=True)
 
-    manager = DeviceManager(device)
+    manager = DeviceManager(brom)
     if args.mode_identify:
         identify_mode(manager)
     elif args.mode_payload or args.mode_simple_payload:

diff --git a/brom-dump/spft-replay/src/device.py → brom-dump/spft-replay/src/brom.py b/brom-dump/spft-replay/src/device.py → brom-dump/spft-replay/src/brom.py
@@ -8,7 +8,7 @@
 from src.common import as_0x, as_hex, from_bytes, to_bytes
 
 
-class Device:
+class BromProtocol:
     def __init__(self, transport):
         self.transport = transport
 

diff --git a/brom-dump/spft-replay/src/manager.py b/brom-dump/spft-replay/src/manager.py
@@ -44,7 +44,7 @@ def identify(self):
     def replay(self, payload, simple_mode, skip_remaining_data):
         self.payload = payload
 
-        hw_code = self.dev.get_hw_code()
+        hw_code = self.brom.get_hw_code()
         logging.replay(f"HW code: {as_hex(hw_code, 2)}")
 
         if hw_code == 0x6573: