Python代码实现Twikoo品评中的图片显示

利用服务器公网IP和域名，通过简单的Python代码实现Twikoo品评中的图片功能。

由于没有购买和注册图床服务，且没有绑定腾讯云服务，所以只能自行想办法实现Twikoo的品评带图片功能。
想起来之前工作中有利用FastAPI实现过在Streamlit大模型RAG检索对话中的图片显示功能，因此考虑在此复用。
检察Twikoo代码中的哀求方式

Twikoo官方GitHub地址: twikoo
代码路径: src/server/function/twikoo/utils/image.js
代码地址: https://github.com/twikoojs/twikoo/blob/main/src/server/function/twikoo/utils/image.js
品评图床引用代码片段:

1. if (!config.IMAGE_CDN || !config.IMAGE_CDN_TOKEN) {
2.     throw new Error('未配置图片上传服务')
3.     }
4.     // tip: qcloud 图床走前端上传，其他图床走后端上传
5.     if (config.IMAGE_CDN === '7bu') {
6.     await fn.uploadImageToLskyPro({ photo, fileName, config, res, imageCdn: 'https://7bu.top' })
7.     } else if (config.IMAGE_CDN === 'smms') {
8.     await fn.uploadImageToSmms({ photo, fileName, config, res, imageCdn: 'https://smms.app/api/v2/upload' })
9.     } else if (isUrl(config.IMAGE_CDN)) {
10.     await fn.uploadImageToLskyPro({ photo, fileName, config, res, imageCdn: config.IMAGE_CDN })
11.     } else if (config.IMAGE_CDN === 'lskypro') {
12.     await fn.uploadImageToLskyPro({ photo, fileName, config, res, imageCdn: config.IMAGE_CDN_URL })
13.     } else if (config.IMAGE_CDN === 'piclist') {
14.     await fn.uploadImageToPicList({ photo, fileName, config, res, imageCdn: config.IMAGE_CDN_URL })
15.     } else {
16.     throw new Error('不支持的图片上传服务')
17. }

复制代码

在源代码的第24行，有 if (isUrl(config.IMAGE_CDN))，即支持自定义图片功能的地址，因此只需要在方法中搞清楚哀求头的相干参数即可。
哀求代码如下：

1. async uploadImageToLskyPro({ photo, fileName, config, res, imageCdn }) {
2.     // 自定义兰空图床（v2）URL
3.     const formData = new FormData()
4.     formData.append('file', fn.base64UrlToReadStream(photo, fileName))
5.     if (process.env.TWIKOO_LSKY_STRATEGY_ID) {
6.         formData.append('strategy_id', parseInt(process.env.TWIKOO_LSKY_STRATEGY_ID))
7.     }
8.     const url = `${imageCdn}/api/v1/upload`
9.     let token = config.IMAGE_CDN_TOKEN
10.     if (!token.startsWith('Bearer')) {
11.         token = `Bearer ${token}`
12.     }
13.     const uploadResult = await axios.post(url, formData, {
14.         headers: {
15.             ...formData.getHeaders(),
16.             Authorization: token
17.         }
18.     })
19.     if (uploadResult.data.status) {
20.         res.data = uploadResult.data.data
21.         res.data.url = res.data.links.url
22.     } else {
23.         throw new Error(uploadResult.data.message)
24.     }
25. },

复制代码

确定好哀求头的相干参数，下面即可利用Python的FastAPI包实现图片服务，详细代码如下：

1. from fastapi import FastAPI, UploadFile, File, Form, HTTPException, Header, Depends
2. from fastapi.middleware.cors import CORSMiddleware
3. from fastapi.staticfiles import StaticFiles
4. from typing import Optional
5. import os
7. app = FastAPI()
9. UPLOAD_Folder = 'Saved_Images'
11. # 确保有一个目录来保存上传的图片
12. os.makedirs(UPLOAD_Folder, exist_ok=True)
14. # 确定图片在网页浏览的基础地址，这里默认/images，即`https://xxxx.com/images/file_name.jpg`
15. app.mount("/images", StaticFiles(directory=UPLOAD_Folder), name="static")
18. @app.post("/")
19. async def image_server():
20.     return {'message': 'Hello'}
23. @app.post("/api/v1/upload")
24. async def upload_image(
25.     file: UploadFile = File(...),  # 接收文件
26.     strategy_id: int = Form(None),  # 接收表单字段，设置默认值为None
27.     authorization: str = Header(...)  # 接收Authorization请求头
28. ):
29.     if authorization != 'Bearer xxx':  # 这里可以设置相关鉴权的KEY等 --> 对应 `IMAGE_CDN_TOKEN`
30.         reponse = {
31.             'status': False,
32.             'data': {
33.                 'links': {
34.                     'url': '--'
35.                 }
36.             },
37.             'message': 'Toekn Invalid..'
38.         }
39.         return reponse
40.     # 读取上传的文件
41.     file_content = await file.read()
43.     # 保存图片到服务器
44.     cur_folder = authorization.replace('Bearer ', '')
45.     if cur_folder:
46.         os.makedirs(f'{UPLOAD_Folder}/{cur_folder}', exist_ok=True)
47.         file_path = f'{UPLOAD_Folder}/{cur_folder}/{file.filename}'
48.     else:
49.         os.makedirs(f'{UPLOAD_Folder}', exist_ok=True)
50.         file_path = f'{UPLOAD_Folder}/{file.filename}'
51.     with open(file_path, "wb") as f:
52.         f.write(file_content)
53.    
54.     upload_success = True  # 假设上传总是成功的，你可以根据实际情况修改这个值
55.     if not os.path.exists(file_path):
56.         upload_success = False
57.     message = "File and form data received successfully"
58.     data = {
59.         "links": {
60.             "url": f"https://{域名orIP地址}/images/{cur_folder}/{file.filename}"  # 假设的URL  --> 对应 `IMAGE_CDN`
61.         }
62.     }
64.     # 构造响应
65.     response = {
66.         "status": upload_success,
67.         "data": data if upload_success else None,
68.         "message": message if upload_success else "Upload failed"
69.     }
71.     # 如果上传失败，你可以抛出一个HTTPException
72.     if not upload_success:
73.         raise HTTPException(status_code=400, detail=response["message"])
74.     return response
77. # 在你的主程序中运行Uvicorn服务器
78. if __name__ == "__main__":
79.     import uvicorn
80.     uvicorn.run(app, host="0.0.0.0", port=8005)

复制代码

至此，基于Pythond FastAPI的图片服务已经实现，附Python代码哀求的脚本：

1. import requests
2. from fastapi import HTTPException
3. from typing import Optional
5. # 假设的服务器地址和端点
6. url = "http://127.0.0.1:8005/api/v1/upload"
8. # 要上传的文件路径
9. file_path = "star.png"
11. # 表单数据
12. form_data = {
13.     "strategy_id": 123  # 假设的表单字段值
14. }
16. # 请求头
17. headers = {
18.     "Authorization": "Bearer xxx"  # 替换为实际的鉴权令牌
19. }
21. # 打开文件
22. with open(file_path, "rb") as file:
23.     files = {"file": (file_path, file, "multipart/form-data")}
25.     # 发送请求
26.     try:
27.         response = requests.post(url, files=files, data=form_data, headers=headers)
28.         response.raise_for_status()  # 如果请求返回了失败的状态码，将抛出HTTPError异常
29.         print(response.json())  # 打印服务器响应
30.     except requests.exceptions.HTTPError as http_err:
31.         print(f"HTTP error occurred: {http_err}")  # 打印HTTP错误
32.     except Exception as err:
33.         print(f"An error occurred: {err}")  # 打印其他错误

复制代码

Twikoo管理页面配置

需配置的地方: IMAGE_CDN, IMAGE_CDN_TOKEN

配置完成，即可实现Twikoo品评带图片啦~

Note: 为确保能正常显示在公网，建议利用https域名进行配置，单纯IP可能不一定正常显示，未测试。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。