def test_data_are_kept_upon_closing_database_connection(self):
database = Database(self.conn)
warn_with_culprit = create_warning_with_culprit()
database.insert_warning(warn_with_culprit)
self.conn.close()
self.conn = sqlite3.connect(self.db_file.name)
database = Database(self.conn)
self.assertTrue(database.has_warning(warn_with_culprit))
def main(argc, argv):
args = parse_args(argv)
config = parse_config("config.ini", "config.local.ini")
if config['logging'].getboolean('enabled'):
logging.basicConfig(
filename=config['logging']['log_file'],
level=logging.INFO,
format='%(asctime)s: %(levelname)s: %(message)s'
)
logging.info('{} started with arguments {}'.format(
argv[0], ' '.join(argv[1:])))
try:
if args.file:
text = read_file(args.file)
else:
text = read_stdin()
warnings = parse_warnings(text)
all_warnings_with_culprit = list(map(create_warning_with_culprit, warnings))
with sqlite3.connect(config['db']['path']) as db_conn:
db = Database(db_conn)
# Do not use filter() because we need the filtered warnings right
# away, before db.make_all_warnings_old() is called.
warnings_with_culprit = [w for w in all_warnings_with_culprit
if not db.has_warning(w)]
db.make_all_warnings_old()
server = config['email']['server'] or input('Email server: ')
port = config['email']['port'] or input('Email server port: ')
use_ssl = config['email'].getboolean('use_ssl')
username = config['email']['username'] or input('Email server username: '******'email']['password'] or getpass('Email server password: '******'email']['from'] or input('From address: ')
subject = config['email']['subject']
to_addr = config['email']['to']
reply_to_addr = config['email']['reply_to']
SMTPServer = SMTP_SSL if use_ssl else SMTP
with SMTPServer(server, port) as smtp_server:
smtp_server.login(username, password)
for culprit, warnings in group_by_culprit(warnings_with_culprit):
email = create_email(culprit, warnings, from_addr, subject,
to_addr=to_addr, reply_to_addr=reply_to_addr)
smtp_server.send_message(email)
logging.info('email sent to {}'.format(email['To']))
for w in all_warnings_with_culprit:
db.insert_warning(w)
except Exception as ex:
logging.error('{}: {}'.format(ex.__class__.__name__, str(ex)))
raise
logging.info('{} ended successfully'.format(argv[0]))
return 0