Initial commit

Author: alon-dotan-starkware

.github/workflows/main.yml

@@ -0,0 +1,18 @@
+name: docker-build
+
+on:
+  workflow_dispatch:
+
+jobs:
+   runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+      - name: Build Papyrus utilites docker image
+        uses: docker/[email protected]
+        with:
+          push: true
+          context: .
+          cache-from: type=gha,scope=buildkit-ci
+          cache-to: type=gha,mode=max,scope=buildkit-ci

Dockerfile

@@ -0,0 +1,7 @@
+FROM python:3.13
+
+COPY . /app
+
+RUN python -m pip install -r /app/src/requirements.txt
+
+CMD python /app/src/main.py

README.md

@@ -0,0 +1 @@
+# simple-python-scheduler

deployment/simple-python-scheduler-deployment.yaml

@@ -0,0 +1,17 @@
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  name: simple-python-scheduler
+spec:
+  replicas: 1
+  selector:
+    matchLabels:
+      app: simple-python-scheduler
+  template:
+    metadata:
+      labels:
+        app: simple-python-scheduler
+    spec:
+      containers:
+      - name: simple-python-scheduler
+        image: simple-python-scheduler:latest

src/main.py

@@ -0,0 +1,57 @@
+#!/usr/bin/env python
+
+import time
+import random
+import json
+
+from kubernetes import client, config, watch
+
+config.load_kube_config()
+v1 = client.CoreV1Api()
+
+scheduler_name = "foobar"
+
+
+def nodes_available():
+    ready_nodes = []
+    for n in v1.list_node().items:
+        for status in n.status.conditions:
+            if status.status == "True" and status.type == "Ready":
+                ready_nodes.append(n.metadata.name)
+    return ready_nodes
+
+
+def scheduler(name, node, namespace="default"):
+    body = client.V1Binding()
+
+    target = client.V1ObjectReference()
+    target.kind = "Node"
+    target.apiVersion = "v1"
+    target.name = node
+
+    meta = client.V1ObjectMeta()
+    meta.name = name
+
+    body.target = target
+    body.metadata = meta
+
+    return v1.create_namespaced_binding_binding(name, namespace, body)
+
+
+def main():
+    w = watch.Watch()
+    for event in w.stream(v1.list_namespaced_pod, "default"):
+        if (
+            event["object"].status.phase == "Pending"
+            and event["object"].spec.scheduler_name == scheduler_name
+        ):
+            try:
+                res = scheduler(
+                    event["object"].metadata.name, random.choice(nodes_available())
+                )
+            except client.rest.ApiException as e:
+                print(json.loads(e.body)["message"])
+
+
+if __name__ == "__main__":
+    main()

src/requirements.txt

@@ -0,0 +1 @@
+kubernetes