oledir: added main to make it a CLI script (issue #69)

decalage2
1 parent 34d3f3fd
Showing 2 changed files with 4 additions and 1 deletions
oletools/oledir.py
setup.py
@@ -113,7 +113,7 @@ def sid_display(sid):
 # === MAIN ===================================================================
-if __name__ == '__main__':
+def main():
     # print banner with version
     print('oledir %s - http://decalage.info/python/oletools' % __version__)
@@ -182,3 +182,5 @@ if __name__ == &#39;__main__&#39;:
     # print t
+if __name__ == '__main__':
+    main()
 \ No newline at end of file
@@ -268,6 +268,7 @@ entry_points = {
         'mraptor=oletools.mraptor:main',
         'olebrowse=oletools.olebrowse:main',
         'oleid=oletools.oleid:main',
+        'oledir=oletools.oledir:main',
         'pyxswf=oletools.pyxswf:main'
     ],
 }