56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
import argparse
|
|
from datetime import datetime, timedelta
|
|
|
|
from sqlalchemy.exc import SQLAlchemyError
|
|
|
|
|
|
if 'PAGURE_CONFIG' not in os.environ \
|
|
and os.path.exists('/etc/pagure/pagure.cfg'):
|
|
print 'Using configuration file `/etc/pagure/pagure.cfg`'
|
|
os.environ['PAGURE_CONFIG'] = '/etc/pagure/pagure.cfg'
|
|
|
|
import pagure
|
|
from pagure import SESSION
|
|
from pagure.lib import model
|
|
|
|
|
|
def main(debug=False):
|
|
''' The function that actually sends the email
|
|
in case the expiration date is near'''
|
|
|
|
current_time = datetime.utcnow()
|
|
day_diff_for_mail = [5, 3, 1]
|
|
email_dates = [email_day.date() for email_day in \
|
|
[current_time + timedelta(days=i) for i in day_diff_for_mail]]
|
|
|
|
tokens = SESSION.query(model.Token).all()
|
|
|
|
for token in tokens:
|
|
if token.expiration.date() in email_dates:
|
|
user = token.user
|
|
user_email = user.default_email
|
|
project = token.project
|
|
days_left = token.expiration.day - datetime.utcnow().day
|
|
subject = 'Pagure API key expiration date is near!'
|
|
text = '''Hi %s, \nYour Pagure API key for the project %s will expire
|
|
in %s day(s). Please get a new key for non-interrupted service. \n
|
|
Thanks, \nYour Pagure Admin. ''' % (user.fullname, project.name, days_left)
|
|
msg = pagure.lib.notify.send_email(text, subject, user_email)
|
|
if debug:
|
|
print 'Sent mail to %s' % user.fullname
|
|
if debug:
|
|
print 'Done'
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
description='Script to send email before the api token expires')
|
|
parser.add_argument(
|
|
'--debug', dest='debug', action='store_true', default=False,
|
|
help='Print the debugging output')
|
|
args = parser.parse_args()
|
|
main(debug=args.debug)
|