Fixes part of google#51: uses os.path.basename(sys.argv[0]) as default program name (google#59)

saurabhkpatel · dbieber · commit 3b04b0edb969 · 2017-04-13T14:02:34.000-07:00
* Uses os.path.basename(sys.argv[0]) for default program name
diff --git a/fire/core.py b/fire/core.py
@@ -54,6 +54,7 @@ def main(argv):
 
 import inspect
 import json
+import os
 import pipes
 import shlex
 import sys
@@ -100,7 +101,7 @@ def Fire(component=None, command=None, name=None):
   # Get args as a list.
   if command is None:
     # Use the command line args by default if no command is specified.
-    name = name or sys.argv[0]
+    name = name or os.path.basename(sys.argv[0])
     args = sys.argv[1:]
   else:
     # Otherwise use the specified command.
diff --git a/fire/fire_test.py b/fire/fire_test.py
@@ -18,6 +18,7 @@
 from __future__ import division
 from __future__ import print_function
 
+import os
 import sys
 import unittest
 
@@ -43,6 +44,15 @@ def testFire(self):
     self.assertEqual(fire.Fire(tc.OldStyleWithDefaults, 'double 2'), 4)
     self.assertEqual(fire.Fire(tc.OldStyleWithDefaults, 'triple 4'), 12)
 
+  def testFireDefaultBaseName(self):
+    with mock.patch.object(sys, 'argv',
+                           [os.path.join('python-fire', 'fire',
+                                         'base_filename.py')]):
+      # positive case
+      with self.assertOutputMatches(stdout='Usage:       base_filename.py',
+                                    stderr=None):
+        fire.Fire(tc.Empty)
+
   def testFireNoArgs(self):
     self.assertEqual(fire.Fire(tc.MixedDefaults, 'ten'), 10)
 
