charles/smssend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
ping 2026-03-21 09:58:35 +08:00
parent 2558c55dbb
commit 2d13b9696b
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
smssend/smssend.py
View File

@ -21,6 +21,7 @@ from sqlor.dbpools import DBPools
from appPublic.uniqueID import getID as uuid from appPublic.uniqueID import getID as uuid
from ahserver.serverenv import ServerEnv from ahserver.serverenv import ServerEnv
from sqlor.dbpools import get_sor_context from sqlor.dbpools import get_sor_context
from appPublic.worker import awaitify
SMS_TEMPLATE_TABLE = { SMS_TEMPLATE_TABLE = {
"summary": [{"name": "sms_template", "primary": "id"}], "summary": [{"name": "sms_template", "primary": "id"}],
@ -84,7 +85,7 @@ class SMSEngine:
required_vars) if val is None] required_vars) if val is None]
raise EnvironmentError(f"Missing required environment variables: {', '.join(missing)}") raise EnvironmentError(f"Missing required environment variables: {', '.join(missing)}")
self.sms_client = self.create_client() self.sms_client = (awaitifyself.create_client())
self.sms_types = {} self.sms_types = {}
def create_client(self): def create_client(self):
@ -155,9 +156,9 @@ class SMSEngine:
await sor.C('sms_record', log) await sor.C('sms_record', log)
return {'status': False, 'msg': msg} return {'status': False, 'msg': msg}
async def generate_sms_code(self, length: int = 6, expire_minutes: int = 5) -> str: async def generate_sms_code(self, length=None, expire_minutes=None, phone=None):
length = int(length) length = int(length) if length is not None else 6
expire_minutes = int(expire_minutes) expire_minutes = int(expire_minutes) if expire_minutes is not None else 5
code = ''.join(random.choices(string.digits, k=length)) code = ''.join(random.choices(string.digits, k=length))
code_id = uuid() code_id = uuid()
expire_time = datetime.datetime.now() + datetime.timedelta(minutes=expire_minutes) expire_time = datetime.datetime.now() + datetime.timedelta(minutes=expire_minutes)