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)