feat: add route for diceroll

Vyvy-vi · Aug 10, 2021 · f76f213 · f76f213
1 parent 0ec031e
commit f76f213
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/routes/diceroll.py b/routes/diceroll.py
@@ -0,0 +1,19 @@
+import random
+from fastapi import APIRouter, Query
+
+router = APIRouter()
+
+
+@router.get("/diceroll")
+def roll_dice(
+    num: int = Query(1, ge=1, le=10000),
+    sides: int = Query(6, ge=3, le=10000)
+    ):
+    return {
+        "task": f"diceroll - d{sides} x {num}",
+        "result": random.choices(
+            range(1, sides + 1),
+            k = num
+        )
+    }
+