134 lines
4.5 KiB
Python
134 lines
4.5 KiB
Python
|
import os
|
||
|
import frontmatter
|
||
|
from wordpress_xmlrpc import Client, WordPressPost
|
||
|
from wordpress_xmlrpc.methods.posts import GetPosts, NewPost
|
||
|
from datetime import datetime
|
||
|
|
||
|
# 配置部分
|
||
|
hexo_posts_dir = '/home/songtianlun/Sync/Develop/frytea/source/_posts' # Hexo 文章目录
|
||
|
wordpress_url = "https://blog.lnf1.skybyte.me/xmlrpc.php"
|
||
|
wordpress_username = 'songtianlun'
|
||
|
wordpress_password = 'sotilu,WP2024'
|
||
|
|
||
|
# 初始化 WordPress 客户端
|
||
|
wp = Client(wordpress_url, wordpress_username, wordpress_password)
|
||
|
|
||
|
# 上传统计信息
|
||
|
total_posts = 0
|
||
|
successful_posts = 0
|
||
|
failed_posts = 0
|
||
|
duplicate_posts = []
|
||
|
|
||
|
|
||
|
def get_all_wp_posts():
|
||
|
print("Fetching existing posts from WordPress...")
|
||
|
all_posts = []
|
||
|
offset = 0
|
||
|
increment = 20 # 每次获取的文章数量,可以调整
|
||
|
while True:
|
||
|
batch = wp.call(GetPosts({'number': increment, 'offset': offset}))
|
||
|
if not batch:
|
||
|
break
|
||
|
all_posts.extend(batch)
|
||
|
offset += increment
|
||
|
print(f"Total existing posts fetched: {len(all_posts)}")
|
||
|
return {post.title: post.id for post in all_posts}
|
||
|
|
||
|
|
||
|
def scan_directory_for_posts(directory, category_prefix=""):
|
||
|
posts = []
|
||
|
post_count = 1
|
||
|
for root, dirs, files in os.walk(directory):
|
||
|
category = category_prefix + os.path.basename(root) # 使用文件夹作为类别
|
||
|
for file in files:
|
||
|
if file.endswith('.md'):
|
||
|
file_path = os.path.join(root, file)
|
||
|
post_title, post_data = process_md_file(file_path)
|
||
|
if post_title and post_data:
|
||
|
posts.append((post_title, post_data, category, post_count))
|
||
|
post_count += 1
|
||
|
return posts
|
||
|
|
||
|
|
||
|
def process_md_file(file_path):
|
||
|
with open(file_path, 'r', encoding='utf-8') as f:
|
||
|
post_data = frontmatter.load(f)
|
||
|
|
||
|
# 构造 WordPress 文章
|
||
|
post_title = post_data['title'] if 'title' in post_data else os.path.basename(file_path)
|
||
|
return post_title, post_data
|
||
|
|
||
|
|
||
|
def create_wordpress_post(post_title, post_data, category):
|
||
|
# 构造 WordPress 文章
|
||
|
post = WordPressPost()
|
||
|
post.title = post_title
|
||
|
post.content = post_data.content
|
||
|
post.terms_names = {
|
||
|
'category': [category],
|
||
|
'post_tag': post_data['tags'] if 'tags' in post_data else [],
|
||
|
}
|
||
|
|
||
|
if 'date' in post_data:
|
||
|
post.date = datetime.strptime(str(post_data['date']), '%Y-%m-%d %H:%M:%S')
|
||
|
|
||
|
post.post_status = 'publish' # 或 'draft' 保存为草稿
|
||
|
|
||
|
return post
|
||
|
|
||
|
|
||
|
def upload_post_with_retries(post, post_number, retries=3):
|
||
|
global successful_posts, failed_posts
|
||
|
attempt = 0
|
||
|
while attempt < retries:
|
||
|
try:
|
||
|
post_id = wp.call(NewPost(post))
|
||
|
print(f"Post {post_number}: '{post.title}' - Upload successful. WordPress ID: {post_id}")
|
||
|
successful_posts += 1
|
||
|
return True
|
||
|
except Exception as e:
|
||
|
attempt += 1
|
||
|
print(f"Post {post_number}: '{post.title}' - Attempt {attempt} failed with error: {e}")
|
||
|
if attempt == retries:
|
||
|
print(f"Post {post_number}: '{post.title}' - Failed to upload after {retries} attempts.")
|
||
|
failed_posts += 1
|
||
|
return False
|
||
|
|
||
|
|
||
|
def upload_posts_to_wordpress(posts_to_upload):
|
||
|
global total_posts
|
||
|
total_posts = len(posts_to_upload)
|
||
|
for post, post_number in posts_to_upload:
|
||
|
try:
|
||
|
upload_post_with_retries(post, post_number)
|
||
|
except Exception:
|
||
|
print(f"Post {post_number}: '{post.title}' - Failed permanently.")
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
existing_wp_posts = get_all_wp_posts()
|
||
|
posts = scan_directory_for_posts(hexo_posts_dir)
|
||
|
posts_to_upload = []
|
||
|
|
||
|
for post_title, post_data, category, post_number in posts:
|
||
|
if post_title in existing_wp_posts:
|
||
|
print(f"Post '{post_title}' skipped: already exists.")
|
||
|
duplicate_posts.append((post_title, post_number))
|
||
|
else:
|
||
|
post = create_wordpress_post(post_title, post_data, category)
|
||
|
posts_to_upload.append((post, len(posts_to_upload) + 1))
|
||
|
|
||
|
upload_posts_to_wordpress(posts_to_upload)
|
||
|
|
||
|
print("Migration completed!")
|
||
|
print(f"Total posts processed: {total_posts}")
|
||
|
print(f"Successfully uploaded: {successful_posts}")
|
||
|
print(f"Failed uploads: {failed_posts}")
|
||
|
if duplicate_posts:
|
||
|
print("\nDuplicate posts found:")
|
||
|
for title, post_number in duplicate_posts:
|
||
|
print(f"Title: {title} - File Number: {post_number}")
|
||
|
else:
|
||
|
print("\nNo duplicate posts found.")
|
||
|
|