werobot_server.py 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. import random
  2. import time
  3. import requests
  4. import werobot
  5. from werobot.replies import ArticlesReply, Article, ImageReply, TextReply, MusicReply
  6. robot = werobot.WeRoBot(token='token')
  7. base_url = "https://yourdomain/api"
  8. headers = {
  9. "j-auth-token": "your_token"
  10. }
  11. def make_request(endpoint, params=None):
  12. response = requests.get(f"{base_url}/{endpoint}", params=params, headers=headers)
  13. if response.status_code == 200:
  14. return response.json()
  15. elif response.status_code == 404:
  16. #print(f"未找到数据,URL: {base_url}/{endpoint}")
  17. return None
  18. else:
  19. #print(f"请求接口失败,状态码: {response.status_code}, URL: {base_url}/{endpoint}")
  20. return None
  21. # 获取影片详情
  22. def get_movie_details(movie_id):
  23. return make_request(f"movies/{movie_id}")
  24. # 获取磁力链接
  25. def get_movie_magnets(movie_id):
  26. # 获取影片详情数据
  27. movie_details_data = get_movie_details(movie_id)
  28. if movie_details_data:
  29. # 提取影片的 gid 和 uc
  30. gid = movie_details_data.get("gid", "")
  31. uc = movie_details_data.get("uc", "")
  32. # 调用磁力链接接口
  33. magnet_url = f"magnets/{movie_id}?gid={gid}&uc={uc}"
  34. movie_magnet_data = make_request(magnet_url)
  35. if movie_magnet_data is not None:
  36. link = movie_magnet_data[0]["link"]
  37. title = movie_magnet_data[0]["title"]
  38. size = movie_magnet_data[0]["size"]
  39. sharedate = movie_magnet_data[0]["shareDate"]
  40. return link, title, size, sharedate
  41. return "未找到"
  42. #获取种子
  43. @robot.text
  44. def zz(message):
  45. if message.content[:2] == "种子":
  46. id = message.content[2:].replace(" ", "")
  47. magnets = get_movie_magnets(id)
  48. if magnets is not None:
  49. return f"磁力链接:{magnets[0]}\n标题:{magnets[1]}\n文件大小:{magnets[2]}\n分享日期:{magnets[3]}"
  50. return f"未找到{id}"
  51. @robot.text
  52. def replay(msg):
  53. return "要发关键词哦!"
  54. robot.config['HOST'] = '0.0.0.0'
  55. robot.config['PORT'] = 4080
  56. robot.run()