summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--krebs/5pkgs/simple/krebsdance/default.nix16
1 files changed, 10 insertions, 6 deletions
diff --git a/krebs/5pkgs/simple/krebsdance/default.nix b/krebs/5pkgs/simple/krebsdance/default.nix
index f200625c9..cdfe23ef2 100644
--- a/krebs/5pkgs/simple/krebsdance/default.nix
+++ b/krebs/5pkgs/simple/krebsdance/default.nix
@@ -1,5 +1,5 @@
{ writers }:
-writers.writePython3Bin "krebsdance" {} ''
+writers.writePython3Bin "krebsdance" { flakeIgnore = [ "E501" ]; } ''
import argparse
import random
import itertools
@@ -80,12 +80,16 @@ writers.writePython3Bin "krebsdance" {} ''
yield f'{claw["up"]} {body["left"]}{eye}{mouth}{eye}{body["right"]} {claw["up"]}'
+ def escape_graph(text):
+ return text.replace("\\", "\\\\")
+
+
def krebs_graph() -> str:
- return "\n".join(
- ["digraph {"]
- + [f'"{krebs}"->"{generate(seed=krebs)}"' for krebs in all_krebses()]
- + ["}"]
- )
+ return "\n".join(itertools.chain(
+ ["digraph {"],
+ [escape_graph(f'"{krebs}"->"{generate(seed=krebs)}"') for krebs in all_krebses()],
+ "}",
+ ))
def generate(*, seed: str, dancing: bool = False) -> str: