Add support for environment variables (#4)

CHANGELOG.md

@@ -1,3 +1,7 @@
+## v0.2.1
+
+- Custom enviroment variables can be passed to commands
+
 ## v0.2.0
 
 - Support for cwd when executing commands

docs/api_documentation.rst

@@ -13,6 +13,8 @@ API Documentation
     :type sudo: string
     :param cwd: Working directory of the command being executed
     :type cwd: string
+    :param env: Environment variables that would be available when the command is being executed
+    :type env: dict(str, str)
     :param stream: When set to true, the output of your command will be streamed. (It does not work with piped commands)
     :type stream: bool
     :param timeout: The timeout for the process in seconds

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name='soldier',
-    version='0.2.0',
+    version='0.2.1',
     author='Yash Mehrotra',
     author_email='[email protected]',
     packages=['soldier'],

soldier/soldier.py

@@ -51,6 +51,7 @@ def __init__(self, command, **kwargs):
         self._in_shell = kwargs.get('shell', False)
         self._is_alive = False
         self._cwd = kwargs.get('cwd', None)
+        self._env = kwargs.get('env', None)
         self._stream = kwargs.get('stream', False)
         self._std_in = kwargs.get('std_in', False)
         self._output = kwargs.get('std_in', None)  # Hack, think of better way
@@ -120,6 +121,7 @@ def _run(self):
                 self._process = Popen(comm,
                                       shell=self._in_shell,
                                       cwd=self._cwd,
+                                      env=self._env,
                                       stdin=PIPE,
                                       stdout=PIPE,
                                       stderr=PIPE,

tests/test_main.py

@@ -24,6 +24,14 @@ def test_cwd(self):
         output = soldier.run('ls | grep {}'.format(self.testdir), cwd=self.testdir).\
                 output.strip()
 
+        assert output == ''
+
+    def test_cwd(self):
+        output = soldier.run('printenv', env={'TEST_VAR': 'VALUE'}).\
+                output.strip()
+
+        assert output == 'TEST_VAR=VALUE'
+
     def test_rmdir(self):
         soldier.run('rmdir {}'.format(self.testdir))
         assert not os.path.exists(os.getcwd() + '/' + self.testdir)