直接上代码:
importrequestsimportcsvfrom contextlib importclosing#保存csv文件
defsave_csv(f_name, data):#1. 创建文件对象
f = open(f_name, 'w', encoding='utf-8', newline='')#2. 基于文件对象构建 csv写入对象
csv_writer =csv.writer(f)#4. 写入csv文件内容
for row indata:
csv_writer.writerow(row)#5. 关闭文件
f.close()#下载csv文件
defget_pag():
url= '********'
#读取数据
with closing(requests.get(url, stream=True)) as r:
f= (line.decode('utf-8') for line inr.iter_lines())
reader= csv.reader(f, delimiter=',', quotechar='"')
save_csv(f_name, reader)if __name__ == '__main__':
f_name= '123.csv'reader= get_pag()
亮点来了:
contextlib.closing 这个库,学习了.
1、之前的我,只知道with会用来关闭文件,数据库资源,这很好。
只要实现了__enter__() 和 __exit__()这两个方法的类都可以轻松创建上下文管理器,就能使用with