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