fix: if disc count lower than 3, it's crash

aicostsqo-io · Mar 18, 2024 · 2bdee34 · 2bdee34
1 parent 63f255f
commit 2bdee34
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/marble-sculp/api.py b/marble-sculp/api.py
@@ -286,7 +286,11 @@ async def extend(request: Request, payload: DiscModel):
                     # if i == 0 and j == 0:
                     #     continue
                     # for d in range(len(payload.data)):
-                    for d in range(3):
+                    if len(payload.data) >= 3:
+                        adet = 3
+                    else:
+                        adet = len(payload.data)
+                    for d in range(adet):
                         discon = deepcopy(payload.data[d])
 
                         temp_circ = Circle(radius=10)