Move endpoints of Chinese iCloud account to icloud.com.cn

picklepete · Jun 11, 2022 · ac0e579 · ac0e579
1 parent 332cc9f
commit ac0e579
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pyicloud/base.py b/pyicloud/base.py
@@ -5,7 +5,7 @@
 import logging
 from requests import Session
 from tempfile import gettempdir
-from os import path, mkdir
+from os import environ, path, mkdir
 from re import match
 import http.cookiejar as cookielib
 import getpass
@@ -201,8 +201,10 @@ class PyiCloudService:
     """
 
     AUTH_ENDPOINT = "https://idmsa.apple.com/appleauth/auth"
-    HOME_ENDPOINT = "https://www.icloud.com"
-    SETUP_ENDPOINT = "https://setup.icloud.com/setup/ws/1"
+
+    icloud_china = environ.get("icloud_china", "1") == "1"
+    HOME_ENDPOINT = f"https://www.icloud.com{icloud_china * '.cn'}"
+    SETUP_ENDPOINT = f"https://setup.icloud.com{icloud_china * '.cn'}/setup/ws/1"
 
     def __init__(
         self,