Create milestone content and append it

gogmilestone2.py

@@ -60,25 +60,30 @@ def extract_csrf_token(html):
         return match.group(1)
     return None
 
-def update_milestone(title):
+def update_milestone(pr_data):
+    login()
     """Update milestone with PR title using CSRF token."""
     csrf_token = get_csrf_token()
     if not csrf_token:
         print("Failed to get CSRF token.")
         return False
 
+    content = create_milestone_content(pr_data)
+
+    with open("content.md", "w") as file:
+        file.write(content)
+
     payload = {
         "_csrf": csrf_token,
-        "title": title,
-        "content": "PR updated via webhook",
+        "content": content,
         "deadline": "2025-03-31"
     }
     headers = {
         "Content-Type": "application/x-www-form-urlencoded"
     }
 
-    print(f"Sending POST request to update milestone with title: '{title}'...")
     response = session.post(EDIT_MILESTONE_URL, data=payload, headers=headers)
+    print("Response:", payload)
 
     if response.status_code == 200:
         print("Milestone updated successfully!")
@@ -86,11 +91,59 @@ def update_milestone(title):
     elif response.status_code == 401:  # Unauthorized → Need to re-login
         print("Session expired, re-logging in...")
         if login():
-            return update_milestone(title)
+            return update_milestone(pr_data)
     else:
         print(f"Failed to update milestone: {response.status_code} - {response.text}")
         return False
 
+def create_milestone_content(pr_data):
+    content = get_milestone()
+    if content:
+        content['content'] = append_item_to_identifier(content['content'], 'feature_sigma', pr_data)
+        return f"{content['content']}\n- [x] {pr_data['title']}"
+    else:
+        return f"- [] {pr_data['title']}"
+
+def get_milestone():
+    import re
+    response = session.get(EDIT_MILESTONE_URL)
+    if response.status_code == 200:
+        title_match = re.search(r'<input[^>]*name=["\']title["\'][^>]*value=["\'](.*?)["\']', response.text, re.IGNORECASE)
+        content_match = re.search(r'<textarea[^>]*name=["\']content["\'][^>]*>(.*?)</textarea>', response.text, re.IGNORECASE | re.DOTALL)
+        title = title_match.group(1) if title_match else None
+        content = content_match.group(1).strip() if content_match else None
+        return {'title': title, 'content': content}
+    else:
+        
+        return None
+
+def append_item_to_identifier(encoded_content, identifier, new_item):
+    # Extract the block delimited by the identifier markers
+    import re
+    import html
+    content = html.unescape(encoded_content)
+    block_pattern = rf'(<!--\s*{identifier}\s*-->)(.*?)(<!--\s*/{identifier}\s*-->)'
+    block_match = re.search(block_pattern, content, re.DOTALL)
+    if not block_match:
+        print(f"Block with identifier '{identifier}' not found.")
+        return False
+
+    # Extract the parts of the block
+    start_marker = block_match.group(1)
+    block_content = block_match.group(2).strip()
+    end_marker = block_match.group(3)
+
+    # Append the new item to the block content
+    formatted_item = f"- [x] {new_item['username']}: [{new_item['title']}]({new_item['branch']})"
+    updated_block_content = f"{block_content}\n{formatted_item}"
+
+    # Reconstruct the content with the updated block
+    updated_content = f"{start_marker}\n{updated_block_content}\n{end_marker}"
+    content = content.replace(block_match.group(0), updated_content)
+
+    print(f"New item appended to '{identifier}' block.")
+    return content
+
 @app.route('/webhook', methods=['POST'])
 def handle_webhook():
     if request.method != 'POST':
@@ -104,12 +157,20 @@ def handle_webhook():
         # Handle Pull Request Event
         if 'pull_request' in payload:
             pr = payload.get('pull_request', {})
+
             pr_title = pr.get('title')
+            pr_username = pr.get('user').get('username')
+            pr_branch = pr.get('head_branch')
+            pr_data = {
+                'title': pr_title,
+                'username': pr_username,
+                'branch': pr_branch
+            }
 
             print(f"Received PR: {pr_title}")
 
             # Update milestone with PR title
-            if update_milestone(pr_title):
+            if update_milestone(pr_data):
                 print("Milestone updated with PR title.")
             else:
                 print("Failed to update milestone.")
@@ -123,6 +184,8 @@ def handle_webhook():
 if __name__ == '__main__':
     if login():  # Attempt login at startup
         print("Initial login successful.")
+        milestone = get_milestone()
+        print(milestone)
     else:
         print("Initial login failed.")
     app.run(port=4001)