Merge pull request #6 from outbit/feature/select_python_version

Feature/select python version

README.md

@@ -38,6 +38,17 @@ Virtual environment loading from /var/folders/4b/dnp21z017cg_rbgfdtzclqlm0000gn/
 (tempyenv)(venv) $ echo "now you can pip install in your virtual environment"
 ```
 
+To specify a specific version of python
+```bash
+$ tempyenv -p python3.10
+```
+
+or
+
+```bash
+$ python3.10 -m tempyenv
+```
+
 License
 =======
 

setup.py

@@ -29,7 +29,8 @@
     url='https://github.com/outbit/tempyenv',
     license='MIT',
     install_requires=[
-        'setuptools'],
+        'setuptools',
+        'venv'],
     include_package_data=True,
     package_dir={
         '': 'src'},

src/tempyenv/__main__.py

@@ -0,0 +1,7 @@
+def main():
+    from tempyenv import cli
+    cli.main()
+
+
+if __name__ == "__main__":
+    main()

src/tempyenv/cli.py

@@ -3,18 +3,23 @@
 import os
 import sys
 
+
 class TemporaryVenvCreator:
-    def __init__(self):
+    def __init__(self, python_exec=None):
         self.temp_dir = None
         self.venv_path = None
+        if python_exec is None:
+            python_exec=sys.executable
+        else:
+            self.python_exec = python_exec
 
     def create_temporary_directory(self):
         self.temp_dir = tempfile.TemporaryDirectory()
         self.venv_path = os.path.join(self.temp_dir.name, 'venv')
 
     def create_virtual_environment(self):
         try:
-            subprocess.run(['python3', '-m', 'venv', self.venv_path], check=True)
+            subprocess.run([self.python_exec, '-m', 'venv', self.venv_path], check=True)
             print(f"Virtual environment created at {self.venv_path}")
         except subprocess.CalledProcessError as e:
             print(f"Error creating virtual environment: {e}")
@@ -33,13 +38,21 @@ def load_virtual_environment(self):
         except subprocess.CalledProcessError as e:
             print(f"Error loading virtual environment: {e}")
 
+
 def main():
+    import argparse
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-p', '--python', help='Specify the Python executable', dest='python_exec', default=sys.executable)
+    args = parser.parse_args()
+    python_exec = args.python_exec
+
     print("(tempyenv) is setting up your virtual environment...hold tight")
-    venv_creator = TemporaryVenvCreator()
+    venv_creator = TemporaryVenvCreator(python_exec)
     venv_creator.create_temporary_directory()
     venv_creator.create_virtual_environment()
     venv_creator.load_virtual_environment()
 
+
 if __name__ == "__main__":
     main()
 

tests/units/test_cli.py

@@ -17,3 +17,7 @@ def test_create_temporary_directory(self):
         cli.create_temporary_directory()
         assert(cli.temp_dir != None)
         assert(cli.venv_path != None)
+
+    def test_python_exec(self):
+        cli = TemporaryVenvCreator(python_exec="python_test")
+        assert(cli.python_exec == "python_test")