"""  The CountryMapping module performs the necessary CS gymnastics to resolve country codes  """
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

__RCSID__ = "c59810311 (2021-10-01 10:44:25 +0200) Chris Burr <christopher.burr@cern.ch>"

from DIRAC import gConfig, S_OK, S_ERROR


def getCountryMapping(country):
    """Determines the associated country from the country code"""
    mappedCountries = [country]
    while True:
        mappedCountry = gConfig.getValue("/Resources/Countries/%s/AssignedTo" % country, country)
        if mappedCountry == country:
            break
        elif mappedCountry in mappedCountries:
            return S_ERROR("Circular mapping detected for %s" % country)
        else:
            country = mappedCountry
            mappedCountries.append(mappedCountry)
    return S_OK(mappedCountry)


def getCountryMappingTier1(country):
    """Returns the Tier1 site mapped to a country code"""
    res = getCountryMapping(country)
    if not res["OK"]:
        return res
    mappedCountry = res["Value"]
    tier1 = gConfig.getValue("/Resources/Countries/%s/Tier1" % mappedCountry, "")
    if not tier1:
        return S_ERROR("No Tier1 assigned to %s" % mappedCountry)
    return S_OK(tier1)