From 6c1d7d587b7564e7e6cfd277ea94232e0a0df8c8 Mon Sep 17 00:00:00 2001
From: James Miller <james.ryland.miller@gmail.com>
Date: Fri, 24 Apr 2020 15:07:08 -0500
Subject: [PATCH] working on staves

---
 data/yaml/{items.yaml => itemcategories.yaml} |  0
 data/yaml/tmp-sql-staves-to-yaml.py           | 99 +++++++++++++++++++
 2 files changed, 99 insertions(+)
 rename data/yaml/{items.yaml => itemcategories.yaml} (100%)
 create mode 100644 data/yaml/tmp-sql-staves-to-yaml.py

diff --git a/data/yaml/items.yaml b/data/yaml/itemcategories.yaml
similarity index 100%
rename from data/yaml/items.yaml
rename to data/yaml/itemcategories.yaml
diff --git a/data/yaml/tmp-sql-staves-to-yaml.py b/data/yaml/tmp-sql-staves-to-yaml.py
new file mode 100644
index 0000000..e8f0f5b
--- /dev/null
+++ b/data/yaml/tmp-sql-staves-to-yaml.py
@@ -0,0 +1,99 @@
+import sqlite3
+import yaml
+import pprint
+
+def main():
+    pp = pprint.PrettyPrinter(indent=1, width=180)
+    conn = sqlite3.connect('../deprecated/pf2.db')
+    conn.row_factory = sqlite3.Row
+
+    q = """
+    SELECT * FROM staff;
+    """
+
+    c = conn.cursor()
+    c.execute(q)
+    data = [dict(row) for row in c.fetchall()]
+
+    for i in data:
+        # ADD THE ITEM CATEGORY
+        i['itemcategory'] = "Staves"
+
+        # convert gp prices to cp prices to avoid float issues
+        if i['price'] == '':
+            i['price'] = '0'
+        i['price_cp'] = int(float(i['price']) * 100)
+        del i['price']
+
+        # make bulk a string
+        i['bulk'] = str(i['bulk'])
+
+        # handle source entries
+        i['source'] = [
+            {
+                     'abbr': 'CRB',
+                     'page_start': int(i['source_pages']),
+                     'page_stop': int(i['source_pages'])
+            }
+        ]
+        del i['source_id']
+        del i['source_pages']
+
+        # Get staffactivations and effects
+        qq = """
+        SELECT activation, effect FROM staffactivations
+        INNER JOIN
+          staff_staffactivations ON
+          staffactivations.staffactivations_id = staff_staffactivations.staffactivations_id
+        INNER JOIN
+          staff ON staff.staff_id = staff_staffactivations.staff_id
+          WHERE staff.staff_id = ?;
+        """
+
+        # Get staffactivations
+        cc = conn.cursor()
+        cc.execute(qq, (i['staff_id'],))
+        res = cc.fetchall()
+        # print("\n\nStaff: {}".format(i['name']))
+        slist = []
+        for j in res:
+            x = tuple(j)[0]
+            y = tuple(j)[1]
+            xy = {'activation': x, 'effect': y}
+            slist.append(xy)
+        if len(slist) > 0:
+            i['staffactivations'] = slist
+        else:
+            i['staffactivations'] = None
+
+        # Get traits
+        qq = """
+        SELECT traits.short_name
+            FROM traits
+            INNER JOIN
+                staff_trait ON traits.trait_id = staff_trait.trait_id
+            INNER JOIN
+                staff ON staff.staff_id = staff_trait.staff_id
+                WHERE staff.staff_id = ?;
+        """
+        cc = conn.cursor()
+        cc.execute(qq, (i['staff_id'],))
+        res = cc.fetchall()
+        # print("\n\nWeapon: {}".format(i['name']))
+        tlist = []
+        for j in res:
+            x = tuple(j)[0]
+            tlist.append(x)
+        if len(tlist) > 0:
+            i['traits'] = tlist
+        else:
+            i['traits'] = None
+
+        # DO THIS LAST
+        del i['staff_id']
+
+    pp.pprint(data)
+
+
+if __name__ == '__main__':
+   main()