mirror of https://github.com/jumpserver/jumpserver
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
|
import os
|
||
|
|
|
||
|
|
from collections import OrderedDict
|
||
|
|
|
||
|
|
from django.conf import settings
|
||
|
|
from django.utils.translation import gettext_lazy as _
|
||
|
|
from django.utils.module_loading import import_string
|
||
|
|
|
||
|
|
from common.utils import get_logger
|
||
|
|
from common.exceptions import JMSException
|
||
|
|
from jumpserver.settings import get_file_md5
|
||
|
|
|
||
|
|
from .base import BaseSMSClient
|
||
|
|
|
||
|
|
|
||
|
|
logger = get_logger(__file__)
|
||
|
|
|
||
|
|
|
||
|
|
custom_sms_method = None
|
||
|
|
SMS_CUSTOM_FILE_MD5 = settings.SMS_CUSTOM_FILE_MD5
|
||
|
|
SMS_CUSTOM_FILE_PATH = os.path.join(settings.PROJECT_DIR, 'data', 'sms', 'main.py')
|
||
|
|
if SMS_CUSTOM_FILE_MD5 == get_file_md5(SMS_CUSTOM_FILE_PATH):
|
||
|
|
try:
|
||
|
|
custom_sms_method_path = 'data.sms.main.send_sms'
|
||
|
|
custom_sms_method = import_string(custom_sms_method_path)
|
||
|
|
except Exception as e:
|
||
|
|
logger.warning('Import custom sms method failed: {}, Maybe not enabled'.format(e))
|
||
|
|
|
||
|
|
|
||
|
|
class CustomFileSMS(BaseSMSClient):
|
||
|
|
@classmethod
|
||
|
|
def new_from_settings(cls):
|
||
|
|
return cls()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def need_pre_check():
|
||
|
|
return False
|
||
|
|
|
||
|
|
def send_sms(self, phone_numbers: list, template_param: OrderedDict, **kwargs):
|
||
|
|
if not callable(custom_sms_method):
|
||
|
|
raise JMSException(_('The custom sms file is invalid'))
|
||
|
|
|
||
|
|
try:
|
||
|
|
logger.info(f'Custom file sms send: phone_numbers={phone_numbers}, param={template_param}')
|
||
|
|
custom_sms_method(phone_numbers, template_param, **kwargs)
|
||
|
|
except Exception as err:
|
||
|
|
raise JMSException(_('SMS sending failed[%s]: %s') % (f"{_('Custom type')}({_('File')})", err))
|
||
|
|
|
||
|
|
|
||
|
|
client = CustomFileSMS
|