tools/extracticon.py: use command line argument

zecho · May 11, 2019 · 5b25e08 · 5b25e08
1 parent a4787e1
commit 5b25e08
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tools/extracticon.py b/tools/extracticon.py
@@ -1,12 +1,19 @@
-#!/usr/bin/python
+#!/usr/bin/env python
 # vim: tabstop=4 shiftwidth=4 softtabstop=4 expandtab
 #
 # Extract ICONs from MZ/NE MS Windows executable file
 #
 
 from struct import *
+import sys
 
-with open('../bin/data/DUNE2.ICO', 'rb') as content_file:
+if len(sys.argv) > 1:
+    filename = sys.argv[1]
+else:
+    filename = '../bin/data/DUNE2.ICO'
+
+print 'reading %s' % (filename)
+with open(filename, 'rb') as content_file:
     content = content_file.read()
 
 icon_header = {}