From 5941ed7909da12609e874ff51fcb3f423c289354 Mon Sep 17 00:00:00 2001
From: julius <julius@0124816.xyz>
Date: Tue, 13 Aug 2024 16:14:29 +0200
Subject: [PATCH] feat: choose marker

---
 main.py | 34 +++++++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index ebfde40..8ae7aa2 100644
--- a/main.py
+++ b/main.py
@@ -1,4 +1,5 @@
 import io
+from typing import Literal
 import itertools
 import logging
 import random
@@ -36,6 +37,35 @@ MEDIA_TYPES = {
 }
 
 
+Marker = Literal[
+    ".",
+    ",",
+    "o",
+    "v",
+    "^",
+    "<",
+    ">",
+    "1",
+    "2",
+    "3",
+    "4",
+    "8",
+    "s",
+    "p",
+    "P",
+    "*",
+    "h",
+    "H",
+    "+",
+    "x",
+    "X",
+    "D",
+    "d",
+    "|",
+    "_",
+]
+
+
 @app.get("/speckles/")
 def make_wallpaper(
     speckle_colours: str,
@@ -44,8 +74,9 @@ def make_wallpaper(
     fileformat: str = "svg",
     orientation: str | None = "landscape",
     local: bool = False,
+    marker: str | None = ".",
 ):
-    if not fileformat in MEDIA_TYPES:
+    if fileformat not in MEDIA_TYPES:
         return
     speckle_colours = speckle_colours.split(",")
     background = speckle_colours.pop(0)
@@ -83,6 +114,7 @@ def make_wallpaper(
             [random.random() * y / 8 for _ in range(speckles_per_colour)],
             c=color,
             s=size,
+            marker=random.choice(marker),
         )
 
     fig.tight_layout()