from collections import OrderedDict

# Fields in order returned by the query
QUERY_FIELDS = [
    "txn_id",
    "collaboratorid",
    "contract_id",
    "statement_period_id",
    "statement_period_name",
    "transaction_type",
    "transactiontypedesc",
    "display_upc",
    "project_code",
    "product_code",
    "isrc",
    "cd_id",
    "track_id",
    "trackname",
    "track_artists",
    "quantity",
    "account_payee_currency",
    "track_count",
    "ratetype",
    "countryname",
    "storename",
    "labelname",
    "productname",
    "artistname",
    "royalty_basis",
    "splitrate_sum",
    "collaborator_split",
    "collaborator_share",
]

# Mapping of query fields to column names in the order they should appear in the report
REPORT_FIELDS = OrderedDict(
    [
        ("statement_period_name", "Statement Period"),
        ("countryname", "Country"),
        ("storename", "Store"),
        ("labelname", "Label"),
        ("artistname", "Artist"),
        ("productname", "Product"),
        ("display_upc", "Display UPC"),
        ("project_code", "Project Code"),
        ("product_code", "Product Code"),
        ("trackname", "Track"),
        ("track_artists", "Track Artists"),
        ("isrc", "ISRC"),
        ("quantity", "Quantity"),
        ("transaction_type", "Transaction Type"),
        ("transactiontypedesc", "Transaction Type Description"),
        ("royalty_basis", "Royalty Basis"),
        ("collaborator_split", "Collaborator Split"),
        ("collaborator_share", "Collaborator Share"),
        ("account_payee_currency", "Currency"),
    ]
)