39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Update docs/index.html to add model download gate via Gitea API"""
|
||
|
|
import requests, json, base64
|
||
|
|
|
||
|
|
TOKEN = '2d924a6e19b44d14bf367f5dd1ce238510128183'
|
||
|
|
URL = 'https://guanghulab.com/code/api/v1/repos/bingshuo/guanghulab/contents/docs/index.html'
|
||
|
|
HEADERS = {'Authorization': f'Bearer {TOKEN}'}
|
||
|
|
|
||
|
|
# Get current file
|
||
|
|
resp = requests.get(URL, headers=HEADERS)
|
||
|
|
data = resp.json()
|
||
|
|
sha = data['sha']
|
||
|
|
content = base64.b64decode(data['content']).decode('utf-8')
|
||
|
|
|
||
|
|
# Add download models gate after the first gate (光湖智库)
|
||
|
|
new_gate = '''
|
||
|
|
<a class="gate" href="/download-models.html">
|
||
|
|
<div class="gate-head"><span class="gate-icon">📥</span><span class="gate-title">模型下载</span></div>
|
||
|
|
<div class="gate-desc">1.5B人格模型模板下载</div>
|
||
|
|
<div class="gate-url">download-models.html →</div>
|
||
|
|
</a>'''
|
||
|
|
|
||
|
|
# Insert after the first </a> in the gates section
|
||
|
|
insert_pos = content.find('</a>', content.find('<section class="gates')) + 4
|
||
|
|
new_content = content[:insert_pos] + new_gate + content[insert_pos:]
|
||
|
|
|
||
|
|
# Encode and push
|
||
|
|
b64 = base64.b64encode(new_content.encode('utf-8')).decode('utf-8')
|
||
|
|
payload = {
|
||
|
|
'content': b64,
|
||
|
|
'sha': sha,
|
||
|
|
'message': '首页添加模型下载入口 gate'
|
||
|
|
}
|
||
|
|
resp2 = requests.put(URL, headers=HEADERS, json=payload)
|
||
|
|
if resp2.status_code == 200:
|
||
|
|
print(f'OK: commit {resp2.json()["commit"]["sha"][:12]}')
|
||
|
|
else:
|
||
|
|
print(f'FAIL: {resp2.status_code} {resp2.text}')
|