From 85a2fc85a38145e0d1e4e24d10ae3c5bea384e81 Mon Sep 17 00:00:00 2001
From: gyulaid <gyulaid@gyulai.cloud>
Date: Thu, 7 Apr 2022 18:03:51 +0000
Subject: [PATCH] PyPiRunner class

---
 alice-ci/src/alice/cli.py                  |  1 -
 alice-ci/src/alice/runnerfactory.py        |  4 ++-
 alice-ci/src/alice/runners/__init__.py     |  1 +
 alice-ci/src/alice/runners/pypirunner.py   | 29 ++++++++++++++++++++++
 alice-ci/src/alice/runners/pythonrunner.py |  2 ++
 ci-examples/full.yaml                      | 14 ++++++++++-
 6 files changed, 48 insertions(+), 3 deletions(-)
 create mode 100644 alice-ci/src/alice/runners/pypirunner.py

diff --git a/alice-ci/src/alice/cli.py b/alice-ci/src/alice/cli.py
index d80f2c7..0820469 100644
--- a/alice-ci/src/alice/cli.py
+++ b/alice-ci/src/alice/cli.py
@@ -33,7 +33,6 @@ def parse_jobs(args):
                 print(f"[Alice][Step] {step}: {status}")
             else:
                 raise ConfigException(f"Step {step} not found in {args.input}")
-                exit(1)
     except ConfigException as e:
         print(f"Configuration error-> {e}")
         exit(1)
diff --git a/alice-ci/src/alice/runnerfactory.py b/alice-ci/src/alice/runnerfactory.py
index 6218362..5d7feee 100644
--- a/alice-ci/src/alice/runnerfactory.py
+++ b/alice-ci/src/alice/runnerfactory.py
@@ -1,4 +1,5 @@
 from alice.runners.pythonrunner import PythonRunner
+from alice.runners.pypirunner import PyPiRunner
 from alice.exceptions import ConfigException
 
 
@@ -15,7 +16,8 @@ class Factory():
         # https://git.gyulai.cloud/gyulaid/alice/issues/4
         # module = __import__("module_file")
         # my_class = getattr(module, "class_name")
-        runners = {"python": PythonRunner}
+        runners = {"python": PythonRunner,
+                   "pypi": PyPiRunner}
 
         if (self.verbose):
             print(f"[Alice] Available runners: {'|'.join(runners.keys())}")
diff --git a/alice-ci/src/alice/runners/__init__.py b/alice-ci/src/alice/runners/__init__.py
index 973c468..b67238e 100644
--- a/alice-ci/src/alice/runners/__init__.py
+++ b/alice-ci/src/alice/runners/__init__.py
@@ -1,2 +1,3 @@
 # flake8: noqa F401
 from alice.runners.pythonrunner import PythonRunner
+from alice.runners.pypirunner import PyPiRunner
diff --git a/alice-ci/src/alice/runners/pypirunner.py b/alice-ci/src/alice/runners/pypirunner.py
new file mode 100644
index 0000000..3eda5e9
--- /dev/null
+++ b/alice-ci/src/alice/runners/pypirunner.py
@@ -0,0 +1,29 @@
+import json
+import sys
+from urllib import request    
+from pkg_resources import parse_version   
+
+from alice.exceptions import NonZeroRetcode, RunnerError, ConfigException
+
+
+class PyPiRunner():
+    def __init__(self, params, user_defaults) -> None:
+        self.verbose = params["verbose"]
+        if self.verbose:
+            print("[PyPiRunner] Initializing")
+        self.workdir = user_defaults["workdir"]
+        # config only contains env and workdir
+        self.config = user_defaults
+        self.default_repo = "https://pypi.python.org/pypi"
+
+    def __versions(self, pkg_name):
+        # TODO: Error handling
+        url = f'{self.default_repo}/{pkg_name}/json'
+        releases = json.loads(request.urlopen(url).read())['releases']
+        return sorted(releases, key=parse_version, reverse=True) 
+
+    def update_config(self, config):
+        print(config)
+
+    def run(self, job_spec):
+        print(self.__versions("alice-ci"))
\ No newline at end of file
diff --git a/alice-ci/src/alice/runners/pythonrunner.py b/alice-ci/src/alice/runners/pythonrunner.py
index 4dbcd0b..cda91b8 100644
--- a/alice-ci/src/alice/runners/pythonrunner.py
+++ b/alice-ci/src/alice/runners/pythonrunner.py
@@ -42,6 +42,8 @@ class PythonRunner():
     # Also - dependency install by config is only allowed in this step
     def update_config(self, config):
         if "dependencies" in config:
+            # TODO: Use pip freeze to list installed packages, only install missing
+            # ...or just dump the dependencies in a single list, and let pip deal with it.
             for dependency in config["dependencies"]:
                 # TODO: Check what happens with fixed version
                 command = [self.vpython, "-m", "pip", "install", dependency, "--upgrade"]
diff --git a/ci-examples/full.yaml b/ci-examples/full.yaml
index f5b9191..565c477 100644
--- a/ci-examples/full.yaml
+++ b/ci-examples/full.yaml
@@ -28,6 +28,18 @@ jobs:
     commands:
       - "-c \"import os; print(os.environ)\""
   - name: lint
+    type: python
     workdir: alice-ci
     commands:
-      - "-m flake8 --ignore E501"
\ No newline at end of file
+      - "-m flake8 --ignore E501"
+  - name: pkg
+    type: pypi
+    workdir: .
+    upload: false
+    repo:
+      uri: example.com
+      username: asdf
+      password:
+        from_env: PYPIPASS
+    packages:
+     - alice-ci
\ No newline at end of file