From 85d3b98b647b50816a06bafa5c06c72e831fc164 Mon Sep 17 00:00:00 2001 From: jpic Date: Tue, 28 Jan 2020 02:03:46 +0100 Subject: [PATCH] Support requirements --- podctl/visitors/pip.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/podctl/visitors/pip.py b/podctl/visitors/pip.py index 3dbba31..32b6fac 100644 --- a/podctl/visitors/pip.py +++ b/podctl/visitors/pip.py @@ -2,9 +2,10 @@ import os class Pip: - def __init__(self, *pip_packages, pip=None): + def __init__(self, *pip_packages, pip=None, requirements=None): self.pip_packages = pip_packages self.pip = pip + self.requirements = requirements def build(self, script): if self.pip: @@ -34,3 +35,6 @@ class Pip: nonsource = [p for p in self.pip_packages if not p.startswith('/')] if nonsource: script.run(f'sudo $_pip install --upgrade {" ".join(source)}') + + if self.requirements: + script.run(f'sudo $_pip install --upgrade -r {self.requirements}')