抖音分享链接视频下载

import requests
import re
from urllib.parse import unquote
import json
from pprint import pprint
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import os

def get_video_id(url):
    # 创建一个Chrome WebDriver实例
    chrome_options = Options()
    chrome_options.add_argument("--headless")  # This line makes Chrome run in headless mode
    chrome_options.add_argument("--disable-logging")
    # Initialize the WebDriver with the specified options
    driver = webdriver.Chrome(options=chrome_options) 
    # 打开网页
    driver.get(url)
    # 获取跳转后的网址
    final_url = driver.current_url
    # 打印跳转后的网址
    print("跳转后的网址:", final_url)
    # 关闭WebDriver实例
    #driver.quit()
    video_id = final_url.split("/video/")[1]
    return video_id

def download_byid(video_id):
    url ="https://www.douyin.com/discover?modal_id="+video_id
    print(url)
    headers = {
    'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36'
    # 'Cookie':''
    }
    response = requests.get(url= url,headers=headers)
    print(response)
    html = response.text
    #
    info = re.findall('<script id="RENDER_DATA" type="application/json">(.*?)</script>',html)
    info_json = unquote(info[0])

    #pprint(info_json)
    json_data = json.loads(info_json)

    video_url = 'https:'+json_data['app']['videoDetail']['video']['bitRateList'][0]['playAddr'][0]['src']
    video_title = json_data['app']['videoDetail']['desc']
    pprint(video_url)
    pprint(video_title)
    video_content = requests.get(url = video_url,headers=headers).content
    with open ("download/"+video_id+'.mp4',mode= 'wb')as f:
        f.write(video_content)
    with open ("download/"+video_id+'.txt',mode= 'w', encoding='utf-8')as f:
        f.write(video_title)
    return "download/"+video_id+'.mp4',video_title,video_url

if __name__ == "__main__":
    url2 = 'https://v.douyin.com/i2G6aAcM/'
    print(get_url(url2))
    download(url2)