X-Git-Url: http://wien.tomnetworks.com/gitweb/?a=blobdiff_plain;f=frprog.py;h=2522bfd5449e5e317b29c012536ebbe3303db264;hb=66102daf335a37e5a9451416c3c7a77e53b18772;hp=7ea1596f4d75bb348330f77e062a2c8fb12b76b6;hpb=57f45381d4add85a253804c995ce2af4a67fb0a7;p=pyfrprog.git

diff --git a/frprog.py b/frprog.py
index 7ea1596..2522bfd 100755
--- a/frprog.py
+++ b/frprog.py
@@ -113,7 +113,8 @@ class FlashSequence(object):
 		self.address = address
 		self.data = data
 
-def read_mhx_file(fp): # needs a file handle to the desired mhx file
+def readMHXFile(filename): # desired mhx filename
+	fp = open(filename, "r")
 	retval = [] # returns a list of FlashSequence objects
 	linecount = 0
 	for line in fp:
@@ -142,6 +143,7 @@ def read_mhx_file(fp): # needs a file handle to the desired mhx file
 
 			# add flash sequence to our list
 			retval.append(FlashSequence(address, data))
+	fp.close()
 	return retval
 
 
@@ -152,15 +154,11 @@ if len(sys.argv) != 2:
 
 # read in data from mhx-file before starting
 try:
-	fp = open(sys.argv[1], "r")
-except IOError:
-	print sys.argv[0] + ": Error - couldn't open file " + sys.argv[1] + "!"
+	flashseqs = readMHXFile(sys.argv[1])
+except IOError as error:
+	print sys.argv[0] + ": Error - couldn't open file " + error.filename + "!"
 	sys.exit(1)
 
-# get list of all our address/data pairs to flash
-flashseqs = read_mhx_file(fp)
-fp.close()
-
 print "The following flash sequences have been read in:"
 for seq in flashseqs:
 	print hex(seq.address) + ":", seq.data