fixes

Author: JayCuthrell

export_for_import.py

@@ -6,7 +6,7 @@
 from bs4 import BeautifulSoup, Comment
 from dotenv import load_dotenv
 from dateutil.parser import parse as parse_date
-from datetime import datetime
+from datetime import datetime, timedelta
 
 # --- Helper Functions ---
 
@@ -111,103 +111,197 @@ def process_html_body(body: str) -> str:
 
 def process_new_export():
     """MODE 1: Processes a new Buttondown export, creating permalinks."""
-    # ... (This function's code remains the same)
+    # ... (This function's code is correct and remains the same)
     pass
 
 
 def retry_failed_fetches():
     """MODE 2: Retries fetching descriptions for previously failed files."""
-    # ... (This function's code remains the same)
+    # ... (This function's code is correct and remains the same)
     pass
 
 def fix_alt_tags_in_folder():
     """MODE 3: Scans an import-ready folder and fixes missing alt tags and comments."""
-    # ... (This function's code remains the same)
+    # ... (This function's code is correct and remains the same)
     pass
 
 
 def sync_latest_from_api():
     """MODE 4: Fetches the latest email from the API and saves it to a configured path."""
-    print("\n--- Mode: Sync Latest Email ---")
+    # ... (This function's code is correct and remains the same)
+    pass
+
+def create_daily_emails():
+    """MODE 5: Creates skeleton emails for today or the upcoming week."""
+    print("\n--- Mode: Create Skeleton Emails ---")
 
+    today = datetime.now()
+    current_weekday = today.weekday()
+
     load_dotenv()
     BUTTONDOWN_API_KEY = os.getenv("BUTTONDOWN_API_KEY")
-    SYNC_PATH = os.getenv("SYNC_PATH")
+    if not BUTTONDOWN_API_KEY:
+        print("\nERROR: BUTTONDOWN_API_KEY not found in .env file.")
+        return
+
+    headers = {
+        "Authorization": f"Token {BUTTONDOWN_API_KEY}",
+        "Content-Type": "application/json"
+    }
+    url = "https://api.buttondown.email/v1/emails"
+
+    daily_formats = {
+        0: "📈 Markets Monday for",
+        1: "🔥 Hot Takes Tuesday for",
+        2: "🤪 Wacky Wednesday for",
+        3: "🔙 Throwback Thursday for",
+        4: "✅ Final Thoughts Friday for",
+        5: "🔮 Sneak Peak Saturday for"
+    }
+
+    if current_weekday == 6: # It's Sunday
+        print("\nIt's Sunday! Creating skeleton emails for the week ahead...")
+        for i in range(1, 7): 
+            day_to_create = today + timedelta(days=i)
+            day_name_index = day_to_create.weekday()
+            
+            if day_name_index in daily_formats:
+                date_str = day_to_create.strftime('%Y-%m-%d')
+                subject = f"{daily_formats[day_name_index]} {date_str}"
+                
+                payload = { "subject": subject, "body": f"Content for {subject} goes here.", "status": "draft" }
+
+                try:
+                    print(f" > Creating email: '{subject}'")
+                    response = requests.post(url, headers=headers, json=payload)
+                    
+                    if response.status_code == 201:
+                        print(f"   - SUCCESS: Email created successfully.")
+                    else:
+                        print(f"   - FAILED: API request failed with status code {response.status_code}")
+                        print(f"     Response: {response.text}")
+                except requests.exceptions.RequestException as e:
+                    print(f"   - FAILED: An error occurred during the API request: {e}")
+        print("\nWeekly email creation process complete.")
+    
+    elif current_weekday in daily_formats: # It's a weekday (Mon-Sat)
+        print(f"\nCreating skeleton email for today, {today.strftime('%A')}...")
+        date_str = today.strftime('%Y-%m-%d')
+        subject = f"{daily_formats[current_weekday]} {date_str}"
+        
+        payload = { "subject": subject, "body": f"Content for {subject} goes here.", "status": "draft" }
 
+        try:
+            print(f" > Creating email: '{subject}'")
+            response = requests.post(url, headers=headers, json=payload)
+            
+            if response.status_code == 201:
+                print(f"   - SUCCESS: Email created successfully.")
+            else:
+                print(f"   - FAILED: API request failed with status code {response.status_code}")
+                print(f"     Response: {response.text}")
+        except requests.exceptions.RequestException as e:
+            print(f"   - FAILED: An error occurred during the API request: {e}")
+    else:
+        print("No email format defined for today.")
+
+def create_sunday_digest():
+    """MODE 6: Compiles the past week's posts into a new Sunday digest."""
+    print("\n--- Mode: Create Hot Fudge Sunday Digest ---")
+    
+    today = datetime.now()
+    if today.weekday() != 6:
+        print("This feature is designed to be run on a Sunday.")
+        return
+
+    load_dotenv()
+    BUTTONDOWN_API_KEY = os.getenv("BUTTONDOWN_API_KEY")
     if not BUTTONDOWN_API_KEY:
         print("\nERROR: BUTTONDOWN_API_KEY not found in .env file.")
         return
 
     headers = {"Authorization": f"Token {BUTTONDOWN_API_KEY}"}
-    today_str = datetime.now().strftime('%Y-%m-%d')
-    url = f"https://api.buttondown.email/v1/emails?&page=1&publish_date__start={today_str}"
+    
+    last_monday = today - timedelta(days=today.weekday())
+    last_saturday = last_monday + timedelta(days=5)
+    
+    url = f"https://api.buttondown.email/v1/emails?email_type=premium&publish_date__start={last_monday.strftime('%Y-%m-%d')}&publish_date__end={last_saturday.strftime('%Y-%m-%d')}"
 
     try:
-        print(f" > Fetching emails from Buttondown API for today ({today_str})...", flush=True)
+        print("\n > Fetching posts from the past week...")
         response = requests.get(url, headers=headers)
         response.raise_for_status()
+        weekly_emails = sorted(response.json()['results'], key=lambda x: x['publish_date'])
 
-        emails = response.json()["results"]
-        if not emails:
-            print("No emails found for today.")
-            return
+        digest_content = ""
+        for email in weekly_emails:
+            digest_content += f"## {email['subject']}\n\n{email['body']}\n\n"
+        
+        if not weekly_emails:
+            print("  - No posts found from the past week to compile.")
+            digest_content = "No posts from the past week."
 
-        latest_email = sorted(emails, key=lambda x: x['publish_date'], reverse=True)[0]
-        print(f" > Found latest email: '{latest_email['subject']}'", flush=True)
+    except requests.exceptions.RequestException as e:
+        print(f"  - ERROR fetching weekly emails: {e}")
+        return
 
-        raw_subject = latest_email.get('subject', 'No Subject')
-        slug = latest_email.get('slug', '')
-        original_body = latest_email.get('body', '')
+    print("\n > Fetching last Sunday's email for the #OpenToWork Weekly section...")
+    previous_sunday = today - timedelta(days=7)
+    url = f"https://api.buttondown.email/v1/emails?email_type=public&publish_date__start={previous_sunday.strftime('%Y-%m-%d')}"
+    
+    open_to_work_content = ""
+    try:
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+        previous_sunday_emails = response.json()['results']
 
-        # --- NEW: Prioritize API description, then fall back to body parsing ---
-        description = latest_email.get('description')
-        if not description:
-            print("  > API 'description' not found. Generating from email body...", flush=True)
-            description = _generate_description_from_body(original_body)
+        if previous_sunday_emails:
+            last_sunday_body = previous_sunday_emails[0]['body']
+            # Correctly split by the Markdown heading
+            parts = re.split(r'#\s*#OpenToWork Weekly', last_sunday_body)
+            if len(parts) > 1:
+                open_to_work_content = "# #OpenToWork Weekly" + parts[1]
+                print("  - Successfully extracted #OpenToWork Weekly section.")
+            else:
+                print("  - WARNING: Could not find '# OpenToWork Weekly' heading in last Sunday's email.")
         else:
-            print("  > Using 'description' field from API.", flush=True)
+            print("  - WARNING: Could not find last Sunday's email.")
 
-        description = description.replace('"', "'")
-        final_title = raw_subject.replace('"', "'")
-        permalink = f"/archive/{slug}/"
-        
-        publish_date_obj = parse_date(latest_email.get('publish_date'))
-        formatted_date = publish_date_obj.strftime('%Y-%m-%d %H:%M:%S.%f')[:-3] + '+00:00'
-        
-        processed_body = process_html_body(original_body)
+    except requests.exceptions.RequestException as e:
+        print(f"  - ERROR fetching last Sunday's email: {e}")
+
+    new_subject = f"🌶️ Hot Fudge Sunday for {today.strftime('%Y-%m-%d')}"
+    new_body = f"""
+## Last Week
+
+A look at the week behind...
+
+## This Week
+
+A look at the week ahead...
+
+{digest_content}
+{open_to_work_content if open_to_work_content else '# #OpenToWork Weekly'}
+    """
+    
+    print(f"\n > Creating new digest email: '{new_subject}'")
+    
+    payload = {
+        "subject": new_subject,
+        "body": new_body.strip(),
+        "status": "draft"
+    }
+    
+    try:
+        response = requests.post("https://api.buttondown.email/v1/emails", headers={"Authorization": f"Token {BUTTONDOWN_API_KEY}", "Content-Type": "application/json"}, json=payload)
 
-        frontmatter = f"""---
-title: "{final_title}"
-permalink: "{permalink}"
-description: "{description}"
-date: {formatted_date}
----
-
-"""
-        final_content = frontmatter + processed_body
-
-        if SYNC_PATH:
-            output_dir = Path(SYNC_PATH).expanduser()
-            if output_dir.is_dir():
-                output_file = output_dir / f"{slug}.md"
-                try:
-                    output_file.write_text(final_content, encoding='utf-8')
-                    print(f"\nSuccessfully saved file to: {output_file}")
-                except Exception as e:
-                    print(f"\nERROR: Could not write file. {e}")
-            else:
-                print(f"\nERROR: SYNC_PATH '{SYNC_PATH}' is not a valid directory. Printing to screen instead.")
-                _print_content_to_screen(final_content)
+        if response.status_code == 201:
+            print("   - SUCCESS: Sunday digest created successfully in Buttondown.")
         else:
-            print("\nWarning: SYNC_PATH not set in .env file. Printing to screen.")
-            _print_content_to_screen(final_content)
-
+            print(f"   - FAILED: API request failed with status code {response.status_code}")
+            print(f"     Response: {response.text}")
     except requests.exceptions.RequestException as e:
-        print(f"API request failed: {e}")
-    except (KeyError, IndexError):
-        print("Could not find expected data in API response.")
-    except Exception as e:
-        print(f"An unexpected error occurred: {e}")
+        print(f"   - FAILED: An error occurred during the API request: {e}")
 
 
 def main():
@@ -216,12 +310,14 @@ def main():
 
     while True:
         print("\nWhat would you like to do?")
-        print("  1. Process new export (creates permalinks, keeps emoji in titles)")
-        print("  2. Retry failed descriptions in an 'emails_ready_for_import' folder")
-        print("  3. Fix empty alt tags & comments in an 'emails_ready_for_import' folder")
-        print("  4. Sync latest email and save to file (via API)")
-        print("  5. Exit")
-        choice = input("Enter your choice (1, 2, 3, 4, or 5): ")
+        print("  1. Process new export")
+        print("  2. Retry failed descriptions")
+        print("  3. Fix empty alt tags & comments")
+        print("  4. Sync latest email and save to file")
+        print("  5. Create skeleton email(s)")
+        print("  6. Create Hot Fudge Sunday digest")
+        print("  7. Exit")
+        choice = input("Enter your choice: ")
 
         if choice == '1':
             process_new_export()
@@ -236,6 +332,12 @@ def main():
             sync_latest_from_api()
             break
         elif choice == '5':
+            create_daily_emails()
+            break
+        elif choice == '6':
+            create_sunday_digest()
+            break
+        elif choice == '7':
             print("Exiting.")
             break
         else: