import_enl_stats.py

# example usage: ./bin/instance run import_enl_stats.py --errors-filenam=missing.csv --issue-path=/enl-dx-clean/test-newsletter-issue /Plone

from datetime import datetime
from plone import api
from Products.EasyNewsletter.content.newsletter_issue import ISendStatus
from transaction import commit
from zope.site.hooks import setSite

import argparse
import csv
import logging
import os
import sys


# from Acquisition import aq_inner
# from Products.CMFPlone.utils import safe_unicode


def get_base_parser():
    parser = argparse.ArgumentParser(description=SCRIPTNAME)
    parser.add_argument(
        "plonesite_path",
        default="/Plone",
        # metavar='"/Plone"',
        action="store",
        # dest="plonesite_path",
        type=str,
        nargs="?",
        help="Path to the Plone site",
    )
    parser.add_argument(
        "-b",
        "--commit_batch_size",
        type=int,
        action="store",
        dest="commit_batch_size",
        default=100,
        metavar="N",
        nargs="?",
        help="Do a transaction commit every N items. Default: 100",
    )
    parser.add_argument(
        "--hostname",
        action="store",
        dest="hostname",
        default="nohost",
        nargs="?",
        help="Define the hostname, Plone should use for creating urls. Default: nohost",  # NOQA: E501
    )
    parser.add_argument(
        "--protocol",
        action="store",
        dest="protocol",
        default="http",
        nargs="?",
        help="Define the protocol, Plone should use for creating urls. Default: http",  # NOQA: E501
    )
    parser.add_argument(
        "--quiet", action="store_true", help="Only show errors. Useful for cronjobs.",
    )
    parser.add_argument(
        "--verbose", action="store_true", help="Show debug infos.",
    )
    return parser


def get_logger(name, args):
    log = logging.getLogger(name)
    if args.quiet:
        log.setLevel(logging.ERROR)
    elif args.verbose:
        log.setLevel(logging.DEBUG)
    else:
        log.setLevel(logging.INFO)

    handler = logging.StreamHandler(sys.stdout)
    formatter = logging.Formatter(
        "%(asctime)s %(levelname)s %(name)s: %(message)s", datefmt="%m/%d/%Y %H:%M:%S"
    )
    handler.setFormatter(formatter)
    log.addHandler(handler)
    return log


class BaseScriptWrapper:
    """
    """

    def __init__(self, app, args):
        """
        """
        plonesite_path = args.plonesite_path
        portal = app.unrestrictedTraverse(plonesite_path)  # noqa: F821
        setSite(portal)
        self.portal = portal
        self.request = self.portal.REQUEST
        self.request["PARENTS"] = [self.portal]
        self.commit_batch_size = args.commit_batch_size
        protocol = args.protocol
        hostname = args.hostname
        port = protocol == "http" and "80" or "443"

        self.request.setServerURL(
            protocol=protocol, hostname=hostname, port=port,
        )
        self.request.setVirtualRoot("")
        self.args = args



#################
## customization:
#################


SCRIPTNAME = u"import_enl_stats: "


parser = get_base_parser()

parser.add_argument(
    '--errors-filename',
    action='store',
    dest="errors_filename",
    default="missing.csv",
    type=str,
    help='CSV filename to import sendiong error addresses from.',
)

parser.add_argument(
    '--issue-path',
    action='store',
    dest="issue_path",
    type=str,
    help='ENL issue where we import the stats to.',
)

# remove -c script_name from args before argparse runs:
if "-c" in sys.argv:
    index = sys.argv.index("-c")
    del sys.argv[index]
    del sys.argv[index]

args = parser.parse_args()
log = get_logger(SCRIPTNAME, args)


class ScriptWrapper(BaseScriptWrapper):
    """
    """

    def run(self):
        enl_issue = api.content.get(args.issue_path)
        enl = enl_issue.get_newsletter()
        subscribers = [obj.email for obj in enl.objectValues() if obj.portal_type == "Newsletter Subscriber"]
        status_adapter = ISendStatus(enl_issue)

        with open(args.errors_filename) as f:
            error_lines = f.readlines()
        error_emails = []
        for line in error_lines:
            error_emails.append(line.strip('\n'))

        successful_subscribers = set(subscribers).difference(set(error_emails))
        records = []
        record_status = {'successful': True, 'error': None, 'datetime': datetime.now()}
        for sub in successful_subscribers:
            records.append({"email": sub, "status": record_status})
            print("Add subscriber '{}' as successful to stattistics.".format(sub))
        status_adapter.add_records(records)
        commit()


if __name__ == "__main__":
    script_wrapper = ScriptWrapper(app, args,)
    script_wrapper.run()