修改上面代码,用input来输入视频地址,下载视频后保存在本地"e:\\下载短视频"
时间: 2024-02-28 08:53:03 浏览: 88
可以使用Python内置的`input()`函数来获取用户输入的视频地址,并使用`os`模块来创建保存视频的目录。
以下是修改后的示例代码:
```python
import asyncio
import re
import os
import aiohttp
from pyppeteer import launch
async def download_douyin_video(url, save_dir):
# 创建浏览器实例
browser = await launch(headless=True)
page = await browser.newPage()
# 访问抖音网页
await page.goto(url)
# 等待视频加载完成
await page.waitForSelector('.video-player video')
# 获取视频url和标题
video_url = await page.evaluate('document.querySelector(".video-player video").src')
title = await page.title()
# 关闭浏览器
await browser.close()
# 去除水印
video_url = re.sub('watermark/.*?/', '', video_url)
# 下载视频到本地
os.makedirs(save_dir, exist_ok=True)
with open(f'{save_dir}/{title}.mp4', 'wb') as f:
async with aiohttp.ClientSession() as session:
async with session.get(video_url) as resp:
f.write(await resp.read())
if __name__ == '__main__':
url = input('请输入抖音视频地址:')
save_dir = 'e:/下载短视频'
asyncio.get_event_loop().run_until_complete(download_douyin_video(url, save_dir))
```
注意,保存视频的目录需要使用正确的文件路径格式,例如`e:/下载短视频`。
阅读全文
相关推荐
![zip](https://img-home.csdnimg.cn/images/20241231045053.png)
![zip](https://img-home.csdnimg.cn/images/20241231045053.png)
![zip](https://img-home.csdnimg.cn/images/20241231045053.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)