25
25
# If this fails, we can't continue and will bomb below
26
26
try :
27
27
import esptool
28
- except ( ImportError , ModuleNotFoundError ) as e :
28
+ except ImportError :
29
29
sys .stderr .write (
30
30
"\n *** pyserial or esptool directories not found next to upload.py tool (this script) ***\n "
31
31
)
38
38
def make_erase_pair (addr : str , dest_size : int , block_size = 2 ** 16 ):
39
39
dest , path = tempfile .mkstemp ()
40
40
41
- buffer = b"\xff " * block_size
41
+ buffer = bytearray ( b"\xff " * block_size )
42
42
while dest_size :
43
43
unaligned = dest_size % block_size
44
44
45
- src_size = block_size
46
45
if unaligned :
47
46
src = buffer [unaligned :]
48
47
src_size = unaligned
49
48
else :
50
49
src = buffer
50
+ src_size = block_size
51
51
52
52
os .write (dest , src )
53
53
dest_size -= src_size
@@ -57,7 +57,7 @@ def make_erase_pair(addr: str, dest_size: int, block_size=2**16):
57
57
def maybe_remove (path ):
58
58
try :
59
59
os .remove (path )
60
- except :
60
+ except Exception :
61
61
pass
62
62
63
63
atexit .register (maybe_remove , path )
@@ -66,9 +66,9 @@ def maybe_remove(path):
66
66
67
67
argv = sys .argv [1 :] # Remove executable name
68
68
69
- cmdline = [] # type: List[str ]
70
- write_options = ["--flash_size" , "detect" ] # type: List[str ]
71
- erase_options = []
69
+ cmdline : List [ str ] = [ ]
70
+ write_options : List [ str ] = ["--flash_size" , "detect" ]
71
+ erase_options : List [ str ] = []
72
72
73
73
thisarg = ""
74
74
lastarg = ""
@@ -115,11 +115,11 @@ def maybe_remove(path):
115
115
116
116
try :
117
117
esptool .main (cmdline )
118
- except :
118
+ except Exception :
119
119
etype , evalue , _ = sys .exc_info ()
120
120
estring = "\n " .join (traceback .format_exception_only (etype , value = evalue ))
121
121
122
- sys .stderr .write (f "\n *** A fatal upload.py error occurred ***\n " )
122
+ sys .stderr .write ("\n *** upload.py fatal error ***\n " )
123
123
sys .stderr .write (estring )
124
124
sys .stderr .flush ()
125
125
0 commit comments