Update

BABA_YAGA_Updater/services/readme_editor.py

@@ -1,30 +1,49 @@
-import re
+import os
 from config.settings import settings
 
 class ReadmeEditor:
     def __init__(self, file_path: str = None):
         self.file_path = file_path or settings.README_PATH
 
-    def update_section(self, new_content: str):
-        with open(self.file_path, 'r', encoding='utf-8') as f:
-            content = f.read()
+    def update_whole_readme(self, sections_content: str):
+        """
+        Ghi đè toàn bộ nội dung README.md bằng nội dung đã gộp từ các sections.
+        """
+        try:
+            # Đảm bảo thư mục cha tồn tại nếu đường dẫn là ../README.md
+            os.makedirs(os.path.dirname(os.path.abspath(self.file_path)), exist_ok=True)
+            
+            with open(self.file_path, 'w', encoding='utf-8') as f:
+                f.write(sections_content)
+            return True
+        except Exception as e:
+            print(f"Error writing to {self.file_path}: {e}")
+            return False
+
+    def inject_content_between_markers(self, file_path: str, content_to_inject: str, start_marker: str, end_marker: str):
+        """
+        Tiêm nội dung vào giữa 2 markers trong một file cụ thể.
+        """
+        import re
+        if not os.path.exists(file_path):
+            return False
+            
+        with open(file_path, 'r', encoding='utf-8') as f:
+            full_text = f.read()
 
-        # Regex to find content between markers
         pattern = re.compile(
-            f"({re.escape(settings.START_MARKER)})(.*?)({re.escape(settings.END_MARKER)})",
+            f"({re.escape(start_marker)})(.*?)({re.escape(end_marker)})",
             re.DOTALL
         )
 
-        if not pattern.search(content):
-            print(f"Markers {settings.START_MARKER} and {settings.END_MARKER} not found in {self.file_path}")
+        if not pattern.search(full_text):
             return False
 
-        updated_content = pattern.sub(
-            f"\\1\n\n{new_content}\n\n\\3",
-            content
+        updated_text = pattern.sub(
+            f"\\1\n\n{content_to_inject}\n\n\\3",
+            full_text
         )
 
-        with open(self.file_path, 'w', encoding='utf-8') as f:
-            f.write(updated_content)
-        
+        with open(file_path, 'w', encoding='utf-8') as f:
+            f.write(updated_text)
         return True