makedemo.py

# coding=utf-8
from __future__ import unicode_literals

import os
import glob
import shutil
import pinyin
from collections import defaultdict
from jinja2 import Environment, FileSystemLoader
import codecs


def list_base(src_folder):
    for folder in glob.glob(src_folder):
        if os.path.isdir(folder):
            yield folder


def list_a_directory(src_folder):
    for f in glob.glob(src_folder):
        file_name = os.path.basename(f).split('.')[0]
        if '市' in file_name:
            file_name = file_name[:-1]
        pinyin_name = pinyin.get(file_name, format="numerical", delimiter="_")
        yield f, file_name, pinyin_name


if __name__ == '__main__':
    name_dict = {}
    rendering_dict = defaultdict(list)
    for folder in list_base('src/*'):
        cfolder = os.path.basename(folder)
        pfolder = pinyin.get(cfolder, format="numerical", delimiter="_")
        _dest_folder = os.path.join('dist', pfolder)
        name_dict[cfolder] = pfolder
        if not os.path.exists(_dest_folder):
            os.mkdir(_dest_folder)
        all_files = list_a_directory(os.path.join(folder, "*.js"))
        for src_file, cname, pname in all_files:
            _dest_file = os.path.join(_dest_folder, "%s.js" % pname)
            print("%s-> %s, %s -> %s" % (cname, pname, src_file, _dest_file))
            shutil.copy(src_file, _dest_file)
            rendering_dict[cfolder].append((cname, pname))
    jinja2_env = Environment(
        loader=FileSystemLoader('./templates'),
        keep_trailing_newline=True,
        trim_blocks=True,
        lstrip_blocks=True)
    template = jinja2_env.get_template('index.html')
    html = template.render(names=name_dict, registry=rendering_dict)
    with codecs.open('index.html', 'wb', 'utf-8') as f:
        f.write(html)

    config = jinja2_env.get_template('config.json')
    config_json = config.render(names=name_dict, registry=rendering_dict)
    with codecs.open(os.path.join('dist', 'config.json'), 'wb', 'utf-8') as f:
        f.write(config_json)

    count = 0
    for cprovince in rendering_dict.keys():
        count += len(rendering_dict[cprovince])
    provinces, cities = len(rendering_dict.keys()), count

    readme = jinja2_env.get_template('README.md')
    readme_txt = readme.render(
        names=name_dict, registry=rendering_dict,
        num_provinces=provinces, num_cities=cities
    )
    with codecs.open(os.path.join('README.md'), 'wb', 'utf-8') as f:
        f.write(readme_txt)