如何终止调用GDAL函数的Python QThread线程？

这个问题我之前也踩过坑——QThread的terminate()方法在面对GDAL这类底层C/C++实现的阻塞操作时完全不管用。原因很简单：terminate()只能终止Python层面的线程执行，但GDAL的核心函数（比如BuildVRT、Translate）是在C级运行时里阻塞的，Python线程的终止信号根本穿透不到那里去。

正确的思路是用协作式终止，结合GDAL自带的进度回调机制来实现中途停止，具体步骤如下：

1. 添加线程安全的终止标志位 #

给MapThread类加一个布尔标志，用来通知GDAL操作需要停止：

class MapThread(QThread):
    print_log = Signal(str)
    show_progress = Signal(int)
    change_button_caption = Signal(str)
    
    def __init__(self, path, tab_filelist, targetpath):
        QThread.__init__(self)
        self.path = path
        self.tab_filelist = tab_filelist
        self.targetpath = targetpath
        self.is_running = True  # 终止控制标志

2. 实现GDAL进度回调函数 #

GDAL的大部分耗时操作都支持传入进度回调，我们可以在回调里检查终止标志，如果需要停止就返回非0值，让GDAL主动中断当前操作：

def gdal_progress_callback(self, complete, message, user_data):
    # 将GDAL返回的进度（0-1）转为百分比同步到UI
    self.show_progress.emit(int(complete * 100))
    # 检查终止标志，返回1告诉GDAL停止操作
    if not self.is_running:
        return 1
    return 0

3. 修改run()方法，整合回调与中断处理 #

把原有的GDAL调用改成带回调的版本，同时添加异常捕获，处理操作被中断的情况：

def run(self):
    self.change_button_caption.emit('停止')
    self.print_log.emit('开始处理影像数据...')
    self.is_running = True  # 重置标志位
    
    try:
        from osgeo import gdal
        from osgeo import osr
        gdal.UseExceptions()  # 让GDAL抛出Python异常，方便捕获中断信号
        
        # 打开源数据集
        d = gdal.Open(self.path)
        if not d:
            self.print_log.emit('无法打开源影像文件')
            return
        
        proj = osr.SpatialReference(wkt=d.GetProjection())
        epsg_code = proj.GetAttrValue('AUTHORITY', 1)
        self.print_log.emit(f'影像数据的EPSG代码: {epsg_code}')
        
        # 生成虚拟栅格（绑定进度回调）
        filename = self.tab_filelist[0].rsplit('\\', 1)[1].rsplit('.', 1)[0]
        path2 = f"{self.targetpath}\\{filename}.vrt"
        vrt_options = gdal.BuildVRTOptions(
            resampleAlg='cubic', 
            outputSRS=proj,
            callback=self.gdal_progress_callback,
            callback_data=None
        )
        self.print_log.emit('开始生成VRT文件...')
        gdal.BuildVRT(path2, self.tab_filelist, options=vrt_options)
        if not self.is_running:
            self.print_log.emit('VRT生成已取消')
            return
        self.print_log.emit('VRT文件已生成')
        
        # 生成MB Tiles（绑定进度回调）
        pathout = f"{self.targetpath}\\{filename}.mbtiles"
        creation_options = ["TILE_FORMAT=JPEG", "QUALITY=90"]
        src_ds = gdal.Open(path2)
        if not src_ds:
            self.print_log.emit('无法打开VRT文件')
            return
        
        self.print_log.emit('开始生成MB Tiles...')
        translate_options = gdal.TranslateOptions(
            format='mbtiles',
            creationOptions=creation_options,
            callback=self.gdal_progress_callback,
            callback_data=None
        )
        gdal.Translate(pathout, src_ds, options=translate_options)
        src_ds = None  # 及时释放资源
        if not self.is_running:
            self.print_log.emit('MB Tiles生成已取消')
            return
        self.print_log.emit('MB Tiles已生成')
        
        # 生成概视图（绑定进度回调）
        self.print_log.emit('开始生成概视图...')
        Image = gdal.Open(pathout, 1)
        if not Image:
            self.print_log.emit('无法打开MB Tiles文件')
            return
        
        gdal.SetConfigOption("COMPRESS_OVERVIEW", "DEFLATE")
        Image.BuildOverviews(
            "AVERAGE", 
            [2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048],
            callback=self.gdal_progress_callback,
            callback_data=None
        )
        Image = None  # 及时释放资源
        if not self.is_running:
            self.print_log.emit('概视图生成已取消')
            return
        self.print_log.emit('概视图计算完成')
        self.show_progress.emit(100)
        
    except Exception as err:
        if self.is_running:
            self.print_log.emit(f'处理出错: {str(err)}')
        else:
            self.print_log.emit('操作已取消')
    finally:
        # 无论成功/失败/中断，都重置按钮状态
        self.change_button_caption.emit('开始')
        self.is_running = False

4. 重构terminateThread()方法，移除危险的terminate()调用 #

现在不需要强制终止线程了，只需要设置标志位即可：

def terminateThread(self):
    self.is_running = False
    self.change_button_caption.emit('开始')
    self.print_log.emit('操作已取消')
    # 无需强制重置进度，让回调自然更新最后状态即可

关键注意事项 #

• 永远别用QThread.terminate()：这个方法是强制终止线程，会导致资源泄漏、数据损坏，Qt官方文档明确不推荐使用。
• 必须启用GDAL异常：调用gdal.UseExceptions()才能让GDAL在被中断时抛出Python异常，否则只会返回错误码，你无法及时感知操作被中断。
• 及时释放GDAL数据集：所有用gdal.Open()打开的数据集，使用后要设置为None，避免内存泄漏，尤其是在操作被中断时。

这样修改后，点击停止按钮时，线程会通过标志位通知GDAL回调函数，GDAL会安全中断当前操作，线程也能正常退出，不会出现无法终止的情况。

内容的提问来源于stack exchange，提问作者Myrkjartan