olemap: added main entry point for setup.py - issue #69

decalage2
1 parent eaa43bb6
Showing 2 changed files with 6 additions and 2 deletions
oletools/olemap.py
setup.py
@@ -42,6 +42,7 @@ http://www.decalage.info/python/oletools
 # 2015-11-01 v0.01 PL: - first version
 # 2016-01-13 v0.02 PL: - improved display with tablestream, added colors
 # 2016-07-20 v0.50 SL: - added Python 3 support
+# 2016-09-05       PL: - added main entry point for setup.py
 __version__ = '0.50'
@@ -89,7 +90,7 @@ FAT_COLORS = {
 # === MAIN ===================================================================
-if __name__ == '__main__':
+def main():
     # print banner with version
     print('olemap %s - http://decalage.info/python/oletools' % __version__)
@@ -119,4 +120,5 @@ if __name__ == &#39;__main__&#39;:
     ole.close()
-
+if __name__ == '__main__':
+    main()
@@ -19,6 +19,7 @@ to install this package.
 # 2016-02-08 v0.42 PL: - added colorclass, tablestream
 # 2016-07-19 v0.50 PL: - create CLI scripts using entry points (by 2*yo)
 # 2016-07-29       PL: - use setuptools if available
+# 2016-09-05       PL: - added more entry points
 #--- TODO ---------------------------------------------------------------------
@@ -269,6 +270,7 @@ entry_points = {
         'olebrowse=oletools.olebrowse:main',
         'oleid=oletools.oleid:main',
         'oledir=oletools.oledir:main',
+        'olemap=oletools.olemap:main',
         'pyxswf=oletools.pyxswf:main'
     ],
 }