def get_periodic_alive_reminder_channel_set(channel_name: str,
logger_general: logging.Logger,
redis: Optional[RedisApi],
alerts_log_file: str,
internal_conf:
InternalConfig = InternalConf,
user_conf: UserConfig = UserConf) \
-> ChannelSet:
# Initialise list of channels with default channels
channels = [
_get_console_channel(channel_name, logger_general),
_get_log_channel(alerts_log_file, channel_name, logger_general,
internal_conf)
]
# Initialise backup channel sets with default channels
backup_channels_for_telegram = ChannelSet(channels)
# Add telegram alerts to channel set if they are enabled from config file
if user_conf.telegram_alerts_enabled and \
user_conf.telegram_enabled:
telegram_channel = _get_telegram_channel(channel_name, logger_general,
redis,
backup_channels_for_telegram,
user_conf)
channels.append(telegram_channel)
# Add email alerts to channel set if they are enabled from config file
if user_conf.email_alerts_enabled and \
user_conf.email_enabled:
email_channel = _get_email_channel(channel_name, logger_general, redis,
user_conf)
channels.append(email_channel)
else:
email_channel = None
# Set up email channel as backup channel for telegram and twilio
if email_channel is not None:
backup_channels_for_telegram.add_channel(email_channel)
return ChannelSet(channels)
def get_full_channel_set(channel_name: str, logger_general: logging.Logger,
redis: Optional[RedisApi], alerts_log_file: str,
internal_conf: InternalConfig = InternalConf,
user_conf: UserConfig = UserConf) -> ChannelSet:
# Logger initialisation
logger_alerts = create_logger(alerts_log_file, 'alerts',
internal_conf.logging_level)
# Initialise list of channels with default channels
channels = [
ConsoleChannel(channel_name, logger_general),
LogChannel(channel_name, logger_general, logger_alerts)
]
# Initialise backup channel sets with default channels
backup_channels_for_telegram = ChannelSet(channels)
backup_channels_for_twilio = ChannelSet(channels)
# Add telegram alerts to channel set
if user_conf.telegram_alerts_enabled:
telegram_bot = TelegramBotApi(user_conf.telegram_alerts_bot_token,
user_conf.telegram_alerts_bot_chat_id)
telegram_channel = TelegramChannel(channel_name, logger_general, redis,
telegram_bot,
backup_channels_for_telegram)
channels.append(telegram_channel)
else:
telegram_channel = None
# Add email alerts to channel set
if user_conf.email_alerts_enabled:
email = EmailSender(user_conf.email_smtp, user_conf.email_from)
email_channel = EmailChannel(channel_name, logger_general,
redis, email, user_conf.email_to)
channels.append(email_channel)
else:
email_channel = None
# Add twilio alerts to channel set
if user_conf.twilio_alerts_enabled:
twilio = TwilioApi(user_conf.twilio_account_sid,
user_conf.twilio_auth_token)
twilio_channel = TwilioChannel(channel_name, logger_general, redis,
twilio, user_conf.twilio_phone_number,
user_conf.twilio_dial_numbers,
internal_conf.twiml_instructions_url,
internal_conf.redis_twilio_snooze_key,
backup_channels_for_twilio)
channels.append(twilio_channel)
else:
# noinspection PyUnusedLocal
twilio_channel = None
# Set up email channel as backup channel for telegram and twilio
if email_channel is not None:
backup_channels_for_telegram.add_channel(email_channel)
backup_channels_for_twilio.add_channel(email_channel)
# Set up telegram channel as backup channel for twilio
if telegram_channel is not None:
backup_channels_for_twilio.add_channel(telegram_channel)
return ChannelSet(channels)