tfstate: very basic terraform state backend in flask

app/tfstate.py

@@ -0,0 +1,51 @@
+import json
+
+from flask import Blueprint, request, Response
+
+from app.extensions import db
+from app.models.tfstate import TerraformState
+
+tfstate = Blueprint("tfstate", __name__)
+
+
+@tfstate.route("/<key>", methods=['GET'])
+def handle_get(key):
+    state = TerraformState.query.filter(TerraformState.key == key).first()
+    if state is None or state.state is None:
+        return "Not Found", 404
+    return Response(state.state, content_type="application/json")
+
+
+@tfstate.route("/<key>", methods=['POST', 'DELETE', 'UNLOCK'])
+def handle_update(key):
+    state = TerraformState.query.filter(TerraformState.key == key).first()
+    if not state:
+        if request.method in ["DELETE", "UNLOCK"]:
+            return "OK", 200
+        state = TerraformState(key=key)
+    if state.lock and not (request.method == "UNLOCK" and request.args.get('ID') is None):
+        if json.loads(state.lock)['ID'] != request.args.get('ID'):
+            return Response(state.lock, status=409, content_type="application/json")
+    if request.method == "POST":
+        state.state = json.dumps(request.json)
+    elif request.method == "DELETE":
+        db.session.delete(state)
+    elif request.method == "UNLOCK":
+        state.lock = None
+    db.session.commit()
+    return "OK", 200
+
+
+@tfstate.route("/<key>", methods=['LOCK'])
+def handle_lock(key):
+    state = TerraformState.query.filter(TerraformState.key == key).with_for_update().first()
+    if state is None:
+        state = TerraformState(key=key)
+        db.session.add(state)
+    if state.lock is not None:
+        lock = state.lock
+        db.session.rollback()
+        return Response(lock, status=423, content_type="application/json")
+    state.lock = json.dumps(request.json)
+    db.session.commit()
+    return "OK", 200