From ec80f0458631d7fa1c0055115650313e53f29236 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?tr=C3=A9meur?= <eheu48@gmail.com>
Date: Wed, 22 Mar 2023 15:32:01 +0000
Subject: [PATCH] Add (dodgy) boolean for title inclusion in fic headers

---
 fic-header.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/fic-header.py b/fic-header.py
index 7e32874..49dc6e5 100644
--- a/fic-header.py
+++ b/fic-header.py
@@ -11,12 +11,14 @@ data = {
 if os.path.exists("ficheader.html"):
     os.remove("ficheader.html")
 
-thefile = open("ficheader.html", "a")
+thefile = open("ficheader.html", "a")    
 
 with requests.Session() as s:
     response = s.post(login_url , data)
-    print("Work ID: ")
+    print("Work ID:")
     id = input()
+    print("Use title? (y/n)")
+    thetitle = input()
     url = "https://archiveofourown.org/works/" + str(id)
     ficpage = s.get(url)
     ficsoup = BeautifulSoup(ficpage.content, "html.parser")
@@ -138,7 +140,10 @@ with requests.Session() as s:
     mastersoup = BeautifulSoup(masterlist.content, "html.parser")
     ficno = int(mastersoup.find("span", class_="ficno").text.strip()) + 1
     thefile.write("<div class=\"fic\" id=\"fic" + str(ficno) + "\">\n")
-    thefile.write("  <h1><span class=\"ficno\">" + str(ficno) + "</span> <span class=\"fictitle\">" + title + "</span></h1>\n")
+    if thetitle == "y":
+        thefile.write("  <h1><span class=\"ficno\">" + str(ficno) + "</span> <span class=\"fictitle\">" + title + "</span></h1>\n")
+    else:
+        thefile.write("  <h1><span class=\"ficno\">" + str(ficno) + "</span></h1>\n")
     thefile.write("  <ul class=\"ficmeta\">\n")
     thefile.write("    <li class=\"ficdate\">" + prazedate + "</li>\n")
     thefile.write("    <li class=\"wordcount\">" + str(words) + "</li>\n")