def test_check_config(self):
"""
Test check_config.
"""
self.assertFalse(common.check_config(self.cred1))
self.assertFalse(common.check_config(self.cred2))
self.assertTrue(common.check_config(self.cred3))
def __init__(self, cred, debug=False):
"""Init logger params.
Args:
modulename (str): Script module name
cred (dict): AWS user_email, access_key, secret_key
"""
self.logger = logger.SecureTeaLogger(
self.modulename,
debug
)
self.enabled = common.check_config(cred)
if not self.enabled:
self.logger.log(
"Credentials not present, please set AWS SES config at ~/.securetea/securetea.conf ",
logtype="error"
)
self.user_email = cred['aws_email']
self.access_key = cred['aws_access_key']
self.secret_key = cred['aws_secret_key']
self.email_obj = Email(self.user_email,
"secureTea Security Alert!",
self.access_key,
self.secret_key)
def __init__(self, cred, debug):
"""Init logger params.
Args:
-----
modulename (str): secureTeaTwilio
cred (dict): Twilio credentials
"""
self.logger = logger.SecureTeaLogger(
self.modulename,
debug
)
self.enabled = common.check_config(cred)
if not self.enabled:
self.logger.log(
"Credentials not present, please set Twilio config at ~/.securetea/securetea.conf ",
logtype="error"
)
self.account_sid = cred['twilio_sid']
self.account_token = cred['twilio_token']
self.twilio_from = cred['twilio_from']
self.twilio_to = cred['twilio_to']
self.client = Client(self.account_sid, self.account_token)
def __init__(self, cred, debug=False):
"""Init AWS SES params.
Args:
debug (bool): Log on terminal or not
cred (dict): AWS SES credentials
"""
# Initialize logger
self.logger = logger.SecureTeaLogger(
self.modulename,
debug
)
self.enabled = common.check_config(cred)
if not self.enabled:
self.logger.log(
"Credentials not present, please set AWS SES config at ~/.securetea/securetea.conf ",
logtype="error"
)
self.user_email = cred['aws_email']
self.access_key = cred['aws_access_key']
self.secret_key = cred['aws_secret_key']
self.email_obj = helper_email.Email(self.user_email,
"secureTea Security Alert!",
self.access_key,
self.secret_key)
def __init__(self, cred, debug=False):
"""Init params.
Args:
cred (dict): GMAIL credentials
debug (bool): Run in debug mode or not
"""
# Initialize logger
self.logger = logger.SecureTeaLogger(__name__, debug=debug)
self.enabled = common.check_config(cred)
if not self.enabled:
self.logger.log(
"Credentials not present, please set GMAIL"
"config at ~/.securetea/securetea.conf",
logtype="error")
sys.exit(0)
# Setup credentials
self.sender_email = cred["sender_email"]
self._PASSWORD = cred["password"]
self.to_email = cred["to_email"]
# GMAIL endpoint variables
self._GMAIL_URL = "smtp.gmail.com"
self._GMAIL_PORT = 587
# Setup login credentials
self.setup()
def __init__(self, cred, debug):
"""Init logger params.
Args:
modulename (str): Script module name
cred (dict): Telegram user_id
"""
self.logger = logger.SecureTeaLogger(self.modulename, debug)
self.enabled = common.check_config(cred)
if not self.enabled:
self.logger.log(
"Credentials not present, please set Telegram config at ~/.securetea/securetea.conf ",
logtype="error")
self.token = cred['token']
self.user_id = cred['user_id']
def __init__(self, cred, debug):
"""Init logger params.
Args:
modulename (str): Script module name
cred (dict): Slack user_id, token
"""
self.logger = logger.SecureTeaLogger(self.modulename, debug)
self.enabled = common.check_config(cred)
if not self.enabled:
self.logger.log(
"Credentials not present, please set Slack config at ~/.securetea/securetea.conf ",
logtype="error")
self.slack_token = cred['token']
self.user_id = cred['user_id']
self.slack_channel_open_url = 'https://slack.com/api/im.open'
self.slack_post_message_url = 'https://slack.com/api/chat.postMessage'
self.auth_header = 'Bearer ' + self.slack_token
def __init__(self, cred, debug):
"""Init logger params.
Args:
modulename (str): Script module name
cred (dict): Twitter credentials
username (TYPE): Twitter username to post the message
"""
self.logger = logger.SecureTeaLogger(self.modulename, debug)
self.enabled = common.check_config(cred)
if not self.enabled:
self.logger.log(
"Credentials not set, please set Twitter config at ~/.securetea/securetea.conf ",
logtype="error")
self.baseUrl = "https://api.twitter.com/1.1"
self.auth = OAuth1(cred['api_key'], cred['api_secret_key'],
cred['access_token'], cred['access_token_secret'])
self.id = self.getuserid()
def __init__(self, cred, debug):
"""Init Twitter paramss.
Args:
debug (bool): Log on terminal or not
cred (dict): Twitter credentials
Raises:
None
"""
# Initialize logger
self.logger = logger.SecureTeaLogger(self.modulename, debug)
self.enabled = common.check_config(cred)
self.baseUrl = "https://api.twitter.com/1.1"
self.auth = OAuth1(cred['api_key'], cred['api_secret_key'],
cred['access_token'], cred['access_token_secret'])
if not self.enabled:
self.logger.log(
"Credentials not set, please set Twitter config at ~/.securetea/securetea.conf ",
logtype="error")
else:
self.id = self.getuserid()
def __init__(self, cred, debug):
"""Init Whatsapp params.
Args:
debug (bool): Log on terminal or not
cred (dict): Whatsapp credentials
"""
# Initialize logger
self.logger = logger.SecureTeaLogger(self.modulename, debug)
self.enabled = common.check_config(cred)
if not self.enabled:
self.logger.log(
"Credentials not present, please set Whatsapp config at ./securetea.conf ",
logtype="error")
self.account_sid = cred['whatsapp_sid']
self.account_token = cred['whatsapp_token']
self.whatsapp_from = cred['whatsapp_from']
self.whatsapp_to = cred['whatsapp_to']
self.client = Client(self.account_sid, self.account_token)
