choozmo
/
CMM_API


			
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252
							import copy
import fastapi
import fastapi.staticfiles as fastapiStaticfiles
import linebot
import linebot.models as linebotModels
import suggests
import os
from GoogleNews import GoogleNews
import dataset
import datetime
from fastapi.responses import HTMLResponse
#

from linebot.models import (
    MessageEvent, TextMessage, TextSendMessage,
    SourceUser, SourceGroup, SourceRoom,
    TemplateSendMessage, ConfirmTemplate, MessageAction,
    ButtonsTemplate, ImageCarouselTemplate, ImageCarouselColumn, URIAction,
    PostbackAction, DatetimePickerAction,
    CameraAction, CameraRollAction, LocationAction,
    CarouselTemplate, CarouselColumn, PostbackEvent,
    StickerMessage, StickerSendMessage, LocationMessage, LocationSendMessage,
    ImageMessage, VideoMessage, AudioMessage, FileMessage,
    UnfollowEvent, FollowEvent, JoinEvent, LeaveEvent, BeaconEvent,
    MemberJoinedEvent, MemberLeftEvent,
    FlexSendMessage, BubbleContainer, ImageComponent, BoxComponent,
    TextComponent, IconComponent, ButtonComponent,
    SeparatorComponent, QuickReply, QuickReplyButton,
    ImageSendMessage)

#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
#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

#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

# --keyfile=./key.pem --certfile=./cert.pem 

#   --ssl-cert-reqs 1 
# 
# --ssl-ca-certs=/etc/letsencrypt/live/ptt.cx/fullchain.crt

app = fastapi.FastAPI()
#app.mount(
#    '/static', fastapiStaticfiles.StaticFiles(directory='static'), name='static')


from linebot import (
    LineBotApi, WebhookHandler
)
from linebot.exceptions import (
    InvalidSignatureError
)
from linebot.models import (
    MessageEvent, TextMessage,ImageSendMessage, TextSendMessage,FlexSendMessage, TemplateSendMessage,CarouselTemplate,ConfirmTemplate,PostbackAction,MessageAction,CarouselColumn,URIAction
)
import json
import codecs
seo=False
s_news=False
line_bot_api = LineBotApi('ExfYlsiGTvqJsMNdixocQ6PLYD+szmUj+GAxH99TXFPCn0RCX/iw06acFnL37pyHW1AYMZf34SOJEIRpdWb9heb9btIF2DaMqcBkYvJ3O2jYaKYv9RDUKbS57lUqcvvChK7ESQeuO2yjvYwes3Oq4wdB04t89/1O/w1cDnyilFU=')
handler = WebhookHandler('5b05fed81cd9f6f1e75dbf7dda388479')
#line_bot_api = LineBotApi('YOUR_CHANNEL_ACCESS_TOKEN')
#handler = WebhookHandler('YOUR_CHANNEL_SECRET')

def get_aws():
    result=''
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/hhh?charset=utf8mb4')
    cursor=db.query('SELECT from_unixtime(eventtime) as dt,area FROM hhh.aws_monitor order by eventtime desc limit 4;')
    for c in cursor:
        result+=str(c['dt'])+"\n"+c['area']+"\n"
    return result


def get_idea():
    result=''
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/hhh?charset=utf8mb4')
    cursor=db.query('SELECT query FROM hhh.gsc_weekly where clicks > 500 order by rand() limit 10;')
    for c in cursor:
        result+=str(c['query'])+"\n"
    return result


@app.get("/aws")
async def aws():
    result='<html><head><link href="https://getbootstrap.com/docs/4.1/dist/css/bootstrap.min.css" rel="stylesheet"></head><body>'
    result+='製表時間: '+str(datetime.datetime.now())+'</br></br>\n\n'
    result+='<table class="table table-striped">'
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/hhh?charset=utf8mb4')
    cursor=db.query('SELECT from_unixtime(eventtime) as dt,area FROM hhh.aws_monitor order by eventtime desc limit 30;')
    for c in cursor:
        result+="<tr><td>"+str(c['dt'])+"</td><td>"+c['area']+"</td></tr>\n"
    result+='</table></body></html>'
    return  HTMLResponse(content=result, status_code=200)

@app.get("/msg/{item_id}")
async def coffee_msg(item_id):
    True
    line_bot_api.push_message(item_id, TextSendMessage(text='開啟下方完整券樣（密碼9888）至櫃台掃碼兌換。https://txp.rs/v/EvX69b4Xq9'))
    return {"code":"ok" }


@app.post('/callback')
async def callback(request: fastapi.Request):
    signature = request.headers['X-Line-Signature']
    body = await request.body()
    handler.handle(body.decode('utf-8'), signature)
    return 'OK'


def get_news_by_kw(keyword):
    googlenews = GoogleNews(lang='zh-TW')
    kw=keyword
    googlenews.set_lang('zh-TW')
    googlenews.search(kw)
    resultstr="新聞:"
    idx=0
    rs=googlenews.results()
    for r in rs:
        if idx>0:
            resultstr+=','
        else:
            idx+=1
        resultstr+=r['title']
    return resultstr
#        print(r['desc'])
#        print(r['link'])
#        print(r['datetime'])


def flex_test():
    fname=os.path.abspath(__file__)
    elmts=fname.split(os.path.sep)
    path2=os.path.sep.join(elmts[0:-1])
    keysdir=path2+os.path.sep
    js=json.load(open(keysdir+'test.json','r',encoding='utf-8'))
    return js


@handler.add(FollowEvent)
def handle_follow(event):
    print(event.source.user_id)
    
    # do something


@handler.add(linebotModels.MessageEvent, message=linebotModels.TextMessage)
def message_text(event):
    global seo
    global s_news
##    if event.message.text == 'push':
#        line_bot_api.push_message(
#                event.source.user_id, [
# #                   TextSendMessage(text='PUSH!'),
#                ]
#            )

    if event.message.text.lower() == 'flex':
        FlexMessage = flex_test()
        line_bot_api.reply_message(event.reply_token, FlexSendMessage('ChoozMo',FlexMessage))
        return

    if event.message.text.lower() == 's_news':
        s_news=True
        line_bot_api.reply_message(
            event.reply_token,
            TextSendMessage(text='請輸入要搜尋新聞的關鍵字:'))
        return

    if event.message.text == 'q_aws':
        line_bot_api.reply_message(
            event.reply_token,[TextSendMessage(text=get_aws()),TextSendMessage(text='完整報告: https://api.ptt.cx:5443/aws')])
                    
        return

    if event.message.text == 'q_idea':
        s_news=True
        line_bot_api.reply_message(
            event.reply_token,
            TextSendMessage(text=get_idea()))
        return


    if event.message.text.lower() == 'seo':
        seo=True
        line_bot_api.reply_message(
            event.reply_token,
            TextSendMessage(text='請輸入要找的關鍵字:'))
        return

    if event.message.text == '叫':
        line_bot_api.reply_message(
            event.reply_token, linebotModels.AudioSendMessage(
                original_content_url=f'{baseUrl}/static/audio/noot_noot.mp3', duration=1000))
    if event.message.text.lower()=='ok':
        line_bot_api.reply_message(
            event.reply_token,
            TextSendMessage(text='最欣賞ChoozMo團隊說OK的人!!'))
    if event.message.text=='c' or event.message.text=='C' :
        line_bot_api.reply_message(
                    event.reply_token,
                    TextSendMessage(
                        text='快速鍵',
                        quick_reply=QuickReply(
                            items=[
                                QuickReplyButton(
                                    action=MessageAction(label="幸福空間靈感", text="q_idea")
                                ),
                                QuickReplyButton(
                                    action=MessageAction(label="查AWS", text="q_aws")
                                ),
                                QuickReplyButton(
                                    action=MessageAction(label="關聯字", text="seo")
                                ),

                            ])))
    else:
        if s_news:
            result=get_news_by_kw(event.message.text)
            line_bot_api.reply_message(
                event.reply_token,
                TextSendMessage(text=result))
            s_news=False
            return
        if seo:
            res='相關字:'
            s = suggests.suggests.get_suggests(event.message.text, source='google')
            idx=0
            for sg in s['suggests']:
                if idx>0:
                    res+=','
                else:
                    idx+=1
                res+=sg


            line_bot_api.reply_message(
                event.reply_token,
                TextSendMessage(text=res))
            seo=False

#        print('test')


#if __name__ == "__main__":
#    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'))
#    app.run(host='0.0.0.0', port=14404,ssl_context=('/tmp/cert.pem','/tmp/chain.pem' ))