From 473b35e66792db380bdcbbcbb0e61dbefa09ae91 Mon Sep 17 00:00:00 2001
From: Jonathan Gordon <jdgordy@gmail.com>
Date: Mon, 22 Jun 2015 21:45:46 +1000
Subject: [PATCH] Add card groups to easily exclude upgrades and prizes, etc

---
 card_db/en_us/card_groups.txt |  6 ++++++
 dominion_tabs.py              | 30 +++++++++++++++++++++++++++++-
 2 files changed, 35 insertions(+), 1 deletion(-)
 create mode 100644 card_db/en_us/card_groups.txt

diff --git a/card_db/en_us/card_groups.txt b/card_db/en_us/card_groups.txt
new file mode 100644
index 0000000..add9a68
--- /dev/null
+++ b/card_db/en_us/card_groups.txt
@@ -0,0 +1,6 @@
+Tournament, Bag of Gold, Diadem, Followers, Princess, Trusty Steed, Tournament and Prizes
+Shelters, Hovel, Overgrown Estate, Necropolis, Shelters
+Page, Treasure Hunter, Warrior, Hero, Champion, Page -> Champion
+Peasant, Soldier, Fugitive, Disciple, Teacher, Peasant -> Teacher
+Hermit, Madman, Hermit / Madman
+Urchin, Mercenary, Urchin / Mercenary
diff --git a/dominion_tabs.py b/dominion_tabs.py
index 4dc388e..d15b617 100755
--- a/dominion_tabs.py
+++ b/dominion_tabs.py
@@ -1,4 +1,5 @@
 #!python
+import csv
 import re
 from optparse import OptionParser
 import os.path
@@ -569,6 +570,15 @@ class DominionTabs:
         if not self.options.tabs_only:
             self.drawText(card, useExtra)
 
+    def read_card_groups(self, fname):
+        groups = {}
+        with open(fname, 'r') as f:
+            for row in csv.reader(f, delimiter=','):
+                groups[row[0].strip()] = {
+                         "subcards":[x.strip() for x in row[1:-1]],
+                         "text":row[-1].strip()}
+        return groups
+                
     def read_card_extras(self, fname, cards):
         f = open(fname)
         cardName = re.compile("^:::(?P<name>[ \w\-/']*)", re.UNICODE)
@@ -852,8 +862,10 @@ class DominionTabs:
                           help="include a few dividers with extra text")
         parser.add_option("--exclude_events", action="store_true",
                           default=False, help="exclude individual dividers for events")
+        parser.add_option("--exclude_card_groups", action="store_true",
+                          default=False, help="exclude individual dividers for events")
         parser.add_option("--exclude_prizes", action="store_true",
-                          default=False, help="exclude individual dividers for prizes (cornicopia)")
+                          default=False, help="exclude individual dividers for prizes (cornucopia)")
         parser.add_option("--cardlist", type="string", dest="cardlist", default=None,
                           help="Path to file that enumerates each card to be printed on its own line.")
         parser.add_option("--no-tab-artwork", action="store_true", dest="no_tab_artwork",
@@ -1070,6 +1082,22 @@ class DominionTabs:
         else:
             cards = [card for card in cards if not isBaseExpansionCard(card)]
 
+        if self.options.exclude_card_groups:
+            # Load the card groupos file
+            card_groups = self.read_card_groups(os.path.join(self.filedir, "card_db", options.language, "card_groups.txt"))
+            # pull out any cards which are a subcard, and rename the master card
+            new_cards = []
+            all_subcards = []
+            for subs in [card_groups[x]["subcards"] for x in card_groups]:
+                all_subcards += subs
+            for card in cards:
+                if card.name in card_groups.keys():
+                    card.name = card_groups[card.name]["text"]
+                elif card.name in all_subcards:
+                    continue
+                new_cards.append(card)
+            cards = new_cards
+
         if self.options.expansions:
             self.options.expansions = [o.lower()
                                        for o in self.options.expansions]