刘国栋
355 lines
12 KiB
Markdown
355 lines
12 KiB
Markdown
# 快速开始指南：实现GitHub发布功能
|
||
|
||
> 这是最简单的实现示例，可以作为其他平台的基础模板
|
||
|
||
## 🎯 目标
|
||
|
||
实现GitHub平台的文章自动发布功能，验证整体架构可行性。
|
||
|
||
## 📦 步骤1：安装依赖
|
||
|
||
```bash
|
||
pip install httpx pyperclip
|
||
```
|
||
|
||
## 📝 步骤2：扩展数据库
|
||
|
||
在 `modules/data_storage.py` 的 `_init_sqlite` 方法中添加：
|
||
|
||
```python
|
||
# 平台账号表
|
||
cursor.execute("""
|
||
CREATE TABLE IF NOT EXISTS platform_accounts (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
platform TEXT NOT NULL,
|
||
account_type TEXT NOT NULL,
|
||
account_name TEXT,
|
||
api_key TEXT,
|
||
config_json TEXT,
|
||
is_active INTEGER DEFAULT 1,
|
||
brand TEXT,
|
||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||
)
|
||
""")
|
||
|
||
# 发布记录表
|
||
cursor.execute("""
|
||
CREATE TABLE IF NOT EXISTS publish_records (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
article_id INTEGER,
|
||
platform TEXT NOT NULL,
|
||
publish_method TEXT NOT NULL,
|
||
publish_status TEXT NOT NULL,
|
||
publish_url TEXT,
|
||
publish_id TEXT,
|
||
error_message TEXT,
|
||
published_at TIMESTAMP,
|
||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||
)
|
||
""")
|
||
```
|
||
|
||
## 💻 步骤3：创建GitHub发布器
|
||
|
||
创建文件 `platform_sync/github_publisher.py`：
|
||
|
||
```python
|
||
"""
|
||
GitHub发布器 - 最简单的实现示例
|
||
"""
|
||
import base64
|
||
import httpx
|
||
from typing import Dict, Any, Optional
|
||
|
||
|
||
class GitHubPublisher:
|
||
"""GitHub发布器"""
|
||
|
||
def __init__(self, api_key: str, repo_owner: str, repo_name: str):
|
||
self.api_key = api_key
|
||
self.repo_owner = repo_owner
|
||
self.repo_name = repo_name
|
||
self.base_url = "https://api.github.com"
|
||
self.headers = {
|
||
"Authorization": f"token {api_key}",
|
||
"Accept": "application/vnd.github.v3+json"
|
||
}
|
||
|
||
def publish(self, content: str, title: str, file_path: Optional[str] = None) -> Dict[str, Any]:
|
||
"""
|
||
发布内容到GitHub
|
||
|
||
Args:
|
||
content: Markdown内容
|
||
title: 文章标题
|
||
file_path: 文件路径（可选）
|
||
|
||
Returns:
|
||
{
|
||
'success': bool,
|
||
'publish_url': str,
|
||
'publish_id': str,
|
||
'error': str
|
||
}
|
||
"""
|
||
try:
|
||
# 生成文件路径
|
||
if not file_path:
|
||
safe_title = title.replace(' ', '_').replace('/', '_')
|
||
file_path = f"content/{safe_title}.md"
|
||
|
||
# 编码内容
|
||
content_bytes = content.encode('utf-8')
|
||
content_base64 = base64.b64encode(content_bytes).decode('utf-8')
|
||
|
||
# API URL
|
||
url = f"{self.base_url}/repos/{self.repo_owner}/{self.repo_name}/contents/{file_path}"
|
||
|
||
# 检查文件是否存在
|
||
response = httpx.get(url, headers=self.headers)
|
||
sha = None
|
||
if response.status_code == 200:
|
||
sha = response.json().get('sha')
|
||
|
||
# 准备数据
|
||
data = {
|
||
"message": f"Publish: {title}",
|
||
"content": content_base64,
|
||
"branch": "main"
|
||
}
|
||
if sha:
|
||
data["sha"] = sha
|
||
|
||
# 创建或更新文件
|
||
response = httpx.put(url, json=data, headers=self.headers)
|
||
|
||
if response.status_code in [200, 201]:
|
||
result = response.json()
|
||
html_url = result.get('content', {}).get('html_url', '')
|
||
return {
|
||
'success': True,
|
||
'publish_url': html_url,
|
||
'publish_id': result.get('content', {}).get('sha', ''),
|
||
'error': None
|
||
}
|
||
else:
|
||
return {
|
||
'success': False,
|
||
'publish_url': '',
|
||
'publish_id': '',
|
||
'error': f"GitHub API错误: {response.text}"
|
||
}
|
||
except Exception as e:
|
||
return {
|
||
'success': False,
|
||
'publish_url': '',
|
||
'publish_id': '',
|
||
'error': str(e)
|
||
}
|
||
|
||
def validate_account(self) -> bool:
|
||
"""验证GitHub账号"""
|
||
try:
|
||
response = httpx.get(f"{self.base_url}/user", headers=self.headers)
|
||
return response.status_code == 200
|
||
except:
|
||
return False
|
||
```
|
||
|
||
## 🔧 步骤4：扩展DataStorage
|
||
|
||
在 `modules/data_storage.py` 的 `DataStorage` 类中添加：
|
||
|
||
```python
|
||
def save_platform_account(self, platform: str, account_config: Dict[str, Any], brand: str):
|
||
"""保存平台账号"""
|
||
if self.storage_type == "sqlite":
|
||
conn = sqlite3.connect(self.db_path)
|
||
cursor = conn.cursor()
|
||
cursor.execute("""
|
||
INSERT OR REPLACE INTO platform_accounts
|
||
(platform, account_type, account_name, api_key, config_json, brand, updated_at)
|
||
VALUES (?, ?, ?, ?, ?, ?, ?)
|
||
""", (
|
||
platform,
|
||
account_config.get('account_type', 'api'),
|
||
account_config.get('account_name', ''),
|
||
account_config.get('api_key', ''),
|
||
json.dumps(account_config.get('config', {}), ensure_ascii=False),
|
||
brand,
|
||
datetime.now().isoformat()
|
||
))
|
||
conn.commit()
|
||
conn.close()
|
||
|
||
def get_platform_account(self, platform: str, brand: str) -> Optional[Dict[str, Any]]:
|
||
"""获取平台账号"""
|
||
if self.storage_type == "sqlite":
|
||
conn = sqlite3.connect(self.db_path)
|
||
cursor = conn.cursor()
|
||
cursor.execute("""
|
||
SELECT * FROM platform_accounts
|
||
WHERE platform = ? AND brand = ? AND is_active = 1
|
||
""", (platform, brand))
|
||
row = cursor.fetchone()
|
||
conn.close()
|
||
|
||
if row:
|
||
return {
|
||
'api_key': row[4],
|
||
'config': json.loads(row[5] or '{}')
|
||
}
|
||
return None
|
||
|
||
def save_publish_record(self, article_id: int, platform: str, publish_method: str,
|
||
publish_status: str, publish_url: str = '', publish_id: str = '',
|
||
error_message: str = ''):
|
||
"""保存发布记录"""
|
||
if self.storage_type == "sqlite":
|
||
conn = sqlite3.connect(self.db_path)
|
||
cursor = conn.cursor()
|
||
cursor.execute("""
|
||
INSERT INTO publish_records
|
||
(article_id, platform, publish_method, publish_status, publish_url,
|
||
publish_id, error_message, published_at)
|
||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||
""", (
|
||
article_id, platform, publish_method, publish_status,
|
||
publish_url, publish_id, error_message, datetime.now().isoformat()
|
||
))
|
||
conn.commit()
|
||
conn.close()
|
||
```
|
||
|
||
## 🎨 步骤5：添加UI（在geo_tool.py中）
|
||
|
||
在 `modules/geo_tool.py` 中添加新的Tab或功能：
|
||
|
||
```python
|
||
# 在Tab定义中添加
|
||
tabs = st.tabs([
|
||
"1 关键词蒸馏",
|
||
"2 内容生成",
|
||
"3 内容优化",
|
||
"4 AI验证",
|
||
"5 历史记录",
|
||
"6 平台同步" # 新增
|
||
])
|
||
|
||
with tabs[5]: # 平台同步Tab
|
||
st.header("📤 平台文章同步")
|
||
|
||
# GitHub账号配置
|
||
with st.expander("🔐 GitHub账号配置", expanded=True):
|
||
github_api_key = st.text_input("GitHub Personal Access Token", type="password")
|
||
github_repo_owner = st.text_input("仓库所有者（用户名）")
|
||
github_repo_name = st.text_input("仓库名称")
|
||
|
||
if st.button("保存GitHub配置"):
|
||
if github_api_key and github_repo_owner and github_repo_name:
|
||
storage.save_platform_account(
|
||
platform="GitHub",
|
||
account_config={
|
||
'account_type': 'api',
|
||
'api_key': github_api_key,
|
||
'config': {
|
||
'repo_owner': github_repo_owner,
|
||
'repo_name': github_repo_name
|
||
}
|
||
},
|
||
brand=brand
|
||
)
|
||
st.success("GitHub配置已保存！")
|
||
else:
|
||
st.error("请填写完整信息")
|
||
|
||
# 发布功能
|
||
st.subheader("📝 发布到GitHub")
|
||
|
||
# 选择文章
|
||
articles = storage.get_articles(brand=brand)
|
||
if articles:
|
||
article_options = {f"{a['keyword']} - {a['platform']}": a['id'] for a in articles}
|
||
selected_article_key = st.selectbox("选择要发布的文章", list(article_options.keys()))
|
||
selected_article_id = article_options[selected_article_key]
|
||
|
||
if st.button("🚀 发布到GitHub", type="primary"):
|
||
# 获取账号配置
|
||
account_config = storage.get_platform_account("GitHub", brand)
|
||
if not account_config:
|
||
st.error("请先配置GitHub账号")
|
||
else:
|
||
# 获取文章
|
||
article = next((a for a in articles if a['id'] == selected_article_id), None)
|
||
if article:
|
||
# 创建发布器
|
||
from platform_sync.github_publisher import GitHubPublisher
|
||
publisher = GitHubPublisher(
|
||
api_key=account_config['api_key'],
|
||
repo_owner=account_config['config']['repo_owner'],
|
||
repo_name=account_config['config']['repo_name']
|
||
)
|
||
|
||
# 发布
|
||
with st.spinner("正在发布到GitHub..."):
|
||
result = publisher.publish(
|
||
content=article['content'],
|
||
title=article['keyword']
|
||
)
|
||
|
||
# 保存发布记录
|
||
storage.save_publish_record(
|
||
article_id=selected_article_id,
|
||
platform="GitHub",
|
||
publish_method="api",
|
||
publish_status="success" if result['success'] else "failed",
|
||
publish_url=result.get('publish_url', ''),
|
||
publish_id=result.get('publish_id', ''),
|
||
error_message=result.get('error', '')
|
||
)
|
||
|
||
# 显示结果
|
||
if result['success']:
|
||
st.success(f"✅ 发布成功！")
|
||
st.markdown(f"**发布链接**: {result['publish_url']}")
|
||
else:
|
||
st.error(f"❌ 发布失败: {result.get('error', '未知错误')}")
|
||
else:
|
||
st.info("请先生成文章")
|
||
|
||
# 发布记录
|
||
st.subheader("📊 发布记录")
|
||
# 这里可以显示发布历史记录
|
||
```
|
||
|
||
## ✅ 步骤6：测试
|
||
|
||
1. **获取GitHub Token**：
|
||
- 访问 [https://github.com/settings/tokens](https://github.com/settings/tokens)
|
||
- 创建新的 Personal Access Token
|
||
- 选择 `repo` 权限
|
||
2. **运行测试**：
|
||
```bash
|
||
streamlit run geo_tool.py
|
||
```
|
||
3. **测试流程**：
|
||
- 配置GitHub账号
|
||
- 生成一篇文章
|
||
- 发布到GitHub
|
||
- 检查GitHub仓库是否成功创建文件
|
||
|
||
## 🎉 完成！
|
||
|
||
如果GitHub发布功能正常工作，说明架构是正确的。接下来可以：
|
||
|
||
1. 按照相同模式实现其他7个API平台
|
||
2. 实现一键复制功能
|
||
3. 添加批量发布功能
|
||
|
||
## 📚 参考
|
||
|
||
- [GitHub API文档](https://docs.github.com/en/rest)
|
||
- [完整实现指南](./PLATFORM_SYNC_IMPLEMENTATION.md)
|
||
|