157 lines
4.2 KiB
Python
157 lines
4.2 KiB
Python
import os, sys, re, unicodedata
|
|
|
|
def slugify(value, allow_unicode=False):
|
|
"""
|
|
Taken from https://github.com/django/django/blob/master/django/utils/text.py
|
|
Convert to ASCII if 'allow_unicode' is False. Convert spaces or repeated
|
|
dashes to single dashes. Remove characters that aren't alphanumerics, whitespace,
|
|
underscores, or hyphens. Convert to lowercase. Also strip leading and
|
|
trailing whitespace.
|
|
"""
|
|
value = str(value)
|
|
if allow_unicode:
|
|
value = unicodedata.normalize('NFKC', value)
|
|
else:
|
|
value = unicodedata.normalize('NFKD', value).encode('ascii', 'ignore').decode('ascii')
|
|
value = re.sub(r'[^\w\s-]', '', value)
|
|
return value.strip()
|
|
|
|
def snippet(text, seq1, seq2):
|
|
try:
|
|
start = text.index(seq1)+len(seq1)
|
|
except ValueError:
|
|
start=0
|
|
end =text.index(seq2, start+1)
|
|
return text[start:end]
|
|
|
|
with open('kibbles_generic_raw.txt') as filein:
|
|
raw_txt = filein.read().split('\n\n')
|
|
|
|
if not os.path.exists('./spells/'):
|
|
os.mkdir('spells')
|
|
base_dir = './spells/KIBBLES'
|
|
if not os.path.exists(base_dir):
|
|
os.mkdir(base_dir)
|
|
|
|
|
|
level = 0
|
|
|
|
for para in raw_txt:
|
|
|
|
spell = {}
|
|
if not para:
|
|
continue
|
|
|
|
try :
|
|
|
|
lines = para.split('\n')
|
|
|
|
while True:
|
|
if not lines[0]:
|
|
lines.pop(0)
|
|
else:
|
|
break
|
|
|
|
while True:
|
|
if not lines[-1]:
|
|
lines.pop()
|
|
else:
|
|
break
|
|
|
|
spell['name'] = lines.pop(0)
|
|
info = lines.pop(0).lower()
|
|
|
|
spell['level'] = 0 if 'cantrip' in info.lower() else int(info[0])
|
|
spell['school'] = info.split()[0].title() if 'cantrip' in info.lower() else info.split()[1].title()
|
|
spell['ritual'] = 'ritual' in info
|
|
spell['classes'] = lines.pop(0)[9:]
|
|
spell['casting'] = lines.pop(0)[14:]
|
|
spell['range'] = lines.pop(0)[7:]
|
|
spell['components'] = lines.pop(0)[12:]
|
|
|
|
if '(' in spell['components'] and ')' not in spell['components']:
|
|
while ')' not in spell['components']:
|
|
spell['components'] += lines.pop(0)
|
|
|
|
spell['duration'] = lines.pop(0)[10:]
|
|
spell['upcasttag'] = 'false'
|
|
|
|
|
|
breaks = []
|
|
for idx, line in enumerate(lines):
|
|
if line and line[-1] in ['.', ':']:
|
|
breaks.append(idx)
|
|
|
|
if line.startswith('**At Higher Levels.**'):
|
|
spell['upcasttag'] = 'true'
|
|
|
|
else:
|
|
if idx not in breaks:
|
|
lines[-1] = lines[-1]+'.'
|
|
breaks.append(idx)
|
|
|
|
if len(breaks) >1:
|
|
for idx in breaks[:-1]:
|
|
lines[idx]+='\n'
|
|
|
|
spell['description'] = ' '.join(lines).replace('\n ','\n')
|
|
|
|
|
|
if 'reaction' in spell['casting'].lower():
|
|
spell['cast_time'] = 'Reaction'
|
|
elif 'bonus' in spell['casting'].lower():
|
|
spell['cast_time'] = 'Bonus acn.'
|
|
elif 'action' in spell['casting'].lower():
|
|
spell['cast_time'] = 'Action'
|
|
else:
|
|
spell['cast_time'] = spell['casting']
|
|
|
|
spell['concentration'] = 'true' if 'concentration' in spell['duration'].lower() else 'false'
|
|
spell['mats'] = spell['components'].split(' (')[0]
|
|
spell['priced'] = 'true' if 'worth' in spell['components'].lower() else 'false'
|
|
spell['consumed'] = 'true' if 'consume' in spell['components'].lower() else 'false'
|
|
|
|
tags = [spell['school']]
|
|
if spell['concentration']:
|
|
tags.append('Concentration')
|
|
if spell['ritual']:
|
|
tags.append('Ritual')
|
|
spell['tags'] = ', '.join(tags)
|
|
|
|
|
|
spell['book'] = "Kibbles' Generic Spells"
|
|
|
|
spell['subtitle'] = info
|
|
cardtext = """---
|
|
level: {level}
|
|
components: [{mats}]
|
|
material: {priced}
|
|
consumed: {consumed}
|
|
cast_time: {cast_time}
|
|
upcast: {upcasttag}
|
|
classes: [{classes}]
|
|
tags: [{tags}]
|
|
---
|
|
#### {name}
|
|
*{subtitle}*
|
|
___
|
|
- **Casting Time:** {casting}
|
|
- **Range:** {range}
|
|
- **Components:** {components}
|
|
- **Duration:** {duration}
|
|
---
|
|
{description}
|
|
|
|
*{book}*
|
|
""".format(**spell)
|
|
|
|
with open(os.path.join(base_dir, slugify(spell['name'])+'.md'),'w') as fileout:
|
|
fileout.write(cardtext)
|
|
|
|
except Exception as e:
|
|
print(para)
|
|
raise e
|
|
|
|
|
|
|
|
|