Manual mechanism for deleting builds (#94)

* Minimal implementation of deletion of conda environments

* Black formatting

conda-store-server/conda_store_server/api.py

@@ -86,7 +86,7 @@ def list_build_artifacts(db, limit: int = 25, build_id: int = None, key: str = N
     if key:
         filters.append(orm.BuildArtifact.key == key)
 
-    return db.query(orm.BuildArtifact).filter(**filters).limit(limit).all()
+    return db.query(orm.BuildArtifact).filter(*filters).limit(limit).all()
 
 
 def get_build_artifact(db, build_id, key):

conda-store-server/conda_store_server/app.py

@@ -272,3 +272,15 @@ def update_environment_build(self, name, build_id):
         from conda_store_server.worker import tasks
 
         tasks.task_update_environment_build.si(environment.id, build.id).apply_async()
+
+    def delete_build(self, build_id):
+        build = api.get_build(self.db, build_id)
+        if build.status not in [orm.BuildStatus.FAILED, orm.BuildStatus.COMPLETED]:
+            raise ValueError("cannot delete build since not finished building")
+
+        self.celery_app
+
+        # must import tasks after a celery app has been initialized
+        from conda_store_server.worker import tasks
+
+        tasks.task_delete_build.si(build.id).apply_async()

conda-store-server/conda_store_server/server/views/api.py

@@ -99,6 +99,17 @@ def api_put_build(build_id):
         return jsonify({"status": "error", "error": "build id does not exist"}), 400
 
 
+@app_api.route("/api/v1/build/<build_id>/", methods=["DELETE"])
+def api_delete_build(build_id):
+    conda_store = get_conda_store()
+    build = api.get_build(conda_store.db, build_id)
+    if build is not None:
+        conda_store.delete_build(build_id)
+        return jsonify({"status": "ok"})
+    else:
+        return jsonify({"status": "error", "error": "build id does not exist"}), 400
+
+
 @app_api.route("/api/v1/build/<build_id>/logs/", methods=["GET"])
 def api_get_build_logs(build_id):
     conda_store = get_conda_store()

conda-store-server/conda_store_server/worker/tasks.py

@@ -1,3 +1,5 @@
+import shutil
+
 from celery.decorators import task
 import yaml
 
@@ -83,3 +85,21 @@ def task_update_environment_build(environment_id, build_id):
     environment.build_id = build.id
     environment.specification_id = build.specification.id
     conda_store.db.commit()
+
+
+@task(name="task_delete_build")
+def task_delete_build(build_id):
+    conda_store = create_worker().conda_store
+    build = api.get_build(conda_store.db, build_id)
+
+    conda_prefix = build.build_path(conda_store.store_directory)
+
+    shutil.rmtree(conda_prefix)
+
+    for build_artifact in api.list_build_artifacts(
+        conda_store.db, limit=None, build_id=build_id
+    ):
+        conda_store.storage.delete(conda_store.db, build_id, build_artifact.key)
+
+    conda_store.db.delete(build)
+    conda_store.db.commit()