import smtplib
from email.mime.text import MIMEText

from stratum import settings

import stratum.logger
log = stratum.logger.get_logger('Notify_Email')

class NOTIFY_EMAIL():

    def notify_start(self):
	if settings.NOTIFY_EMAIL_TO != '':
	    self.send_email(settings.NOTIFY_EMAIL_TO,'Stratum Server Started','Stratum server has started!')
    
    def notify_found_block(self,worker_name):
	if settings.NOTIFY_EMAIL_TO != '':
	    text = '%s on Stratum server found a block!' % worker_name
	    self.send_email(settings.NOTIFY_EMAIL_TO,'Stratum Server Found Block',text)
	    
    def notify_dead_coindaemon(self,worker_name):
	if settings.NOTIFY_EMAIL_TO != '':
	    text = 'Coin Daemon Has Crashed Please Report' % worker_name
	    self.send_email(settings.NOTIFY_EMAIL_TO,'Coin Daemon Crashed!',text)
	    
    def send_email(self,to,subject,message):
	msg = MIMEText(message)
	msg['Subject'] = subject
	msg['From'] = settings.NOTIFY_EMAIL_FROM
	msg['To'] = to
	try:
	    s = smtplib.SMTP(settings.NOTIFY_EMAIL_SERVER)
	    if settings.NOTIFY_EMAIL_USERNAME != '':
		if settings.NOTIFY_EMAIL_USETLS:
		    s.ehlo()
		    s.starttls()
		s.ehlo()
		s.login(settings.NOTIFY_EMAIL_USERNAME, settings.NOTIFY_EMAIL_PASSWORD)
	    s.sendmail(settings.NOTIFY_EMAIL_FROM,to,msg.as_string())
	    s.quit()
	except smtplib.SMTPAuthenticationError as e:
	    log.error('Error sending Email: %s' % e[1])
	except Exception as e:
	    log.error('Error sending Email: %s' % e[0])