from cerbos.sdk.model import Principal

# Define the principals we are going to use in the demo.

Michelle = Principal(
    id="Michelle",
    roles={"user"},
    attr={
        "type": "human",
        "apps": ["Abacus"],
        "tenants": {
            "*": {
                "account_id": "*",
                "attachments": {
                    "Abacus_contract_admin": {
                        "role": "Abacus_contract_admin"
                    }
                }
            }
        }
    }
)

Elaine = Principal(
    id="Elaine",
    roles={"user"},
    attr={
        "type": "human",
        "apps": ["Abacus"],
        "tenants": {
            "*": {
                "account_id": "*",
                "attachments": {
                    "Abacus_royalties": {
                        "role": "Abacus_royalties"
                    }
                }
            }
        }
    }
)

Oliver = Principal(
    id="Oliver",
    roles={"user"},
    attr={
        "type": "human",
        "apps": ["Documents", "Content"],
        "tenants": {
            "999": {
                "account_id": "999",
                "attachments": {
                    "Documents_payee_management": {
                        "role": "Documents_payee_management"
                    },
                    "Content_label_artist_user": {
                        "role": "Content_label_artist_user",
                        "content_types": ["digital_audio"]
                    },
                }
            }
        }
    }
)

Conrad = Principal(
    id="Conrad",
    roles={"user"},
    attr={
        "type": "human",
        "apps": ["Content"],
        "tenants": {
            "*": {
                "account_id": "*",
                "attachments": {
                    "Content_distribution_reviewer": {
                        "role": "Content_distribution_reviewer",
                        "content_types": ["digital_audio"]
                    }
                }
            }
        }
    }
)

Paul = Principal(
    id="Paul",
    roles={"user"},
    attr={
        "type": "human",
        "apps": ["Content"],
        "tenants": {
            "888": {
                "account_id": "888",
                "attachments": {
                    "Content_label_artist_user": {
                        "role": "Content_label_artist_user",
                        "content_types": ["digital_audio", "sr_performance"]
                    }
                }
            }
        }
    }
)

Allan = Principal(
    id="Allan",
    roles={"user"},
    attr={
        "type": "human",
        "profiles": ["Content"],
        "tenants": {
            "888": {
                "account_id": "888",
                "attachments": {
                    "Content_label_artist_user": {
                        "role": "Content_label_artist_user",
                        "content_types": ["sr_performance"]
                    }
                }
            },
            "222": {
                "account_id": "222",
                "attachments": {
                    "Content_label_artist_user": {
                        "role": "Content_label_artist_user",
                        "content_types": ["digital_audio"]
                    }
                }
            },
        }
    }
)


Nate = Principal(
    id="Nate",
    roles={"user"},
    attr={
        "type": "machine",
        "apps": ["Insights"],
        "tenants": {
            "888": {
                "account_id": "888",
                "attachments": {
                    "Insights_analytics": {
                        "role": "Insights_analytics",
                        "content_types": ["digital_audio", "sr_performance"]
                    }
                }
            }
        }
    }
)

PRINCIPALS = [
    Michelle,
    Elaine,
    Oliver,
    Conrad,
    Paul,
    Allan,
    Nate,
]