from http.cookies import SimpleCookie


def get_datadome_cookies(cookiesStr: str):
    _cookies = SimpleCookie()
    _cookies.load(cookiesStr)
    new_cookies = {k: v.value for k, v in _cookies.items()}
    _new_coolies_str = None
    for key in new_cookies:
        if key == "datadome":
            if _new_coolies_str is None:
                _new_coolies_str = ""
            _new_coolies_str = _new_coolies_str + new_cookies[key]
    return _new_coolies_str


def retain_only_dataome_cookies(cookiesStr: str):
    _cookies = SimpleCookie()
    _cookies.load(cookiesStr)
    new_cookies = {k: v.value for k, v in _cookies.items()}
    new_coolies_str = None
    for key in new_cookies:
        if key == "datadome":
            if new_coolies_str is None:
                new_coolies_str = ""
            new_coolies_str = new_coolies_str + key + "=" + new_cookies[key] + ";"
    return new_coolies_str


def get_app_cookies(cookiesStr: str):
    _cookies = SimpleCookie()
    _cookies.load(cookiesStr)
    new_cookies = {k: v.value for k, v in _cookies.items()}
    new_coolies_str = None
    for key in new_cookies:
        if key == "app.sig":
            if new_coolies_str is None:
                new_coolies_str = ""
            new_coolies_str = new_coolies_str + key + "=" + new_cookies[key] + ";"
    return new_coolies_str


def get_lang_cookies(cookiesStr: str):
    _cookies = SimpleCookie()
    _cookies.load(cookiesStr)
    new_cookies = {k: v.value for k, v in _cookies.items()}
    new_coolies_str = None
    for key in new_cookies:
        if key == "app":
            if new_coolies_str is None:
                new_coolies_str = ""
            new_coolies_str = new_coolies_str + key + "=" + new_cookies[key] + ";"
        if key == "policy":
            if new_coolies_str is None:
                new_coolies_str = ""
            new_coolies_str = new_coolies_str + key + "=" + new_cookies[key] + ";"
        if key == "lang":
            if new_coolies_str is None:
                new_coolies_str = ""
            new_coolies_str = new_coolies_str + key + "=" + new_cookies[key] + ";"
    return new_coolies_str


if __name__ == '__main__':
    print(get_datadome_cookies(
        "datadome=HS7HPdbsFTYfxD60ogUwxjplOuqrveIjcTaucXVIWCko6pRayhXuKxC~44Ky5eNR2wnx4PEd2P2AadewmeXgxi4vn3x_QKdxLb6H1a9MbaMLNwmxfMEjQQ~CK8Y_nspJ;app=eyJmbGFzaCI6e30sImNhY2hlZmxhc2giOltdLCJjc3JmU2VjcmV0IjoiNXhvX3ZFaWc0aURla0dYMHZGTHNJX2p4In0=;policy=accepted;lang=fr;"))
    print(get_app_cookies(
        "datadome=HS7HPdbsFTYfxD60ogUwxjplOuqrveIjcTaucXVIWCko6pRayhXuKxC~44Ky5eNR2wnx4PEd2P2AadewmeXgxi4vn3x_QKdxLb6H1a9MbaMLNwmxfMEjQQ~CK8Y_nspJ;app=eyJmbGFzaCI6e30sImNhY2hlZmxhc2giOltdLCJjc3JmU2VjcmV0IjoiNXhvX3ZFaWc0aURla0dYMHZGTHNJX2p4In0=;policy=accepted;lang=fr;"))
    print(get_lang_cookies(
        "datadome=HS7HPdbsFTYfxD60ogUwxjplOuqrveIjcTaucXVIWCko6pRayhXuKxC~44Ky5eNR2wnx4PEd2P2AadewmeXgxi4vn3x_QKdxLb6H1a9MbaMLNwmxfMEjQQ~CK8Y_nspJ;app=eyJmbGFzaCI6e30sImNhY2hlZmxhc2giOltdLCJjc3JmU2VjcmV0IjoiNXhvX3ZFaWc0aURla0dYMHZGTHNJX2p4In0=;policy=accepted;lang=fr;"))