1. REPLACE-ALL added.

Author: Steven Lu

ailice/modules/ABrowser.py

@@ -33,7 +33,8 @@ def ModuleInfo(self):
                                                "SEARCH-UP-BROWSER": {"func": "SearchUp", "prompt": "Search content upward from the current location.", "type": "supportive"},
                                                "GET-LINK": {"func": "GetLink", "prompt": "Get the url on the specified text fragment. The text needs to be one of those text fragments enclosed by square brackets on the page (excluding the square brackets themselves).", "type": "supportive"},
                                                "EXECUTE-JS": {"func": "ExecuteJS", "prompt": "Execute js code on the current web page, especially suitable for form operations such as entering text, clicking buttons, etc. Use triple quotes on your code.", "type": "supportive"},
-                                               "REPLACE": {"func": "Replace", "prompt": "Replace and edit content within the current page. When regexMode==True, you can use regular expressions to represent the pattern and replacement. This function is a simple wrapper for re.sub() in this mode. When regexMode==False, pattern and replacement represent literal strings. Use triple quotes to represent pattern and replacement.", "type": "supportive"},
+                                               "REPLACE": {"func": "Replace", "prompt": "Replace the matching content within the current page. When regexMode==True, you can use regular expressions to represent the pattern and replacement. This function is a simple wrapper for re.sub() in this mode. When regexMode==False, pattern and replacement represent literal strings. Use triple quotes to represent pattern and replacement.", "type": "supportive"},
+                                               "REPLACE-ALL": {"func": "ReplaceAll", "prompt": "Replace all matching content in the entire document. When regexMode==True, you can use regular expressions to represent the pattern and replacement. This function is a simple wrapper for re.sub() in this mode. When regexMode==False, pattern and replacement represent literal strings. Use triple quotes to represent pattern and replacement.", "type": "supportive"},
                                                "SAVETO": {"func": "SaveTo", "prompt": "Save the modified content to a file. If the dstPath parameter is an empty string, save it to the original file.", "type": "supportive"},
                                                }}
 
@@ -140,6 +141,9 @@ def GetLink(self, text: str, session: str) -> str:
     def Replace(self, pattern: str, replacement: str, regexMode: bool, session: str) -> str:
         return self.sessions[session].Replace(pattern, replacement, regexMode) if hasattr(self.sessions[session], "Replace") else "Replace not supported in current browser."
 
+    def ReplaceAll(self, pattern: str, replacement: str, regexMode: bool, session: str) -> str:
+        return self.sessions[session].ReplaceAll(pattern, replacement, regexMode) if hasattr(self.sessions[session], "ReplaceAll") else "ReplaceAll not supported in current browser."
+    
     def SaveTo(self, dstPath: str, session: str) -> str:
         return self.sessions[session].SaveTo(dstPath) if hasattr(self.sessions[session], "SaveTo") else "SaveTo not supported in current browser."
 
@@ -154,7 +158,7 @@ def main():
         makeServer(ABrowser,
                    {"pdfOutputDir": (args.pdfOutputDir if "" != args.pdfOutputDir.strip() else tmpdir)},
                    args.addr,
-                   ["ModuleInfo", "Browse", "Edit", "ScrollDown", "ScrollUp", "SearchDown", "SearchUp", "GetFullText", "GetLink", "ExecuteJS", "Replace", "SaveTo"]).Run()
+                   ["ModuleInfo", "Browse", "Edit", "ScrollDown", "ScrollUp", "SearchDown", "SearchUp", "GetFullText", "GetLink", "ExecuteJS", "Replace", "ReplaceAll", "SaveTo"]).Run()
 
 if __name__ == '__main__':
     main()

ailice/modules/ATextBrowser.py

@@ -11,8 +11,10 @@ def __init__(self, functions: dict[str, str]):
         self.prompt = \
 '''
 The document is in editable mode. You can edit the content using the following functions:
-#Replace and edit content within the current page. When regexMode==True, you can use regular expressions to represent the pattern and replacement. This function is a simple wrapper for re.sub() in this mode. When regexMode==False, pattern and replacement represent literal strings. Use triple quotes to represent pattern and replacement.
+#Replace the matching content within the current page. When regexMode==True, you can use regular expressions to represent the pattern and replacement. This function is a simple wrapper for re.sub() in this mode. When regexMode==False, pattern and replacement represent literal strings. Use triple quotes to represent pattern and replacement.
 REPLACE<!|pattern: str, replacement: str, regexMode: bool, session: str|!> -> str
+#Replace all matching content in the entire document. The parameters are the same as REPLACE.
+REPLACE-ALL<!|pattern: str, replacement: str, regexMode: bool, session: str|!> -> str
 #Save the modified content to a file. If the dstPath parameter is an empty string, save it to the original file.
 SAVETO<!|dstPath: str, session: str|!> -> str
 
@@ -58,8 +60,22 @@ def Replace(self, pattern: str, replacement: str, regexMode: bool) -> str:
             textNew = re.sub(pattern, replacement, self(prompt=False))
         else:
             textNew = self(prompt=False).replace(pattern, replacement)
+        msg = "Pattern NOT FOUND in current visible page. Please check: 1. If the pattern you entered is correct, such as whether you forgot to properly escape characters within the quotes. 2. Ensure that the content to be replaced is within the currently visible page (you can use the SEARCHDOWN/SEARCHUP to locate it, or directly use the REPLACE-ALL to replace all matching content).\n\n---\n\n"
+        if (self(prompt=False) != textNew) or ("" == pattern):
+            msg = "The matching contents has been replaced. \n\n---\n\n"
         self.ReplaceText(textNew, replaceAll=False)
-        return textNew + self.prompt
+        return msg + self() + self.prompt
+    
+    def ReplaceAll(self, pattern: str, replacement: str, regexMode: bool) -> str:
+        if regexMode:
+            textNew = re.sub(pattern, replacement, self.txt)
+        else:
+            textNew = self.txt.replace(pattern, replacement)
+        msg = "Pattern NOT FOUND in the entire document. Please check if the pattern you entered is correct, such as whether you forgot to properly escape characters within the quotes.\n\n---\n\n"
+        if (self.txt != textNew) or ("" == pattern):
+            msg = "The matching contents has been replaced. \n\n---\n\n"
+        self.ReplaceText(textNew, replaceAll=True)
+        return msg + self() + self.prompt
 
     def SaveTo(self, dstPath: str) -> str:
         try: