將Linebot與Google sheet API做結合，製作記帳機器人

LinebotWithGoogle.py

#encoding=UTF-8
"""
Author:Ethan Yang
Datetime:2016/12/29
Purpose:
1.對話是經濟?!
2.利用Line當作輸入指令的窗口。
3.記帳。
4.查詢單字
"""
from flask import Flask,request
import requests
import json
import re
from bs4 import BeautifulSoup as bs
import time
#from linebot import LineBotApi
#from linebot.models import TextSendMessage
"""
輸入範例：D apple 或是 d apple
"""
def searchdictionary(count,inputlist):
	firstcommand=0
	searchstring=''
	for i in inputlist:
        #為了跳過第一個D/d		
        if firstcommand==0:
			firstcommand+=1
			continue
		else:
			searchstring += i+' '
	res = requests.get('https://tw.dictionary.yahoo.com/dictionary?p='+searchstring)
	print res.status_code
	soup = bs(res.content,'html.parser')
	ttt = soup.find_all('div',attrs={'class':'dd algo explain mt-20 lst DictionaryResults'})

	if len(ttt)==0:
		outputstring="查詢失敗"
	else:
        outputstring = ttt[0].text.encode('utf-8')
		#count是作為最後回傳的時候要判斷用的一個(開關)
        count=1
	#mat同樣是作為最後回傳的時候要判斷用的一個(開關)
	mat=[]
	return count,mat,outputstring
"""
透過google API取得sheet資料。
"""
def gettotal():
	
	apikey={Youe API KEY}
	getvalueurl='https://sheets.googleapis.com/v4/spreadsheets/{Your sheet ID}/values/sheet1!A:C?key=%s' % (apikey)
	res = requests.get(getvalueurl)
	data = res.content
	
	jsondata = json.loads(data)
	values = jsondata['values']
    #開關	
    count=0
    #以下3個變數作為加總的時候用的 	
    cash=0
	credit=0
	other=0
	for i in values:
		if count==0:
			count+=1
			continue
		else:
			#1代表cash
			if i[2]=='cash':
				cash+=int(i[1])
			#2代表credit
			elif i[2]=='credit':
				credit+=int(i[1])
			else:
				other+=int(i[1])
	outputvalue = 'Cash:{}, Credit:{}, \nOther:{}, Total:{}'.format(cash,credit,other,cash+credit+other)
	return outputvalue
"""
填寫google表單，進行記帳的動作
範例：1000,1,note 第1個參數為金額，第2個為種類：1表示cash、2表示credit、其他為other，第3個參數為備註。
"""
def editRecord(accountlist):
	actionurl = 'https://docs.google.com/forms/d/e/{sheet ID}/formResponse'
	formurl = 'https://docs.google.com/forms/d/e/{sheet ID}/viewform'
	r = requests.Session()
	res = r.get(formurl)
	soup = bs(res.content,'html.parser')
	inputlist = soup.find_all('input')
	textarealist = soup.find_all('textarea')
	test={}
	for i in inputlist:
		#print i['aria-label']+':'+i['value']
		try:
			if i['aria-label']=='金額'.decode("utf-8"):
				test.update({i['name']:accountlist[0]})
			elif i['aria-label']=='種類'.decode('utf-8'):
				if accountlist[1]=='1':
					inputkind = 'cash'
				elif accountlist[1]=='2':
					inputkind = 'credit'
				else:
					inputkind = 'other'
				test.update({i['name']:inputkind})
		except:
			test.update({i['name']:i['value']})
    #因為表單當中的備註欄位不是必填欄位，所以在記帳的時候，這個部分可有可無，所以需要判斷，有輸入的話才填入。		
    if len(accountlist)==3:	
		for j in textarealist:
			#print j['name']+':'+j['value']
			test.update({j['name']:accountlist[2]})
	res = r.post(actionurl,data=test)
	return res.status_code

app = Flask(__name__)

@app.route("/",methods=["GET"])
def index():
	return "hello world",200
	
@app.route("/callback",methods=["POST"])
def index2():
	temp = request.get_json()
	for i in temp['events']:
		ttt =  i['replyToken']
		print i['source']['userId']
		if i['message']['type']=='text':
			msg = i['message']['text']
		replyapi(ttt,msg)
	return "hello world",200
	
def replyapi(accesstoken,msg):
	channeltoken={Line channel Token}

	t = msg.encode('utf-8')
	inputlist = t.split(' ')
	count=0
	mat=[]
	if len(inputlist)>1:
		if inputlist[0]=='d' or inputlist[0]=='D':
			count,mat,outputstring = searchdictionary(count,inputlist)
	else:
		accountlist = inputlist[0].split(',')
		if inputlist[0]=='l' or inputlist[0]=='L':
			#進行記錄查詢的動作
			outputstring = gettotal()
			count=1
		elif len(accountlist)>1:
			#進行記帳的動作
			if editRecord(accountlist)==200:
				time.sleep(1)
				outputstring = '紀錄成功'
				count=1
			else:
				outputstring = '紀錄失敗'
				count=1
			#print 'hello'
		else:
			pat = re.compile(r".*(掰掰).*")
			print type(msg)
			mat = pat.findall(t)
	
	if len(mat)==0:
		if count==1:
			data={
			'replyToken':accesstoken,
			'messages':[{'type':'text','text':outputstring}]
			}
		else:
			data={
			'replyToken':accesstoken,
			'messages':[{'type':'text','text':'朕知道了'},{'type':'text','text':'可以退下了'}]
			}
	else:
		data={
		'replyToken':accesstoken,
		'messages':[{'type':'text','text':'慢走不送'}]
		}
	
	
	
	headers={
		'Content-Type':'application/json',
		'Authorization':'Bearer '+channeltoken
	}
	urladdress = 'https://api.line.me/v2/bot/message/reply'
	datajson = json.dumps(data)
	#print type(datajson)
	res = requests.post(urladdress,headers=headers,data=datajson)
	print res.status_code
	

if __name__=='__main__':
	app.run()