从blog迁移到hugo脚本

hugo-blog.py

# coding: utf-8

import MySQLdb


connect = MySQLdb.connect(user='root', passwd='', db='blog', host='127.0.0.1', port=3306)
cursor = connect.cursor()
cursor.execute('SELECT title, content, tags, created FROM blogs')

result = cursor.fetchall()

with open('template.md.bak', 'r') as f:
    template = f.read()

for log in result:
    data = {
        'title': log[0],
        'tags': '", "'.join(log[2].split(',')),
        'date': log[3].strftime('%Y-%m-%dT%H:%M:%S+08:00')
    }
    content = log[1].split('\n')
    count = index = 0
    for i, line in enumerate(content):
        count += line.count('```')
        if i >= 9 and (count % 2 == 0):
            index = i+1
            break
    content.insert(index, '<!--more-->')
    data['content'] = '\n'.join(content)
    with open('{}-{}.md'.format(log[3].strftime('%Y-%m-%d'), data['title']), 'w') as f:
        f.write(template.format(**data))