Browse Source
Merge pull request #2153 from 6vision/update_linkaibot
support more file types.master
vision
GitHub
4 months ago
1 changed files with 2 additions and 1 deletions
+ 2
- 1
bot/linkai/link_ai_bot.py
View File
| @@ -399,6 +399,7 @@ class LinkAIBot(Bot): | |||||
| return | return | ||||
| max_send_num = conf().get("max_media_send_count") | max_send_num = conf().get("max_media_send_count") | ||||
| send_interval = conf().get("media_send_interval") | send_interval = conf().get("media_send_interval") | ||||
| file_type = (".pdf", ".doc", ".docx", ".csv", ".xls", ".xlsx", ".txt", ".rtf", ".ppt", ".pptx") | |||||
| try: | try: | ||||
| i = 0 | i = 0 | ||||
| for url in image_urls: | for url in image_urls: | ||||
| @@ -407,7 +408,7 @@ class LinkAIBot(Bot): | |||||
| i += 1 | i += 1 | ||||
| if url.endswith(".mp4"): | if url.endswith(".mp4"): | ||||
| reply_type = ReplyType.VIDEO_URL | reply_type = ReplyType.VIDEO_URL | ||||
| elif url.endswith(".pdf") or url.endswith(".doc") or url.endswith(".docx") or url.endswith(".csv"): | |||||
| elif url.endswith(file_type): | |||||
| reply_type = ReplyType.FILE | reply_type = ReplyType.FILE | ||||
| url = _download_file(url) | url = _download_file(url) | ||||
| if not url: | if not url: | ||||
Loading…