pagure/files/api_key_expire_mail.py

#!/usr/bin/env python

from __future__ import print_function, absolute_import
import os
import argparse
from datetime import datetime, timedelta

from sqlalchemy.exc import SQLAlchemyError

import pagure.config
import pagure.lib.model as model
import pagure.lib.model_base
import pagure.lib.notify
import pagure.lib.query

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"

_config = pagure.config.reload_config()


def main(check=False, debug=False):
    """ The function that actually sends the email
    in case the expiration date is near"""

    current_time = datetime.utcnow()
    day_diff_for_mail = [10, 5, 1]
    email_dates = [
        email_day.date()
        for email_day in [
            current_time + timedelta(days=i) for i in day_diff_for_mail
        ]
    ]

    session = pagure.lib.model_base.create_session(_config["DB_URL"])
    tokens = session.query(model.Token).all()

    for token in tokens:
        if debug:
            print(token.id, token.expiration.date())
        if token.expiration.date() in email_dates:
            user = token.user
            username = user.fullname or user.username
            user_email = user.default_email
            days_left = (token.expiration - datetime.utcnow()).days
            subject = "Pagure API key expiration date is near!"
            if token.project:
                text = """Hi %s,
Your Pagure API key %s linked to the project %s
will expire in %s day(s).
Please get a new key for non-interrupted service.

Thanks,
Your Pagure Admin. """ % (
                    username,
                    token.description,
                    token.project.fullname,
                    days_left,
                )
            else:
                text = """Hi %s,
Your Pagure API key %s will expire in %s day(s).
Please get a new key for non-interrupted service.

Thanks,
Your Pagure Admin. """ % (
                    username,
                    token.description,
                    days_left,
                )
            if not check:
                msg = pagure.lib.notify.send_email(text, subject, user_email)
            else:
                print(
                    "Sending email to %s (%s) about key: %s"
                    % (username, user_email, token.id)
                )
            if debug:
                print("Sent mail to %s" % username)

    session.remove()
    if debug:
        print("Done")


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Script to send email before the api token expires"
    )
    parser.add_argument(
        "--check",
        dest="check",
        action="store_true",
        default=False,
        help="Print the some output but does not send any email",
    )
    parser.add_argument(
        "--debug",
        dest="debug",
        action="store_true",
        default=False,
        help="Print the debugging output",
    )
    args = parser.parse_args()
    main(debug=args.debug)