Improve generate_theme.py script

This commit is contained in:
Lonami Exo 2017-03-03 16:38:57 +01:00
parent 2e864b0017
commit 2833055390
2 changed files with 50 additions and 17 deletions

View file

@ -34,6 +34,10 @@ template = '''{{
}} }}
''' '''
out_dir = '../android/assets/themes/'
theme_list = os.path.join(out_dir, 'theme.list')
def price_ok(price): def price_ok(price):
try: try:
price = int(price) price = int(price)
@ -47,23 +51,50 @@ def price_ok(price):
def main(): def main():
if not os.path.isfile('in.svg'): # Look for all the files which are not called 'template.svg'
print('Error: No "in.svg" found. Aborting.') files = [f for f in os.listdir()
if f.endswith('.svg') and f != 'template.svg']
if not files:
print('No .svg files were found. '
'Please see CREATING-THEMES.txt for more information')
return return
print('Reading "in.svg"') # Work on all the files to generate the corresponding themes
with open('in.svg', 'r', encoding='utf-8') as f: for filename in files:
work(filename)
print('Updating theme.list…')
themes = []
with open(theme_list, 'r', encoding='utf-8') as f:
themes = [line.strip() for line in f]
added_names = [os.path.splitext(f)[0] for f in files]
added_count = 0
for name in added_names:
if name not in themes:
themes.append(name)
added_count += 1
with open(theme_list, 'w', encoding='utf-8') as f:
f.write('\n'.join(themes))
f.write('\n')
print('Added {} new theme(s), updated {}.'.format(
added_count, len(files) - added_count))
def work(filename):
name = os.path.splitext(filename)[0]
with open(filename, 'r', encoding='utf-8') as f:
xml = f.read().replace('\n', '') xml = f.read().replace('\n', '')
print('Finding used colors…')
replacements = {} replacements = {}
for m in color_re.finditer(xml): for m in color_re.finditer(xml):
# Append 'ff' because the themes require the alpha to be set # Append 'ff' because the themes require the alpha to be set
replacements[m.group(1)] = m.group(2)+'ff' replacements[m.group(1)] = m.group(2)+'ff'
print('Almost done, we only need some more information.') replacements['name'] = input('Enter theme name for "{}": '.format(name))
replacements['name'] = input('Enter theme name: ')
replacements['price'] = input('Enter theme price: ') replacements['price'] = input('Enter theme price: ')
replacements['cell_tex'] = \ replacements['cell_tex'] = \
input('Enter cell texture (default "basic.png"): ') input('Enter cell texture (default "basic.png"): ')
@ -76,8 +107,10 @@ def main():
print('No texture specified. Using default "basic.png" texture.') print('No texture specified. Using default "basic.png" texture.')
replacements['cell_tex'] = 'basic.png' replacements['cell_tex'] = 'basic.png'
print('Generating theme…') output = os.path.join(out_dir, name+'.theme')
with open('out.theme', 'w', encoding='utf-8') as f:
print('Saving theme to {}'.format(output))
with open(output, 'w', encoding='utf-8') as f:
f.write(template.format_map(replacements)) f.write(template.format_map(replacements))
print('Done!') print('Done!')

View file

Before

Width:  |  Height:  |  Size: 14 KiB

After

Width:  |  Height:  |  Size: 14 KiB