From 6beb0bf2d369edb2e7efb381e70e77d4d3a4f2c8 Mon Sep 17 00:00:00 2001
From: Jim Miller <retiefjimm@gmail.com>
Date: Mon, 11 Sep 2017 12:28:31 -0500
Subject: [PATCH] Fixes for QQ and AH changes.

---
 .../adapter_forumquestionablequestingcom.py   | 36 +++++++++++++++----
 1 file changed, 30 insertions(+), 6 deletions(-)

diff --git a/fanficfare/adapters/adapter_forumquestionablequestingcom.py b/fanficfare/adapters/adapter_forumquestionablequestingcom.py
index a756b086..f4dbdd87 100644
--- a/fanficfare/adapters/adapter_forumquestionablequestingcom.py
+++ b/fanficfare/adapters/adapter_forumquestionablequestingcom.py
@@ -15,6 +15,7 @@
 # limitations under the License.
 #
 
+import re
 from ..htmlcleanup import stripHTML
 
 from base_xenforoforum_adapter import BaseXenForoForumAdapter
@@ -39,15 +40,38 @@ class QuestionablequestingComAdapter(BaseXenForoForumAdapter):
     ## and QQ enough to require some differentiation.
     def extract_threadmarks(self,souptag):
         # try threadmarks if no '#' in url
-        navdiv = souptag.find('div',{'class':'pageNavLinkGroup'})
-        threadmarksa = navdiv.find('a',{'class':'threadmarksTrigger'})
+        navdiv = souptag.find('div',{'class':'threadmarkMenus'})
+        # was class=threadmarksTrigger.  thread cats are currently
+        # only OverlayTrigger <a>s in threadmarkMenus, but I wouldn't
+        # be surprised if that changed.  Don't want to do use just
+        # href=re because there's more than one copy on the page; plus
+        # could be included in a post.  Would be easier if <noscript>s
+        # weren't being stripped, but that's a different issue.
+        threadmarksas = navdiv.find_all('a',{'class':'OverlayTrigger','href':re.compile('threadmarks.*category_id=')})
         ## Loop on threadmark categories.
         threadmarks=[]
-        if threadmarksa:
+        tmcat_num=None
+
+        for threadmarksa in threadmarksas:
+            tmcat_num = threadmarksa['href'].split('category_id=')[1]
+            # get from earlier <a> now.
+            tmcat_name = stripHTML(threadmarksa.find_previous('a',{'class':'threadmarksTrigger'}))
+            prepend = ""
+            if tmcat_name in self.getConfigList('skip_threadmarks_categories'):
+                continue
+
+            if tmcat_name == 'Apocrypha' and self.getConfig('apocrypha_to_omake'):
+                tmcat_name = 'Omake'
+
+            if tmcat_name != "Threadmarks":
+                prepend = tmcat_name+" - "
+
             soupmarks = self.make_soup(self._fetchUrl(self.getURLPrefix()+'/'+threadmarksa['href']))
             markas = []
-            markas = soupmarks.find('div',{'class':'threadmarks'}).find_all('a',{'class':'PreviewTooltip'})
-            for (atag,url,name) in [ (x,x['href'],stripHTML(x)) for x in markas ]:
+            markas = soupmarks.find('div',{'class':'threadmarkList'}).find_all('a',{'class':'PreviewTooltip'})
+            for tmcat_index, atag in enumerate(markas):
+                url,name = atag['href'],stripHTML(atag)
                 date = self.make_date(atag.find_next_sibling('div',{'class':'extra'}))
-                threadmarks.append({'title':name,'url':self.getURLPrefix()+'/'+url,'date':date})
+                threadmarks.append({"tmcat_name":tmcat_name,"tmcat_num":tmcat_num,"tmcat_index":tmcat_index,'title':name,'url':self.getURLPrefix()+'/'+url,'date':date})
+
         return threadmarks