86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
|
# Start an HTTP server to receive SMS forwarding notifications and store them in Redis.
|
|||
|
|
import re
|
|||
|
|
import json
|
|||
|
|
import asyncio
|
|||
|
|
|
|||
|
|
import aioredis
|
|||
|
|
import tornado.web
|
|||
|
|
|
|||
|
|
import config
|
|||
|
|
|
|||
|
|
|
|||
|
|
def extract_verification_code(message) -> str:
|
|||
|
|
"""
|
|||
|
|
Extract verification code of 6 digits from the SMS.
|
|||
|
|
"""
|
|||
|
|
pattern = re.compile(r'\b[0-9]{6}\b')
|
|||
|
|
codes = pattern.findall(message)
|
|||
|
|
return codes[0] if codes and len(codes) > 0 else ""
|
|||
|
|
|
|||
|
|
|
|||
|
|
class RecvSmsNotificationHandler(tornado.web.RequestHandler):
|
|||
|
|
async def get(self):
|
|||
|
|
self.set_status(404)
|
|||
|
|
self.write("404")
|
|||
|
|
|
|||
|
|
async def post(self):
|
|||
|
|
# GitHub address for the SMS forwarding function：https://github.com/pppscn/SmsForwarder
|
|||
|
|
# Document address:：https://gitee.com/pp/SmsForwarder/wikis/pages?sort_id=6040999&doc_id=1821427
|
|||
|
|
# Forwarding channel definition：
|
|||
|
|
# {
|
|||
|
|
# "platform": "xhs",
|
|||
|
|
# "current_number": "138xxxxxxxx",
|
|||
|
|
# "from_number": "[from]",
|
|||
|
|
# "sms_content": "[org_content]",
|
|||
|
|
# "timestamp": "[timestamp]"
|
|||
|
|
# }
|
|||
|
|
|
|||
|
|
# SMS message body：
|
|||
|
|
# {
|
|||
|
|
# 'platform': 'xhs', # or dy
|
|||
|
|
# 'current_number': '138xxxxxxxx',
|
|||
|
|
# 'from_number': '1069421xxx134',
|
|||
|
|
# 'sms_content': '【小红书】您的验证码是: 171959， 3分钟内有效。请勿向他人泄漏。如非本人操作，可忽略本消息。',
|
|||
|
|
# 'timestamp': '1686720601614'
|
|||
|
|
# }
|
|||
|
|
request_body = self.request.body.decode("utf-8")
|
|||
|
|
req_body_dict = json.loads(request_body)
|
|||
|
|
print("recv sms notification and body content: ", req_body_dict)
|
|||
|
|
redis_obj = aioredis.from_url(url=config.redis_db_host, password=config.redis_db_pwd, decode_responses=True)
|
|||
|
|
sms_content = req_body_dict.get("sms_content")
|
|||
|
|
sms_code = extract_verification_code(sms_content)
|
|||
|
|
if sms_code:
|
|||
|
|
# Save the verification code in Redis and set the expiration time to 3 minutes.
|
|||
|
|
# Use Redis string data structure, in the following format:
|
|||
|
|
# xhs_138xxxxxxxx -> 171959
|
|||
|
|
key = f"{req_body_dict.get('platform')}_{req_body_dict.get('current_number')}"
|
|||
|
|
await redis_obj.set(name=key, value=sms_code, ex=60 * 3)
|
|||
|
|
self.set_status(200)
|
|||
|
|
self.write("ok")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Application(tornado.web.Application):
|
|||
|
|
def __init__(self):
|
|||
|
|
handlers = [(r'/', RecvSmsNotificationHandler)]
|
|||
|
|
settings = dict(
|
|||
|
|
gzip=True,
|
|||
|
|
autoescape=None,
|
|||
|
|
autoreload=True
|
|||
|
|
)
|
|||
|
|
super(Application, self).__init__(handlers, **settings)
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def main():
|
|||
|
|
app = Application()
|
|||
|
|
app.listen(port=9435)
|
|||
|
|
print("Recv sms notification app running ...")
|
|||
|
|
shutdown_event = tornado.locks.Event()
|
|||
|
|
await shutdown_event.wait()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
try:
|
|||
|
|
asyncio.run(main())
|
|||
|
|
except KeyboardInterrupt:
|
|||
|
|
pass
|