Differences

This shows you the differences between two versions of the page.

Link to this comparison view

Both sides previous revision Previous revision
linux:apache:sendmail_python [2017/02/20 15:52]
tomsa [Testing] 		linux:apache:sendmail_python [2017/02/21 14:27] (current)
tomsa [Code]
Line 59: Line 59:
 time_scope = 60 time_scope = 60
    
 +notification_mail = """​To:​ firma@starlab.cz
 +Subject: Alert! Too many outgoing emails from server {server_name}
 +
 +Safe sendmail script started filtering messages on server {server_name}. Threshold reached.
 +"""​.format(server_name='​webhosting.starlab.cz'​)
 +
    
 def check_db(): def check_db():
Line 66: Line 72:
         is_table = len(c.fetchall())         is_table = len(c.fetchall())
         if is_table == 0:         if is_table == 0:
-            c.execute("​CREATE TABLE mails (date text, address_to text)"​)+            c.execute("​CREATE TABLE mails (date text, address_to text, spamcount integer)")
             conn.commit()             conn.commit()
     return True     return True
Line 97: Line 103:
     logger.debug("​Datetime:​ %s" % timestamp)     logger.debug("​Datetime:​ %s" % timestamp)
     past = timestamp - dt.timedelta(minutes=time_scope)     past = timestamp - dt.timedelta(minutes=time_scope)
-    ​with sqlite3.connect(db_file) as conn:+        ​with sqlite3.connect(db_file) as conn:
         c = conn.cursor()         c = conn.cursor()
-        c.execute('​INSERT INTO mails VALUES(?,?)', (timestamp, address)) +        c.execute('​SELECT spamcount FROM mails ORDER BY date DESC LIMIT 1') 
-        ​conn.commit()+        ​last_spam_count = c.fetchone()[0] 
 +        if last_spam_count is None: 
 +            last_spam_count = 0 
         c.execute('​SELECT COUNT(*) FROM mails where date > ?', (past,))         c.execute('​SELECT COUNT(*) FROM mails where date > ?', (past,))
-        count_since_past = c.fetchone()[0]+        count_since_past = c.fetchone()[0] ​+ 1
         logger.debug("​Count since %s %d" % (timestamp, count_since_past))         logger.debug("​Count since %s %d" % (timestamp, count_since_past))
 +        c.execute('​INSERT INTO mails VALUES(?,?,?​)',​ (timestamp, address, count_since_past))
 +        conn.commit()
         if count_since_past < threshold:         if count_since_past < threshold:
             if send_mail(mail):​             if send_mail(mail):​
Line 110: Line 121:
                 sys.exit(2)                 sys.exit(2)
         else:         else:
 +            if last_spam_count < threshold:
 +                send_mail(notification_mail)
             # logger.info("​Parent:​ {}"​.format(parent_process.cmdline())) ​               # uncomment to get parent'​s cmdline             # logger.info("​Parent:​ {}"​.format(parent_process.cmdline())) ​               # uncomment to get parent'​s cmdline
             # logger.info("​Grandparent:​ {}"​.format(grandparent_process.cmdline())) ​     # uncomment to get grandparent'​s cmdline             # logger.info("​Grandparent:​ {}"​.format(grandparent_process.cmdline())) ​     # uncomment to get grandparent'​s cmdline
Line 115: Line 128:
             logger.warning("​Unable to send mail to address %s - quota exceeded! (%d mails sent in last %d minutes - limit %s)" % (address, count_since_past,​ time_scope, threshold))             logger.warning("​Unable to send mail to address %s - quota exceeded! (%d mails sent in last %d minutes - limit %s)" % (address, count_since_past,​ time_scope, threshold))
             sys.exit(1)             sys.exit(1)
- 
  
 </​code>​ </​code>​
  
 
linux/apache/sendmail_python.1487602355.txt.gz · Last modified: 2017/02/20 15:52 by tomsa