main.py 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204
  1. import uvicorn
  2. import fastapi
  3. from fastapi.middleware.cors import CORSMiddleware
  4. from fastapi.responses import HTMLResponse
  5. from linebot import LineBotApi, WebhookHandler
  6. from linebot.models import (
  7. MessageEvent, TextMessage, TextSendMessage, FollowEvent, TemplateSendMessage, ButtonsTemplate, URITemplateAction,
  8. )
  9. import dataset
  10. import requests
  11. import json
  12. import qrcode
  13. from PIL import Image
  14. import base64, io
  15. from random import randrange
  16. import models
  17. app = fastapi.FastAPI()
  18. app.add_middleware(
  19. CORSMiddleware,
  20. allow_origins=['*'],
  21. allow_credentials=True,
  22. allow_methods=["*"],
  23. allow_headers=["*"],
  24. )
  25. # bot config
  26. line_bot_api = LineBotApi("SJT7VPT4RMQFLcS27jQBy3FcC24gtDrkcwJWZ5Xzqesr5T78LOKudHEJzt0k3b2S7n4KPwf27J7DVz2c8NQ4plSaaQylEeB1cYrfejaE/RPG/lCIQBYe4iBTzo26s4i2PcmT89837per/lTyvhVIKAdB04t89/1O/w1cDnyilFU=")
  27. handler = WebhookHandler("411ae3ef7e766739ed2c2c27b249d010")
  28. # callback event
  29. @app.post("/callback")
  30. async def callback(request: fastapi.Request):
  31. signature = request.headers['X-Line-Signature']
  32. body = await request.body()
  33. handler.handle(body.decode('utf-8'), signature)
  34. return 'OK'
  35. # nft collection api
  36. @app.get("/collection/{userid}")
  37. def collection(userid):
  38. xname = userid
  39. sql = 'SELECT a.* FROM nft a INNER JOIN ( SELECT userid FROM users WHERE userid="'+xname+'" ) b ON b.userid = a.userid'
  40. result = db.query(sql)
  41. rows = {}
  42. i = 0
  43. for row in result:
  44. rows[i] = row
  45. i += 1
  46. return rows
  47. # receive handler
  48. @app.get("/receive/{userid}")
  49. def receive(userid):
  50. if not table.find_one(userid=userid):
  51. return "ERROR: User Not Found"
  52. else:
  53. result = table.find_one(userid=userid)
  54. return {"userid": result['userid'], "base": result['base'], "useraddress": result['useraddress']}
  55. # TODO send handler 發送nft
  56. # @app.get("/collection/{userid}/{hash}")
  57. # def blockchain():
  58. # TODO post blockchain api 上鏈nft
  59. # @app.get("/collection/{userid}/{art}")
  60. # def blockchain():
  61. # TODO get transaction update 取得交易結果
  62. # @app.get("/collection/{userid}")
  63. # def blockchain():
  64. # follow event
  65. @handler.add(FollowEvent)
  66. def handle_follow(event):
  67. # get user id when follow
  68. real_user_id = event.source.user_id
  69. rand_num = str(randrange(99999))
  70. # db connect
  71. db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/arkcard?charset=utf8mb4')
  72. table = db['users']
  73. print(rand_num)
  74. user_id = event.source.user_id + rand_num
  75. result = table.find_one(userid=real_user_id)
  76. if result:
  77. print(real_user_id)
  78. line_bot_api.reply_message(
  79. event.reply_token,
  80. TextSendMessage(text='很高興再見到您!'))
  81. else:
  82. print("n")
  83. # create user account
  84. url = 'https://nft-api-staging.joyso.io/api/v1/accounts'
  85. headers = {'Authorization': 'Basic bmZ0OmMxOTEzOWMzYjM3YjdjZWU3ZmY3OTFiZGU3NzdjZWNl'}
  86. # setup for temp use (unique id)
  87. data = 'uid=' + user_id
  88. r = requests.post(url=url, headers=headers, data=data)
  89. # extract the account address
  90. dict_str = json.loads(r.text)
  91. user_account = dict_str['account']
  92. user_address = user_account['address']
  93. # generate qr code from user id
  94. qr = qrcode.QRCode(
  95. version=1,
  96. box_size=10,
  97. border=5)
  98. qr.add_data(user_address)
  99. qr.make(fit=True)
  100. img_qr = qr.make_image(fill='black', back_color='white')
  101. filename = "qrcode/" + real_user_id + '.png'
  102. img_save = img_qr.save(filename)
  103. # open file
  104. im = Image.open(filename)
  105. # convert
  106. imgByteArr = io.BytesIO()
  107. im.save(imgByteArr, format='PNG')
  108. # decode to ascii
  109. x = imgByteArr.getvalue()
  110. base64.b64encode(x).decode('ascii')
  111. encode = base64.b64encode(x).decode('ascii')
  112. # add to db
  113. data = dict(base64=encode)
  114. result = data['base64']
  115. # table.update(dict(userid=userid, base=result), ['userid'])
  116. # write in sql
  117. data = dict(userid=real_user_id, useraddress=user_address, base=result)
  118. table.insert(data)
  119. line_bot_api.reply_message(
  120. event.reply_token,
  121. TextSendMessage(text='歡迎加入好友'))
  122. # message handler
  123. @handler.add(MessageEvent, message=TextMessage)
  124. def message(event):
  125. if '我要發送' in event.message.text:
  126. button_template_message = ButtonsTemplate(
  127. title=' ',
  128. text='點擊並打開收藏的NFT,可以選擇想要發送的NFT給對方!',
  129. actions=[
  130. URITemplateAction(
  131. label='打開發送頁面',
  132. uri='http://ark.cards/collect.html?' + event.source.user_id),])
  133. line_bot_api.reply_message(
  134. event.reply_token,
  135. TemplateSendMessage(
  136. alt_text="Receive",
  137. template=button_template_message))
  138. elif '我要接收' in event.message.text:
  139. button_template_message = ButtonsTemplate(
  140. title=' ',
  141. text='點擊並打開接收頁面,即可分享接收地址給對方!',
  142. actions=[
  143. URITemplateAction(
  144. label='打開接收頁面',
  145. uri='http://ark.cards/qr-code.html?' + event.source.user_id),])
  146. line_bot_api.reply_message(
  147. event.reply_token,
  148. TemplateSendMessage(
  149. alt_text="Receive",
  150. template=button_template_message))
  151. elif '我的NFT收藏' in event.message.text:
  152. button_template_message = ButtonsTemplate(
  153. title=' ',
  154. text='點擊並打開收藏頁面,就可以查看所有的收藏品哦!',
  155. actions=[
  156. URITemplateAction(
  157. label='打開我的NFT收藏頁面',
  158. uri='http://ark.cards/collect.html?' + event.source.user_id),])
  159. line_bot_api.reply_message(
  160. event.reply_token,
  161. TemplateSendMessage(
  162. alt_text="Receive",
  163. template=button_template_message))
  164. else:
  165. button_template_message = ButtonsTemplate(
  166. title=' ',
  167. text='更多的服務內容,歡迎請上我們的官網!',
  168. actions=[
  169. URITemplateAction(
  170. label='ArkCard的官網',
  171. uri='http://ark.cards'),])
  172. line_bot_api.reply_message(
  173. event.reply_token,
  174. TemplateSendMessage(
  175. alt_text="Receive",
  176. template=button_template_message))
  177. if __name__ == '__main__':
  178. uvicorn.run("main:app", host="0.0.0.0", port=8228, reload=True)