Add triple quotes export option for python docstrings (#295)

lewish · May 9, 2024 · 2e804a1 · 2e804a1
1 parent 1f439b7
commit 2e804a1
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/client/export.tsx b/client/export.tsx
@@ -20,7 +20,7 @@ import { useObserver } from "mobx-react";
 import * as React from "react";
 
 export interface IExportConfig {
-  wrapper?: "star" | "star-filled" | "hash" | "slash" | "three-slashes" | "dash" | "apostrophe" | "semicolon" | "backticks" | "four-spaces";
+  wrapper?: "star" | "star-filled" | "triple-quotes" | "hash" | "slash" | "three-slashes" | "dash" | "apostrophe" | "semicolon" | "backticks" | "four-spaces";
   indent?: number;
   characters?: "basic" | "extended";
 }
@@ -84,6 +84,9 @@ export function ExportDialog({
                 <MenuItem value={"star-filled"}>
                   Filled multi-line <CommentTypeChip label="/***/" />
                 </MenuItem>
+                <MenuItem value={"triple-quotes"}>
+                  Quotes multi-line <CommentTypeChip label='""" """' />
+                </MenuItem>
                 <MenuItem value={"hash"}>
                   Hashes <CommentTypeChip label="#" />
                 </MenuItem>
@@ -204,6 +207,13 @@ function applyConfig(text: string, exportConfig: IExportConfig) {
         " */",
       ]);
     }
+    if (exportConfig.wrapper === "triple-quotes") {
+      setLines([
+        exportConfig.characters === "basic" ? "\"\"\"" : "u\"\"\"",
+        ...lines(),
+        "\"\"\"",
+      ]);
+    }
     if (exportConfig.wrapper === "hash") {
       setLines(lines().map((line) => `# ${line}`));
     }