「python」抖音短视频下载助手

前不久抖音宣布日活跃用户达到1.5亿,说实话,当看到这个数据我吓了一跳.1.5亿,什么个概念?基本上年轻人人手一个抖音账号.

最近,我在 github 上看到一个项目,专门用来批量下载抖音短视频.有兴趣的可以在 github 上搜索 Jack-Cherish ,上面有不少爬虫的小项目.

接下来把源码分享给大家:

# -*- coding:utf-8 -*-

from splinter.driver.webdriver.chrome import Options, Chrome

from splinter.browser import Browser

from contextlib import closing

import requests, json, time, re, os, sys, time

from bs4 import BeautifulSoup

class DouYin(object):

def __init__(self, width = 500, height = 300):

"""

抖音App视频下载

"""

# 无头浏览器

chrome_options = Options()

chrome_options.add_argument('user-agent="Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36"')

self.driver = Browser(driver_name='chrome', executable_path='D:/chromedriver', options=chrome_options, headless=True)

def get_video_urls(self, user_id):

"""

获得视频播放地址

Parameters:

user_id：查询的用户ID

Returns:

video_names: 视频名字列表

video_urls: 视频链接列表

nickname: 用户昵称

"""

video_names = []

video_urls = []

unique_id = ''

while unique_id != user_id:

search_url = 'https://api.amemv.com/aweme/v1/discover/search/?cursor=0&keyword=%s&count=10&type=1&retry_type=no_retry&iid=17900846586&device_id=34692364855&ac=wifi&channel=xiaomi&aid=1128&app_name=aweme&version_code=162&version_name=1.6.2&device_platform=android&ssmix=a&device_type=MI+5&device_brand=Xiaomi&os_api=24&os_version=7.0&uuid=861945034132187&openudid=dc451556fc0eeadb&manifest_version_code=162&resolution=1080*1920&dpi=480&update_version_code=1622' % user_id

req = requests.get(url = search_url, verify = False)

html = json.loads(req.text)

aweme_count = html['user_list'][0]['user_info']['aweme_count']

uid = html['user_list'][0]['user_info']['uid']

nickname = html['user_list'][0]['user_info']['nickname']

unique_id = html['user_list'][0]['user_info']['unique_id']

user_url = 'https://www.douyin.com/aweme/v1/aweme/post/?user_id=%s&max_cursor=0&count=%s' % (uid, aweme_count)

req = requests.get(url = user_url, verify = False)

html = json.loads(req.text)

i = 1

for each in html['aweme_list']:

share_desc = each['share_info']['share_desc']

if '抖音-原创音乐短视频社区' == share_desc:

video_names.append(str(i) + '.mp4')

i += 1

else:

video_names.append(share_desc + '.mp4')

video_urls.append(each['share_info']['share_url'])

return video_names, video_urls, nickname

def get_download_url(self, video_url):

"""

获得带水印的视频播放地址

Parameters:

video_url：带水印的视频播放地址

Returns:

download_url: 带水印的视频下载地址

"""

req = requests.get(url = video_url, verify = False)

bf = BeautifulSoup(req.text, 'lxml')

script = bf.find_all('script')[-1]

video_url_js = re.findall('var data = \[(.+)\];', str(script))[0]

video_html = json.loads(video_url_js)

download_url = video_html['video']['play_addr']['url_list'][0]

return download_url

def video_downloader(self, video_url, video_name, watermark_flag=True):

"""

视频下载

Parameters:

video_url: 带水印的视频地址

video_name: 视频名

watermark_flag: 是否下载不带水印的视频

Returns:

无

"""

size = 0

if watermark_flag == True:

video_url = self.remove_watermark(video_url)

else:

video_url = self.get_download_url(video_url)

with closing(requests.get(video_url, stream=True, verify = False)) as response:

chunk_size = 1024

content_size = int(response.headers['content-length'])

if response.status_code == 200:

sys.stdout.write(' [文件大小]:%0.2f MB\n' % (content_size / chunk_size / 1024))

with open(video_name, "wb") as file:

for data in response.iter_content(chunk_size = chunk_size):

file.write(data)

size += len(data)

file.flush()

sys.stdout.write(' [下载进度]:%.2f%%' % float(size / content_size * 100) + '\r')

sys.stdout.flush()

def remove_watermark(self, video_url):

"""

获得无水印的视频播放地址

Parameters:

video_url: 带水印的视频地址

Returns:

无水印的视频下载地址

"""

self.driver.visit('http://douyin.iiilab.com/')

self.driver.find_by_tag('input').fill(video_url)

self.driver.find_by_xpath('//button[@class="btn btn-default"]').click()

html = self.driver.find_by_xpath('//div[@class="thumbnail"]/div/p')[0].html

bf = BeautifulSoup(html, 'lxml')

return bf.find('a').get('href')

def run(self):

"""

运行函数

Parameters:

None

Returns:

None

"""

self.hello()

user_id = input('请输入ID(例如40103580):')

video_names, video_urls, nickname = self.get_video_urls(user_id)

if nickname not in os.listdir():

os.mkdir(nickname)

print('视频下载中:共有%d个作品!\n' % len(video_urls))

for num in range(len(video_urls)):

print(' 解析第%d个视频链接 [%s] 中，请稍后!\n' % (num+1, video_urls[num]))

if '\\' in video_names[num]:

video_name = video_names[num].replace('\\', '')

elif '/' in video_names[num]:

video_name = video_names[num].replace('/', '')

else:

video_name = video_names[num]

self.video_downloader(video_urls[num], os.path.join(nickname, video_name))

print('\n')

print('下载完成!')

def hello(self):

"""

打印欢迎界面

Parameters:

None

Returns:

None

"""

print('*' * 100)

print('\t\t\t\t抖音App视频下载小助手')

print('\t\t作者:Jack Cui')

print('*' * 100)

if __name__ == '__main__':

douyin = DouYin()

douyin.run()

想要 python 和其他语言学习资料的私信我,想要怎么科学上网的也可以私信我,记得回复 python,不然无法给你回信.