newbot.py 8.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242
  1. import copy
  2. import fastapi
  3. import fastapi.staticfiles as fastapiStaticfiles
  4. import linebot
  5. import linebot.models as linebotModels
  6. import suggests
  7. from GoogleNews import GoogleNews
  8. import dataset
  9. import datetime
  10. from fastapi.responses import HTMLResponse
  11. #
  12. from linebot.models import (
  13. MessageEvent, TextMessage, TextSendMessage,
  14. SourceUser, SourceGroup, SourceRoom,
  15. TemplateSendMessage, ConfirmTemplate, MessageAction,
  16. ButtonsTemplate, ImageCarouselTemplate, ImageCarouselColumn, URIAction,
  17. PostbackAction, DatetimePickerAction,
  18. CameraAction, CameraRollAction, LocationAction,
  19. CarouselTemplate, CarouselColumn, PostbackEvent,
  20. StickerMessage, StickerSendMessage, LocationMessage, LocationSendMessage,
  21. ImageMessage, VideoMessage, AudioMessage, FileMessage,
  22. UnfollowEvent, FollowEvent, JoinEvent, LeaveEvent, BeaconEvent,
  23. MemberJoinedEvent, MemberLeftEvent,
  24. FlexSendMessage, BubbleContainer, ImageComponent, BoxComponent,
  25. TextComponent, IconComponent, ButtonComponent,
  26. SeparatorComponent, QuickReply, QuickReplyButton,
  27. ImageSendMessage)
  28. #uvicorn newbot:app --host 0.0.0.0 --port 5443 --ssl-keyfile=/etc/letsencrypt/live/api.ptt.cx/privkey.pem --ssl-certfile=/etc/letsencrypt/live/api.ptt.cx/fullchain.pem
  29. #uvicorn main:app --host 0.0.0.0 --port 443 --ssl-keyfile=/etc/letsencrypt/live/ptt.cx/privkey.pem --ssl-certfile=/etc/letsencrypt/live/ptt.cx/chain.pem
  30. #uvicorn main:app --host 0.0.0.0 --port 443 --key-file=/etc/letsencrypt/live/ptt.cx/privkey.pem --certfile=/etc/letsencrypt/live/ptt.cx/cert.pem
  31. # --keyfile=./key.pem --certfile=./cert.pem
  32. # --ssl-cert-reqs 1
  33. #
  34. # --ssl-ca-certs=/etc/letsencrypt/live/ptt.cx/fullchain.crt
  35. app = fastapi.FastAPI()
  36. #app.mount(
  37. # '/static', fastapiStaticfiles.StaticFiles(directory='static'), name='static')
  38. from linebot import (
  39. LineBotApi, WebhookHandler
  40. )
  41. from linebot.exceptions import (
  42. InvalidSignatureError
  43. )
  44. from linebot.models import (
  45. MessageEvent, TextMessage,ImageSendMessage, TextSendMessage,FlexSendMessage, TemplateSendMessage,CarouselTemplate,ConfirmTemplate,PostbackAction,MessageAction,CarouselColumn,URIAction
  46. )
  47. import json
  48. import codecs
  49. seo=False
  50. s_news=False
  51. line_bot_api = LineBotApi('Gub58t+8u9z7nTBrZLLvXnCbwDR1Gmyrew3nGFlKRxsmvH/oIMGjjup2DygA4XhV1NenM6dFTO8yvpbtCDezCqP2BAV4eVn2Y63/EYqeTCw1S+oJ+BiLLzC8DdVrWu9jSfp7wXqrVNUxcYdk54eoWAdB04t89/1O/w1cDnyilFU=')
  52. handler = WebhookHandler('f761bc6038c94a3baa815124e33dea50')
  53. #line_bot_api = LineBotApi('YOUR_CHANNEL_ACCESS_TOKEN')
  54. #handler = WebhookHandler('YOUR_CHANNEL_SECRET')
  55. def get_aws():
  56. result=''
  57. db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/hhh?charset=utf8mb4')
  58. cursor=db.query('SELECT from_unixtime(eventtime) as dt,area FROM hhh.aws_monitor order by eventtime desc limit 4;')
  59. for c in cursor:
  60. result+=str(c['dt'])+"\n"+c['area']+"\n"
  61. return result
  62. def get_idea():
  63. result=''
  64. db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/hhh?charset=utf8mb4')
  65. cursor=db.query('SELECT query FROM hhh.gsc_weekly where clicks > 500 order by rand() limit 10;')
  66. for c in cursor:
  67. result+=str(c['query'])+"\n"
  68. return result
  69. @app.get("/aws")
  70. async def aws():
  71. result='<html><head><link href="https://getbootstrap.com/docs/4.1/dist/css/bootstrap.min.css" rel="stylesheet"></head><body>'
  72. result+='製表時間: '+str(datetime.datetime.now())+'</br></br>\n\n'
  73. result+='<table class="table table-striped">'
  74. db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/hhh?charset=utf8mb4')
  75. cursor=db.query('SELECT from_unixtime(eventtime) as dt,area FROM hhh.aws_monitor order by eventtime desc limit 30;')
  76. for c in cursor:
  77. result+="<tr><td>"+str(c['dt'])+"</td><td>"+c['area']+"</td></tr>\n"
  78. result+='</table></body></html>'
  79. return HTMLResponse(content=result, status_code=200)
  80. @app.get("/msg/{item_id}")
  81. async def coffee_msg(item_id):
  82. True
  83. line_bot_api.push_message(item_id, TextSendMessage(text='開啟下方完整券樣(密碼9888)至櫃台掃碼兌換。https://txp.rs/v/EvX69b4Xq9'))
  84. return {"code":"ok" }
  85. @app.post('/callback')
  86. async def callback(request: fastapi.Request):
  87. signature = request.headers['X-Line-Signature']
  88. body = await request.body()
  89. handler.handle(body.decode('utf-8'), signature)
  90. return 'OK'
  91. def get_news_by_kw(keyword):
  92. googlenews = GoogleNews(lang='zh-TW')
  93. kw=keyword
  94. googlenews.set_lang('zh-TW')
  95. googlenews.search(kw)
  96. resultstr="新聞:"
  97. idx=0
  98. rs=googlenews.results()
  99. for r in rs:
  100. if idx>0:
  101. resultstr+=','
  102. else:
  103. idx+=1
  104. resultstr+=r['title']
  105. return resultstr
  106. # print(r['desc'])
  107. # print(r['link'])
  108. # print(r['datetime'])
  109. @handler.add(FollowEvent)
  110. def handle_follow(event):
  111. print(event.source.user_id)
  112. # do something
  113. @handler.add(linebotModels.MessageEvent, message=linebotModels.TextMessage)
  114. def message_text(event):
  115. global seo
  116. global s_news
  117. ## if event.message.text == 'push':
  118. # line_bot_api.push_message(
  119. # event.source.user_id, [
  120. # # TextSendMessage(text='PUSH!'),
  121. # ]
  122. # )
  123. if event.message.text == 's_news':
  124. s_news=True
  125. line_bot_api.reply_message(
  126. event.reply_token,
  127. TextSendMessage(text='請輸入要搜尋新聞的關鍵字:'))
  128. return
  129. if event.message.text == 'q_aws':
  130. line_bot_api.reply_message(
  131. event.reply_token,[TextSendMessage(text=get_aws()),TextSendMessage(text='完整報告: https://api.ptt.cx:5443/aws')])
  132. return
  133. if event.message.text == 'q_idea':
  134. s_news=True
  135. line_bot_api.reply_message(
  136. event.reply_token,
  137. TextSendMessage(text=get_idea()))
  138. return
  139. if event.message.text == 'seo':
  140. seo=True
  141. line_bot_api.reply_message(
  142. event.reply_token,
  143. TextSendMessage(text='請輸入要找的關鍵字:'))
  144. return
  145. if event.message.text=='熱門關鍵字':
  146. line_bot_api.reply_message(
  147. event.reply_token,
  148. TextSendMessage(text='最欣賞ChoozMo團隊說ooooK的人!!'))
  149. if event.message.text=='關鍵字歷史走勢':
  150. line_bot_api.reply_message(
  151. event.reply_token,
  152. TextSendMessage(text='最欣賞ChoozMo團隊說OkkkkK的人!!'))
  153. if event.message.text == '叫':
  154. line_bot_api.reply_message(
  155. event.reply_token, linebotModels.AudioSendMessage(
  156. original_content_url=f'{baseUrl}/static/audio/noot_noot.mp3', duration=1000))
  157. if event.message.text=='ok':
  158. line_bot_api.reply_message(
  159. event.reply_token,
  160. TextSendMessage(text='最欣賞ChoozMo團隊說OK的人!!'))
  161. if event.message.text=='c' or event.message.text=='C' :
  162. line_bot_api.reply_message(
  163. event.reply_token,
  164. TextSendMessage(
  165. text='快速鍵',
  166. quick_reply=QuickReply(
  167. items=[
  168. QuickReplyButton(
  169. action=MessageAction(label="幸福空間靈感", text="q_idea")
  170. ),
  171. QuickReplyButton(
  172. action=MessageAction(label="查AWS", text="q_aws")
  173. ),
  174. QuickReplyButton(
  175. action=MessageAction(label="關聯字", text="seo")
  176. ),
  177. ])))
  178. else:
  179. if s_news:
  180. result=get_news_by_kw(event.message.text)
  181. line_bot_api.reply_message(
  182. event.reply_token,
  183. TextSendMessage(text=result))
  184. s_news=False
  185. return
  186. if seo:
  187. res='相關字:'
  188. s = suggests.suggests.get_suggests(event.message.text, source='google')
  189. idx=0
  190. for sg in s['suggests']:
  191. if idx>0:
  192. res+=','
  193. else:
  194. idx+=1
  195. res+=sg
  196. line_bot_api.reply_message(
  197. event.reply_token,
  198. TextSendMessage(text=res))
  199. seo=False
  200. # print('test')
  201. #if __name__ == "__main__":
  202. # app.run(host='0.0.0.0', port=443,ssl_context=('/etc/letsencrypt/live/ptt.cx/fullchain.pem', '/etc/letsencrypt/live/ptt.cx/privkey.pem'))
  203. # app.run(host='0.0.0.0', port=14404,ssl_context=('/tmp/cert.pem','/tmp/chain.pem' ))