百度360必应搜狗淘宝本站头条
当前位置:网站首页 > IT知识 > 正文

Python的with语句:不止文件操作,数据库和多线程也能轻松搞定!

liuian 2025-08-31 04:04 7 浏览

你是不是也遇到过这样的问题:写文件操作时忘记调用close()导致资源泄漏?处理数据库连接时,因为异常没关闭连接让服务器崩溃?别慌!今天咱们聊聊Python里的"资源管家"——with语句,它可不只是用来打开文件那么简单,数据库连接、多线程锁都能轻松拿捏!

什么是with语句?一句话讲明白

with语句就像给资源配了个"智能管家",自动帮你完成"开门-使用-关门"全流程。不管你是正常用完资源,还是中途报错,它都能确保资源被正确释放,再也不用手动调用close()啦!

它的核心秘密在于上下文管理器协议——只要一个对象实现了__enter__()(开门)和__exit__()(关门)方法,就能用with语句管理。比如文件对象、数据库连接、线程锁,都是天生的"乖宝宝",会乖乖配合with语句~

场景一:文件操作——最经典的"自动关门"

说到with语句,文件操作绝对是它的"成名作"!以前写文件读写,得用try-finally手动关文件,又臭又长:

# 传统写法:手动关文件,生怕忘写finally
file = None
try:
    file = open('data.txt', 'r')
    content = file.read()
except FileNotFoundError:
    print("文件不存在!")
finally:
    if file:
        file.close()  # 漏写这行就等着资源泄漏吧!

有了with语句,直接简化成一行,自动关文件,安全感拉满

# with语句:自动关文件,优雅!
with open('data.txt', 'r') as file:
    content = file.read()
    print(content)
# 出了with块,文件已经自动关闭,随便你怎么操作都不怕~

进阶技巧:还能同时管理多个文件,比如读取一个文件并写入另一个,全程自动关:

# 同时操作两个文件,一个读一个写
with open('input.txt', 'r') as infile, open('output.txt', 'w') as outfile:
    outfile.write(infile.read().upper())  # 把内容转大写写入新文件

看图感受下with语句的简洁:


(代码编辑器截图:左侧文件结构,中间with语句操作文件,底部显示运行结果,全程无手动close())

场景二:数据库连接——再也不怕连接泄漏

数据库连接可是"娇贵资源",一个没关就可能让服务器崩溃!但有了with语句,连接自动关闭,安全感爆棚~

举个SQLite的例子:

import sqlite3

# with语句管理数据库连接,自动关闭
with sqlite3.connect('example.db') as conn:
    cursor = conn.cursor()
    cursor.execute('SELECT * FROM users')  # 查询用户表
    results = cursor.fetchall()  # 获取结果
    print(f"查到{len(results)}条数据!")
# 出了with块,连接已经自动关闭,服务器表示:谢谢你啊~

高级玩家:异步数据库连接(asyncpg)

现在异步编程这么火,with语句也能跟上!比如用asyncpg操作PostgreSQL:

# 异步数据库连接,async with更配哦~
async with asyncpg.create_pool(**creds) as pool:  # 创建连接池
    async with pool.acquire() as conn:  # 获取连接
        await conn.execute("INSERT INTO logs VALUES($1)", datetime.now())  # 插入数据

数据库连接的"自动管理"长这样:


(代码截图:with语句嵌套管理连接池和连接,底部显示运行结果,无需手动close())

场景三:多线程锁——线程安全"一键搞定"

多线程编程最怕啥?竞态条件!多个线程抢着改一个变量,结果乱套了…但用with语句管理线程锁,自动加锁解锁,线程安全so easy!

举个栗子:多线程计数器

import threading

lock = threading.Lock()  # 创建锁
shared_counter = 0  # 共享变量

def worker():
    global shared_counter
    with lock:  # 自动加锁,用完自动解锁
        shared_counter += 1  # 线程安全的自增

# 创建5个线程并发执行
threads = [threading.Thread(target=worker) for _ in range(5)]
for t in threads:
    t.start()
for t in threads:
    t.join()

print(f"最终结果:{shared_counter}")  # 输出5,完美!

对比一下:加锁vs不加锁

不加锁时,5个线程可能同时改shared_counter,结果可能是3或4;加了with lock后,结果一定是5! 看下图效果:


(左侧代码加锁,右侧记事本显示结果:加锁后5行整齐输出,不加锁则混乱)

高级玩法:自定义上下文管理器,想管啥就管啥!

with语句不止能用内置对象,还能自己定义"管家"!比如写个计时器,自动统计代码块运行时间:

import time

class Timer:
    def __enter__(self):
        self.start = time.time()  # 开门:记录开始时间
        return self

    def __exit__(self, *args):
        print(f"耗时:{time.time()-self.start:.2f}秒")  # 关门:计算耗时

# 用起来!
with Timer():
    time.sleep(1.5)  # 模拟耗时操作
# 输出:耗时:1.50秒

甚至可以用contextlib装饰器简化:

from contextlib import contextmanager

@contextmanager
def temp_dir():
    path = tempfile.mkdtemp()  # 创建临时目录
    try:
        yield path  # 返回目录路径
    finally:
        shutil.rmtree(path)  # 自动删除目录

with temp_dir() as tmp:
    print(f"使用临时目录:{tmp}")  # 用完自动删,干净!

避坑指南:这些错误别踩!

  1. AttributeError: enter
    → 原因:对象没实现上下文协议(没写
    __enter____exit__) → 解决:检查是否用错对象,比如with 普通变量:就会报错
  2. 嵌套with块
    → 不推荐:
    with A(): with B(): ...
    → 推荐:with A() as a, B() as b: 一行搞定多资源
  3. 异步同步混用
    → 错误:在同步函数里用
    async with
    → 解决:async with只能在async def函数里配合await使用

with语句,Pythoner必备"安全符"

不管是文件、数据库还是多线程,with语句都能帮你自动管理资源,既减少代码量,又避免资源泄漏,简直是"一举两得"!记住:写Python时,遇到需要"打开-关闭"的资源,先想想能不能用with语句,优雅又安全~

最后送大家一句口诀:"with一出,资源无忧;自动管理,bug溜走!"

参考来源:

CSDN博客:Python with语句深度实战指南Python官方文档:Context Managers

相关推荐

小程序微信登录反映慢原来因为login和getUserProfile关系造成的

在我开发算盘记账APP小程序的过程中,遇到了UNIAPP开发微信小程序微信登录反映特别慢的问题。经反复查找原因,原来是因为微信登录界面代码的uni.login中包含了uni.getUserProfil...

uniapp主题切换功能的方式终结篇(全平台兼容)

前面我已经给大家介绍了两种主题切换的方式,每种方式各有自己的优势与缺点,例如“scss变量+vuex”方式兼容好但不好维护与扩展,“scss变量+require”方式好维护但兼容不好,还不清楚的可点下...

UNI又来秀了 新功能不懂玩 这份说明书带你玩遍所有潮科技

当您在车辆仪表盘上看到如下图标时是否一脸迷茫?是否手忙脚乱翻用户手册找不到答案?近段时间,有不少车主反馈长安UNI-T很多功能太新潮了,自己有点跟不上时代的节奏了……莫慌,本期推出长安UNI-T“AR...

一款商用品质的开源商城系统(Yii2+Vue2.0+uniapp)

一、项目简介这是一套很成熟的开源商城系统【开店星】,之前推过一次,后台感兴趣的还不少,今天再来详细介绍一下:基于Yii2+Vue2.0+uniapp框架研发,代码质量堪称商用品质,下载安装无门槛,UI...

惊了,这个国产软件居然这么牛,比 uniapp 还全能

最近跟同事闲聊,大家都在吐槽一个问题:!App是越做越像平台了,但开发却越做越痛苦了。你想加个活动页,产品说今晚上线;你想做个业务扩展,运营说要不你再写个低代码工具;你想适配鸿蒙,领导说最好做个...

JeecgUniapp 全新版本 3.0 发布,采用 uniapp+vue3 最新架构

项目介绍JeecgUniapp是JeecgBoot的配套APP移动框架,项目采用Uniapp、Vue3.0、Vite、Wot-design-uni、TypeScript等最新技术栈,包括...

AIGC+UniApp:自动生成跨平台代码实践

一、颠覆传统的开发革命2024年GitHub统计显示,采用AIGC辅助的UniApp项目平均开发效率提升217%。我们在某APP项目中,用ChatGPT-4o生成基础框架代码,原本需要3人日的登录模块...

长安引力UNI-K iDD OTA无法升级,还算问题吗?车主小题大做?

OTA无法升级,在我看来都不算问题。因为它本来就是锦上添花的配置!并不是雪中送炭的配置!车还可以正常开,导航用手机就可以,听歌连手机蓝牙就可以。但是要说OTA这个配置,最早是特斯拉干的。2012年特斯...

uni-app 多环境部署方案_开发uniapp的环境配置

前言最近几周都在处理公司的移动业务,而为在后期能统一多端,解放自己,迎合公司的技术栈;选用了uni-app来开发。开发前期重新对公司移动业务做深入了解,重构大部分业务逻辑,也抽离出基础组件;但实际...

3分钟拥有一个属于自己的博客网站「腾讯云篇」

一、前言想要搭建一个让全世界的人都可以访问的网站,我们最少需要准备三样东西:①服务器腾讯云服务器首年低至40元/年,「链接」阿里云服务器新用户可以免费使用6个月,新人特惠_云产品推荐_云服务器-阿里云...

3月18号分享一个最新Cursor无限续杯的方法,亲测有效~

1、执行脚本MacOS&Linux系统curl-fsSLhttps://aizaozao.com/accelerate.php/https://raw.githubusercontent.co...

什么是IPFS,如何搭建IPFS节点_ipfs简单理解

IPFS是一个分布式系统,用于存储和访问文件、网站、应用程序和数据。星际文件系统(InterPlanetaryFileSystem).IPFS是一个分布式的web,点到点超媒体协议.可以...

Claude Code:完爆 Cursor 的编程体验

前言最近,听说ClaudeCode这款代码辅助编写产品很强,有人把Cursor比作实习生水平,ClaudeCode比作高级工程师水平。起初不以为意,因为特殊原因,Claude无法直接访问。然而,...

10分钟搞定gitlab-ci自动化部署_gitlab ci 自动化部署

gitlab-ci是持续集成工具/自动化部署工具,类似jenkins。持续集成是将代码集成到共享存储库并尽可能早地自动构建/测试每个更改的实践-通常一天几次。概述在编码完成时都会进行打包发布...

「 小白玩树莓派系列」Raspberry Pi 4 Ubuntu USB 大容量存储启动指南

Canonical对RaspberryPi的官方支持已经走过了漫长的道路。我们现在可以在Pi安装官方支持的Ubuntu!在我之前的Pi上的Ubuntu18.04指南中,Raspb...