main.py 6.6 KB

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