Add --copy-env (#95)

doc/changelog.rst

@@ -1,6 +1,12 @@
 Changelog
 =========
 
+Version 2.2.0 (TBA)
+-------------------
+
+* ``--track-memory`` option now supports macOS by using ``psutil``.
+* Added ``--copy-env`` command line option that inherits all environment variables.
+
 Version 2.1.0 (2021-01-14)
 --------------------------
 
@@ -674,4 +680,3 @@ Version 0.1 (2016-06-02)
 ------------------------
 
 * First public release
-

doc/runner.rst

@@ -96,6 +96,8 @@ Option::
     --python-names REF_NAME:CHANGED_NAME
     --affinity=CPU_LIST
     --inherit-environ=VARS
+    --copy-env
+    --no-locale
     --track-memory
     --tracemalloc
 
@@ -119,6 +121,7 @@ Option::
   only the following variables are inherited: ``PATH``, ``HOME``, ``TEMP``,
   ``COMSPEC``, ``SystemRoot`` and locale environment variables. See the
   ``--no-locale`` below for locale environment variables.
+* ``--copy-env``: Inherit all environment variables.
 * ``--no-locale``: Don't inherit locale environment variables:
 
   - ``LANG``

pyperf/_manager.py

@@ -73,7 +73,8 @@ def worker_cmd(self, calibrate_loops, calibrate_warmups, wpipe):
 
     def spawn_worker(self, calibrate_loops, calibrate_warmups):
         env = create_environ(self.args.inherit_environ,
-                             self.args.locale)
+                             self.args.locale,
+                             self.args.copy_env)
 
         rpipe, wpipe = create_pipe()
         with rpipe:

pyperf/_runner.py

@@ -201,6 +201,9 @@ def __init__(self, values=None, warmups=None, processes=None,
                             help='Comma-separated list of environment '
                                  'variables inherited by worker child '
                                  'processes.')
+        parser.add_argument("--copy-env",
+                            dest="copy_env", action="store_true", default=False,
+                            help="Copy all environment variables")
         parser.add_argument("--no-locale",
                             dest="locale", action="store_false", default=True,
                             help="Don't copy locale environment variables "

pyperf/_utils.py

@@ -270,7 +270,9 @@ def abs_executable(python):
     return os.path.normpath(python)
 
 
-def create_environ(inherit_environ, locale):
+def create_environ(inherit_environ, locale, copy_all):
+    if copy_all:
+        return os.environ
     env = {}
 
     copy_env = ["PATH", "HOME", "TEMP", "COMSPEC", "SystemRoot", "SystemDrive"]