fix: support for local and remote apps when looking for dependencies

abcgitalbadia · Mar 25, 2022 · 2f17da1 · 2f17da1
1 parent 61a864d
commit 2f17da1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/bench/app.py b/bench/app.py
@@ -220,7 +220,7 @@ def _get_dependencies(self):
 		if self.on_disk:
 			required_deps = os.path.join(self.mount_path, self.repo,'hooks.py')
 			try:
-				print(required_apps_from_hooks(required_deps))
+				print(required_apps_from_hooks(required_deps, local=True))
 			except IndexError:
 				print(f"No dependencies for {self.repo}")
 			finally:
@@ -231,8 +231,6 @@ def _get_dependencies(self):
 		except Exception:
 			return []
 
-		return required_apps
-
 	def update_app_state(self):
 		from bench.bench import Bench
 		bench = Bench(self.bench.name)

diff --git a/bench/utils/app.py b/bench/utils/app.py
@@ -183,9 +183,10 @@ def get_required_deps(org, name, branch, deps="hooks.py"):
 	return base64.decodebytes(res["content"].encode()).decode()
 
 
-def required_apps_from_hooks(required_deps):
-	with open(required_deps) as f:
-		required_deps = f.read()
+def required_apps_from_hooks(required_deps, local=False):
+	if local:
+		with open(required_deps) as f:
+			required_deps = f.read()
 	lines = [x for x in required_deps.split("\n") if x.strip().startswith("required_apps")]
 	required_apps = eval(lines[0].strip("required_apps").strip().lstrip("=").strip())
 	return required_apps