python操作数据库批量添加wordpress图片文章，并生成缩略图

# -*- coding:utf-8 -*-

import os
import MySQLdb
import shutil

# 连接数据库
conn = MySQLdb.connect(host='localhost', user='root', passwd='', db='wordpress', port=3306, charset='utf8')
cur = conn.cursor()

# 准备图片文件
img_dir = '/home/images/'
if not os.path.exists(img_dir):
    os.makedirs(img_dir)

# 批量添加文章
for i in range(10):
    # 添加文章
    cur.execute("INSERT INTO wp_posts (post_title, post_content, post_status, comment_status, post_type) VALUES (%s, %s, %s, %s, %s)",
        ('标题' + str(i), '文章内容' + str(i), 'publish', 'open', 'post'))
    post_id = cur.lastrowid
    
    # 添加图片
    cur.execute("INSERT INTO wp_posts (post_title, post_content, post_status, comment_status, post_type, post_parent) VALUES (%s, %s, %s, %s, %s, %s)",
        ('标题图片' + str(i), '图片描述' + str(i), 'inherit', 'open', 'attachment', post_id))
    img_id = cur.lastrowid
    
    # 生成缩略图
    img_name = 'img' + str(i) + '.jpg'
    shutil.copyfile(os.path.join(img_dir, img_name), os.path.join('/var/www/html/wp-content/uploads/', img_name))
    cur.execute("UPDATE wp_posts SET guid = %s WHERE ID = %s",
        ('http://www.example.com/wp-content/uploads/' + img_name, img_id))
    cur.execute("INSERT INTO wp_postmeta (post_id, meta_key, meta_value) VALUES (%s, %s, %s)",
        (img_id, '_wp_attached_file', 'uploads/' + img_name))
    
    # 提交数据
    conn.commit()

# 关闭数据库连接
cur.close()
conn.close()