- [Show page]
- [Old revisions]
- [[unknown link type]]
- []
Differences
This shows you the differences between two versions of the page.
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