from .. import Provider as CompanyProvider


class Provider(CompanyProvider):
    formats = (
        "{{last_name}} {{company_suffix}}",
        "{{last_name}} {{last_name}} {{company_suffix}}",
        "{{last_name}}",
    )
    company_prefixes = ("S.C.", "S.S.I.", "A.D.")
    company_suffixes = (
        "SRL",
        "SA",
        "SCA",
        "SNC",
        "SCS",
        "AFJ",
        "ASF",
        "CON",
        "CRL",
        "INC",
        "LOC",
        "OC1",
        "OC2",
        "OC3",
        "PFA",
        "RA",
        "SCS",
        "SPI",
        "URL",
    )

    def company_suffix(self) -> str:
        return self.random_element(self.company_suffixes)