95 lines
2.1 KiB
Python
95 lines
2.1 KiB
Python
import sqlite3
|
|
import yaml
|
|
import pprint
|
|
|
|
def main():
|
|
pp = pprint.PrettyPrinter(indent=4)
|
|
conn = sqlite3.connect('../../pf2.db')
|
|
conn.row_factory = sqlite3.Row
|
|
|
|
# DO WEAPON GROUPS
|
|
|
|
q = """
|
|
SELECT
|
|
sources_pages,
|
|
name,
|
|
descr
|
|
FROM weapongroups
|
|
"""
|
|
|
|
c = conn.cursor()
|
|
c.execute(q)
|
|
data = [dict(row) for row in c.fetchall()]
|
|
pp.pprint(data)
|
|
|
|
# "source": [
|
|
# {
|
|
# 'abbr': 'CRB',
|
|
# 'page_start': int(i['sources_pages']),
|
|
# 'page_stop': int(i['sources_pages'])
|
|
# },
|
|
|
|
wgdata = []
|
|
for i in data:
|
|
res = {
|
|
'name': i['name'],
|
|
'descr': i['descr'],
|
|
'source': [
|
|
{
|
|
'abbr': 'CRB',
|
|
'page_start': int(i['sources_pages']),
|
|
'page_stop': int(i['sources_pages'])
|
|
}
|
|
]
|
|
}
|
|
wgdata.append(res)
|
|
|
|
pp.pprint(wgdata)
|
|
finalwgdata = {'weapongroups': wgdata}
|
|
pp.pprint(finalwgdata)
|
|
|
|
final = yaml.safe_dump(finalwgdata, allow_unicode=True)
|
|
with open('tmp-weapongroups.yaml', 'w') as f:
|
|
f.write(final)
|
|
|
|
# DO WEAPONS
|
|
|
|
# q = """
|
|
# SELECT
|
|
# price_gp,
|
|
# dice_size,
|
|
# bulk,
|
|
# hands,
|
|
# range,
|
|
# reload,
|
|
# name,
|
|
# descr,
|
|
# (SELECT name FROM actions) as action_name -- use this as template for subqueries
|
|
# FROM weapons;
|
|
# """
|
|
|
|
# c = conn.cursor()
|
|
# c.execute(q)
|
|
# data = [dict(row) for row in c.fetchall()]
|
|
# for i in data:
|
|
# # handle empty bulk entries to match the abbr field in bulks.yaml
|
|
# if i['bulk'] == '':
|
|
# i['bulk'] = '-'
|
|
# # convert gp prices to cp prices to avoid float issues
|
|
# if i['price_gp'] == '':
|
|
# i['price_gp'] = '0'
|
|
# i['price_cp'] = int(float(i['price_gp']) * 100)
|
|
# del i['price_gp']
|
|
|
|
# pp.pprint(data)
|
|
|
|
# fdata = {'weapons': data}
|
|
# final = yaml.safe_dump(fdata, allow_unicode=True)
|
|
# with open('tmp-weapons.yaml', 'w') as f:
|
|
# f.write(final)
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|