#!/usr/bin/env python

from __future__ import print_function

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

import pagure.config

from pagure.lib import model, create_session

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]]

    SESSION = create_session(pagure.config.config['DB_URL'])

    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)