Add auth tokens

scripts/pentest_fixtures.py

@@ -108,11 +108,12 @@ def create_workstations(*, container):
 def create_algorithms(*, container):
     algorithm_count = Algorithm.objects.count()
 
-    for ii in range(2):
+    for ii, public in enumerate([True, True]):  # TODO: Add False, False
         create_algorithm(
             algorithm_num=algorithm_count + 1 + ii,
             container=container,
             create_jobs=True,
+            public=public,
         )
 
 
@@ -400,6 +401,7 @@ def create_algorithm(
     title=None,
     editor=None,
     create_jobs=False,
+    public=False,
 ):
     assert algorithm_num is not None or (
         title is not None and editor is not None
@@ -414,6 +416,11 @@ def create_algorithm(
     inputs = _get_inputs()
     outputs = _get_outputs()
 
+    if public:
+        title += " (Public)"
+    else:
+        title += " (Non-Public)"
+
     algorithm = Algorithm.objects.create(
         title=title,
         logo=create_image_with_text(text=f"{title}, Logo"),
@@ -557,7 +564,11 @@ def _demo_container():
 
 
 def create_image_with_text(
-    *, text, image_size=(1024, 1024), font_size=50, font_path="Roboto-Regular.ttf"
+    *,
+    text,
+    image_size=(1024, 1024),
+    font_size=50,
+    font_path="Roboto-Regular.ttf",
 ):
     # Create a new image with a black background
     image = PILImage.new("RGB", image_size, color="black")