sys.argv is not always available (PyMySQL#739)

methane · web-flow · commit 3ab3b275e3d6 · 2018-10-29T16:41:14.000+09:00
fixes PyMySQL#736
diff --git a/pymysql/connections.py b/pymysql/connections.py
@@ -315,10 +315,12 @@ def _config(key, arg):
             '_pid': str(os.getpid()),
             '_client_version': VERSION_STRING,
         }
+        if program_name is None:
+            argv = getattr(sys, "argv")
+            if argv:
+                program_name = argv[0]
         if program_name:
             self._connect_attrs["program_name"] = program_name
-        elif sys.argv:
-            self._connect_attrs["program_name"] = sys.argv[0]
 
         if defer_connect:
             self._sock = None
