util.py

import pandas

from process_report.invoices import (
    invoice,
    pi_specific_invoice,
    prepay_credits_snapshot,
    NERC_total_invoice,
)

from process_report.processors import (
    coldfront_fetch_processor,
    validate_pi_alias_processor,
    lenovo_processor,
    validate_billable_pi_processor,
    new_pi_credit_processor,
    bu_subsidy_processor,
    prepayment_processor,
    validate_cluster_name_processor,
)


def new_base_invoice(
    name="",
    invoice_month="0000-00",
    data=None,
):
    if data is None:
        data = pandas.DataFrame()
    return invoice.Invoice(invoice_month, data, name)


def new_pi_specific_invoice(
    name="",
    invoice_month="0000-00",
    data=None,
):
    if data is None:
        data = pandas.DataFrame()
    return pi_specific_invoice.PIInvoice(
        invoice_month,
        data,
        name,
    )


def new_nerc_total_invoice(
    name="",
    invoice_month="0000-00",
    data=None,
):
    if data is None:
        data = pandas.DataFrame()
    return NERC_total_invoice.NERCTotalInvoice(
        invoice_month,
        data,
        name,
    )


def new_coldfront_fetch_processor(
    name="",
    invoice_month="0000-00",
    data=None,
    nonbillable_projects=None,
    coldfront_data_filepaths=(),
):
    if data is None:
        data = pandas.DataFrame()
    if nonbillable_projects is None:
        nonbillable_projects = pandas.DataFrame(
            columns=["Project Name", "Cluster", "Is Timed", "Is Billable Override"]
        )
    return coldfront_fetch_processor.ColdfrontFetchProcessor(
        invoice_month,
        data,
        name,
        nonbillable_projects,
        coldfront_data_filepaths,
    )


def new_validate_pi_alias_processor(
    name="", invoice_month="0000-00", data=None, alias_map=None
):
    if data is None:
        data = pandas.DataFrame()
    if alias_map is None:
        alias_map = {}
    return validate_pi_alias_processor.ValidatePIAliasProcessor(
        invoice_month, data, name, alias_map
    )


def new_lenovo_processor(
    name="", invoice_month="0000-00", data=None, su_charge_info=None
):
    if data is None:
        data = pandas.DataFrame()
    if su_charge_info is None:
        su_charge_info = {}
    return lenovo_processor.LenovoProcessor(invoice_month, data, name, su_charge_info)


def new_validate_billable_pi_processor(
    name="",
    invoice_month="0000-00",
    data=None,
    nonbillable_pis=None,
    nonbillable_projects=None,
):
    if data is None:
        data = pandas.DataFrame()
    if nonbillable_pis is None:
        nonbillable_pis = []
    if nonbillable_projects is None:
        nonbillable_projects = pandas.DataFrame(
            columns=["Project Name", "Cluster", "Is Timed", "Is Billable Override"]
        )

    return validate_billable_pi_processor.ValidateBillablePIsProcessor(
        invoice_month,
        data,
        name,
        nonbillable_pis,
        nonbillable_projects,
    )


def new_new_pi_credit_processor(
    name="",
    invoice_month="0000-00",
    data=None,
    old_pi_filepath="",
    credit_amount=1000,
    limit_new_pi_credit_to_partners=False,
    upload_to_s3=False,
):
    if data is None:
        data = pandas.DataFrame()
    return new_pi_credit_processor.NewPICreditProcessor(
        invoice_month,
        data,
        name,
        old_pi_filepath,
        credit_amount,
        limit_new_pi_credit_to_partners,
        upload_to_s3,
    )


def new_bu_subsidy_processor(
    name="",
    invoice_month="0000-00",
    data=None,
    subsidy_amount=0,
):
    if data is None:
        data = pandas.DataFrame()
    return bu_subsidy_processor.BUSubsidyProcessor(
        invoice_month, data, name, subsidy_amount
    )


def new_prepayment_processor(
    name="",
    invoice_month="0000-00",
    data=None,
    prepay_credits=None,
    prepay_debits_filepath="",
    prepay_projects=None,
    prepay_contacts=None,
    upload_to_s3=False,
):
    if prepay_credits is None:
        prepay_credits = pandas.DataFrame()
    if prepay_projects is None:
        prepay_projects = pandas.DataFrame()
    if prepay_contacts is None:
        prepay_contacts = pandas.DataFrame()
    return prepayment_processor.PrepaymentProcessor(
        invoice_month,
        data,
        name,
        prepay_credits,
        prepay_projects,
        prepay_contacts,
        prepay_debits_filepath,
        upload_to_s3,
    )


def new_prepay_credits_snapshot(
    name="",
    invoice_month="0000-00",
    data=None,
    prepay_credits=None,
    prepay_contacts=None,
):
    return prepay_credits_snapshot.PrepayCreditsSnapshot(
        invoice_month, data, name, prepay_credits, prepay_contacts
    )


def new_validate_cluster_name_processor(
    name="",
    invoice_month="0000-00",
    data=None,
):
    return validate_cluster_name_processor.ValidateClusterNameProcessor(
        invoice_month, data, name
    )