fix: fix prefix not function properly

Author: pynickle

.gitignore

@@ -1,6 +1,8 @@
 .idea/
 
 __pycache__/
+/test_env/generated/*
+!/test_env/generated/.gitkeep
 
 package-lock.json
 node_modules/

README.md

@@ -199,8 +199,8 @@ Fork me on [GitHub](https://github.com/pynickle/python-cheatsheet-redefined).
 ... """
 >>> text1_lines = text1.splitlines()
 >>> text2_lines = text2.splitlines()
->>> with open("HtmlDiff.html", "w", encoding="utf-8") as f:   # make it a html file
-...     HtmlDiff = d.make_file(text1_lines, text2_lines)
+>>> with open("generated/HtmlDiff.html", "w", encoding="utf-8") as f:
+...     HtmlDiff = d.make_file(text1_lines, text2_lines)   # make it a html file
 ...     f.write(HtmlDiff)
 ...
 3331

test_env/generated/.gitkeep

@@ -31,7 +31,7 @@ function removePythonPrefixes(code: string): string {
         }
 
         // 检查是否包含前缀
-        const hasPythonPrefix = line.includes('>>>') || line.includes('>>>') || line.includes('...');
+        const hasPythonPrefix = line.startsWith('>>>') || line.startsWith('>>>') || line.startsWith('...');
 
         // 输出行：前面不含>>>或者...的行
         if (!hasPythonPrefix) {
@@ -42,19 +42,15 @@ function removePythonPrefixes(code: string): string {
         let cleanedLine = line;
 
         // 不使用正则表达式处理前缀，保留前缀后的缩进
-        if (line.includes('>>>') && !line.includes('>>>')) {
+        if (line.startsWith('>>>') && !line.startsWith('>>>')) {
             // 处理 >>> 前缀
-            const prefixIndex = line.indexOf('>>>');
-            // 从prefixIndex+3开始，获取前缀后的内容
-            cleanedLine = line.substring(prefixIndex + 3);
-        } else if (line.includes('>>>')) {
+            cleanedLine = line.substring(3);
+        } else if (line.startsWith('>>>')) {
             // 处理HTML转义的 >>> 前缀
-            const prefixIndex = line.indexOf('>>>');
-            cleanedLine = line.substring(prefixIndex + 10); // '>>>'长度为10
-        } else if (line.includes('...')) {
+            cleanedLine = line.substring(10); // '>>>'长度为10
+        } else if (line.startsWith('...')) {
             // 处理 ... 前缀
-            const prefixIndex = line.indexOf('...');
-            cleanedLine = line.substring(prefixIndex + 3);
+            cleanedLine = line.substring(3);
         }
 
         // 去除行首可能存在的多余空格
@@ -189,9 +185,8 @@ function toggleCodePrefix(blockId: string): void {
         const currentText = codeBlock.textContent || '';
 
         // 判断当前状态并切换（检查更多可能的前缀模式）
-        if (currentText.includes('>>>') || currentText.includes('>>>') || currentText.includes('...')) {
+        if (currentText.startsWith(">>>") || currentText.startsWith(">>>")) {
             codeBlock.textContent = removePythonPrefixes(originalCode);
-            console.log(removePythonPrefixes(originalCode))
         } else {
             codeBlock.textContent = originalCode;
         }