下载中国地区的 ASTER GDEM V3 高程数据

download_gdemv3.py

import os
import urllib.request
from concurrent.futures import ThreadPoolExecutor
from typing import List, Tuple

import pandas as pd

# 从控制台获取
SID = ""
UID = ""


df = pd.read_excel("中国经纬度范围.xlsx", header=None)

res: List[Tuple[int, int]] = []

for row in df.itertuples():
    lat = int(row[1])
    for i in range(2, len(row), 2):
        if pd.isna(start := row[i]):
            continue
        end = row[i + 1]
        res.extend((lat, lon) for lon in range(int(start), int(end) + 1))


failed: List[Tuple[int, int]] = []


def download_one(pair: Tuple[int, int]):
    lat, lon = pair
    base = f"ASTGTMV003_N{lat}E{lon:03d}"
    filename = f"{base}.zip"
    uri = (
        f"https://bjdl.gscloud.cn/sources/download/"
        f"aeab8000652a45b38afbb7ff023ddabb/{base}?sid={SID}&uid={UID}"
    )
    if not os.path.exists(filename):
        try:
            urllib.request.urlretrieve(uri, filename)
        except Exception:
            failed.append(pair)


def download_mt():
    from tqdm import tqdm

    with ThreadPoolExecutor(max_workers=5) as executor:
        list(tqdm(executor.map(download_one, res), total=len(res), desc="下载进度: "))

    print(f"\n下载失败: {failed}")


if __name__ == "__main__":
    download_mt()

中国经纬度范围.csv