90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
"""
|
|
Publish plugins to OpenWebUI Community
|
|
使用 OpenWebUICommunityClient 发布插件到官方社区
|
|
|
|
用法：
|
|
python scripts/publish_plugin.py # 只更新有版本变化的插件
|
|
python scripts/publish_plugin.py --force # 强制更新所有插件
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import re
|
|
import argparse
|
|
|
|
# Add current directory to path
|
|
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from openwebui_community_client import OpenWebUICommunityClient, get_client
|
|
|
|
|
|
def find_plugins_with_id(plugins_dir: str) -> list:
|
|
"""查找所有带 openwebui_id 的插件文件"""
|
|
plugins = []
|
|
for root, _, files in os.walk(plugins_dir):
|
|
for file in files:
|
|
if file.endswith(".py"):
|
|
file_path = os.path.join(root, file)
|
|
with open(file_path, "r", encoding="utf-8") as f:
|
|
content = f.read(2000) # 只读前 2000 字符检查 ID
|
|
|
|
id_match = re.search(
|
|
r"(?:openwebui_id|post_id):\s*([a-z0-9-]+)", content
|
|
)
|
|
if id_match:
|
|
plugins.append(
|
|
{"file_path": file_path, "post_id": id_match.group(1).strip()}
|
|
)
|
|
return plugins
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Publish plugins to OpenWebUI Market")
|
|
parser.add_argument(
|
|
"--force", action="store_true", help="Force update even if version matches"
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
client = get_client()
|
|
except ValueError as e:
|
|
print(f"Error: {e}")
|
|
sys.exit(1)
|
|
|
|
base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
plugins_dir = os.path.join(base_dir, "plugins")
|
|
|
|
plugins = find_plugins_with_id(plugins_dir)
|
|
print(f"Found {len(plugins)} plugins with OpenWebUI ID.\n")
|
|
|
|
updated = 0
|
|
skipped = 0
|
|
failed = 0
|
|
|
|
for plugin in plugins:
|
|
file_path = plugin["file_path"]
|
|
file_name = os.path.basename(file_path)
|
|
post_id = plugin["post_id"]
|
|
|
|
print(f"Processing {file_name} (ID: {post_id})...")
|
|
|
|
success, message = client.publish_plugin_from_file(file_path, force=args.force)
|
|
|
|
if success:
|
|
if "Skipped" in message:
|
|
print(f" ⏭️ {message}")
|
|
skipped += 1
|
|
else:
|
|
print(f" ✅ {message}")
|
|
updated += 1
|
|
else:
|
|
print(f" ❌ {message}")
|
|
failed += 1
|
|
|
|
print(f"\n{'='*50}")
|
|
print(f"Finished: {updated} updated, {skipped} skipped, {failed} failed")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|