Fix Black formatting issues in Python scripts

Copilot · commjoen · Copilot · commit 5727024a2bb4 · 2025-07-31T10:13:04.000Z
Co-authored-by: commjoen &lt;1457214+commjoen@users.noreply.github.com&gt;
diff --git a/.github/scripts/remove_pr_from_index.py b/.github/scripts/remove_pr_from_index.py
@@ -4,43 +4,54 @@
 import os
 from datetime import datetime
 
+
 def main():
-    pr_number = os.environ.get('PR_NUMBER', 'unknown')
+    pr_number = os.environ.get("PR_NUMBER", "unknown")
 
     try:
-        with open('/tmp/cleaned-site/index.html', 'r') as f:
+        with open("/tmp/cleaned-site/index.html", "r") as f:
             content = f.read()
-        
+
         # Remove the PR card for this specific PR number
-        card_pattern = f'<div class="pr-card"[^>]*data-pr="{pr_number}"[^>]*>.*?</div>\s*</div>'
-        updated_content = re.sub(card_pattern, '', content, flags=re.DOTALL)
-        
+        card_pattern = (
+            f'<div class="pr-card"[^>]*data-pr="{pr_number}"[^>]*>.*?</div>\s*</div>'
+        )
+        updated_content = re.sub(card_pattern, "", content, flags=re.DOTALL)
+
         # Check if there are any remaining PR cards
-        remaining_cards = re.findall(r'<div class="pr-card"[^>]*data-pr="[^"]*"[^>]*>', updated_content)
-        
+        remaining_cards = re.findall(
+            r'<div class="pr-card"[^>]*data-pr="[^"]*"[^>]*>', updated_content
+        )
+
         if not remaining_cards:
             # No remaining PRs, add a "no previews" message
             no_previews_html = """                          <div class="no-previews" style="text-align: center; color: #666; font-style: italic; grid-column: 1 / -1;">
                               No active PR previews available
                           </div>"""
-            
+
             # Replace any existing content in the grid
             grid_pattern = r'(<div class="pr-grid">)(.*?)(</div>\s*</div>\s*<div style="text-align: center)'
-            updated_content = re.sub(grid_pattern, r'\1\n' + no_previews_html + r'\n            \3', updated_content, flags=re.DOTALL)
-        
+            updated_content = re.sub(
+                grid_pattern,
+                r"\1\n" + no_previews_html + r"\n            \3",
+                updated_content,
+                flags=re.DOTALL,
+            )
+
         # Update the last updated timestamp
-        timestamp_pattern = r'Last updated: [^<•]*'
+        timestamp_pattern = r"Last updated: [^<•]*"
         new_timestamp = f'Last updated: {datetime.now().strftime("%Y-%m-%d %H:%M UTC")}'
         updated_content = re.sub(timestamp_pattern, new_timestamp, updated_content)
-        
-        with open('/tmp/cleaned-site/index.html', 'w') as f:
+
+        with open("/tmp/cleaned-site/index.html", "w") as f:
             f.write(updated_content)
-        
+
         print(f"Successfully removed PR #{pr_number} from index")
-        
+
     except Exception as e:
         print(f"Error updating index: {e}")
         # Don't fail the cleanup if index update fails
 
+
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/.github/scripts/update_pr_index.py b/.github/scripts/update_pr_index.py
@@ -5,12 +5,13 @@
 from datetime import datetime
 import html
 
+
 def main():
     # Get environment variables
-    pr_number = os.environ.get('PR_NUMBER', 'unknown')
-    pr_title = os.environ.get('PR_TITLE', 'Unknown PR')
-    pr_sha = os.environ.get('PR_SHA', 'unknown')
-    existing_index_file = os.environ.get('EXISTING_INDEX', '')
+    pr_number = os.environ.get("PR_NUMBER", "unknown")
+    pr_title = os.environ.get("PR_TITLE", "Unknown PR")
+    pr_sha = os.environ.get("PR_SHA", "unknown")
+    existing_index_file = os.environ.get("EXISTING_INDEX", "")
 
     # Escape HTML characters in PR title
     pr_title_escaped = html.escape(pr_title.strip())
@@ -26,7 +27,8 @@ def main():
                           </div>"""
 
     # Base HTML template
-    base_html = """<!DOCTYPE html>
+    base_html = (
+        """<!DOCTYPE html>
 <html lang="en">
 <head>
     <meta charset="UTF-8">
@@ -85,55 +87,61 @@ def main():
                 <a href="https://github.com/OWASP/wrongsecrets" target="_blank">View Repository</a> |
                 <a href="https://github.com/OWASP/wrongsecrets/pulls" target="_blank">All Pull Requests</a>
             </p>
-            <small>Generated by GitHub Actions • Last updated: """ + datetime.now().strftime("%Y-%m-%d %H:%M UTC") + """</small>
+            <small>Generated by GitHub Actions • Last updated: """
+        + datetime.now().strftime("%Y-%m-%d %H:%M UTC")
+        + """</small>
         </div>
     </div>
 </body>
 </html>"""
+    )
 
     try:
         # Try to read existing index
         if existing_index_file and os.path.exists(existing_index_file):
-            with open(existing_index_file, 'r') as f:
+            with open(existing_index_file, "r") as f:
                 existing_content = f.read()
-            
+
             # Extract existing PR cards
             card_pattern = r'<div class="pr-card"[^>]*>.*?</div>\s*</div>'
             existing_cards = re.findall(card_pattern, existing_content, re.DOTALL)
-            
+
             # Remove the current PR card if it exists
             filtered_cards = []
             for card in existing_cards:
                 if f'data-pr="{pr_number}"' not in card:
                     filtered_cards.append(card)
-            
+
             # Add current PR card at the beginning
             all_cards = [pr_card_template] + filtered_cards
-            
+
         else:
             # No existing index, start fresh
             all_cards = [pr_card_template]
 
         # Generate final HTML
         if all_cards:
-            cards_html = '\n'.join(all_cards)
+            cards_html = "\n".join(all_cards)
         else:
             cards_html = '                          <div class="no-previews">No active PR previews</div>'
-        
-        final_html = base_html.replace('<!-- PR_CARDS_PLACEHOLDER -->', cards_html)
-        
-        with open('static-site/index.html', 'w') as f:
+
+        final_html = base_html.replace("<!-- PR_CARDS_PLACEHOLDER -->", cards_html)
+
+        with open("static-site/index.html", "w") as f:
             f.write(final_html)
-        
+
         print(f"Successfully updated index with PR #{pr_number}")
-        
+
     except Exception as e:
         print(f"Error updating index: {e}")
         # Fallback to simple index
-        fallback_html = base_html.replace('<!-- PR_CARDS_PLACEHOLDER -->', pr_card_template)
-        with open('static-site/index.html', 'w') as f:
+        fallback_html = base_html.replace(
+            "<!-- PR_CARDS_PLACEHOLDER -->", pr_card_template
+        )
+        with open("static-site/index.html", "w") as f:
             f.write(fallback_html)
         print("Created fallback index")
 
+
 if __name__ == "__main__":
-    main()
+    main()
