from serenitas.analytics.dates import prev_business_day
from serenitas.utils.exchange import ExchangeMessage
from serenitas.utils.db import dbconn
import logging
import argparse
import datetime
from .sma import IsoselSMA, BowdstSMA
from .cash import NTCashReport, UMBCashReport, BNYCashReport
from .admin import AccruedReport, AllReport
from .wires import BowdstWire, NTWire
from .custodians import upload_to_custodian

logger = logging.getLogger(__name__)

parser = argparse.ArgumentParser()

parser.add_argument(
    "-s",
    "--sma_positions",
    action="store_true",
    help="upload sma positions to hedgemark",
)
parser.add_argument(
    "-c", "--cash_reports", action="store_true", help="process cash reports to database"
)
parser.add_argument(
    "-i",
    "--isosel_reports",
    action="store_true",
    help="download isosel reports to database",
)
parser.add_argument(
    "-w", "--wire_reports", action="store_true", help="download wires to db"
)
parser.add_argument(
    "-sc",
    "--send_to_custodians",
    action="store_true",
    help="upload trade files to notify custodians",
)
parser.add_argument(
    "-n", "--no-upload", action="store_true", help="do not upload, just create files"
)
parser.add_argument(
    "date",
    nargs="?",
    type=datetime.date.fromisoformat,
    default=datetime.date.today(),
)

args = parser.parse_args()

cob = date = prev_business_day(args.date)

if args.sma_positions:
    for sma_cls in (
        IsoselSMA,
        BowdstSMA,
    ):
        sma = sma_cls(cob)
        try:
            sma.email_positions()
        except ValueError as e:
            logger.warning(e)

if args.cash_reports:
    for report_cls in (
        NTCashReport,
        UMBCashReport,
        BNYCashReport,
    ):
        report = report_cls(args.date)
        try:
            report.to_db()
        except ValueError as e:
            logger.warning(e)

if args.isosel_reports:
    for report_cls in (
        AccruedReport,
        AllReport,
    ):
        try:
            report_cls.to_db(cob)
        except ValueError as e:
            logger.info(e)

if args.wire_reports:
    for wire_report in (BowdstWire, NTWire):
        try:
            wire_report.to_db(args.date)
        except ValueError as e:
            logger.info(e)

if args.send_to_custodians:
    conn = dbconn("dawndb")
    em = ExchangeMessage()
    for account in ("UMB", "BBH"):
        try:
            upload_to_custodian(account, args.date, conn, not args.no_upload, em)
        except ValueError as e:
            logger.info(e)