iScript/flv_cmd.py at master · fengqiaoye/iScript · GitHub

Repository navigation

History

executable file

186 lines (164 loc) · 5.38 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

#!/usr/bin/env python2

# vim: set fileencoding=utf8

import re

import requests

import os

import sys

import argparse

import random

from HTMLParser import HTMLParser

import urllib

import select

s = '\x1b[%d;%dm%s\x1b[0m' # terminual color template

parser = HTMLParser()

############################################################

# wget exit status

wget_es = {

0: "No problems occurred.",

2: "User interference.",

1<<8: "Generic error code.",

2<<8: "Parse error - for instance, \

when parsing command-line optio.wgetrc or .netrc...",

3<<8: "File I/O error.",

4<<8: "Network failure.",

5<<8: "SSL verification failure.",

6<<8: "Username/password authentication failure.",

7<<8: "Protocol errors.",

8<<8: "Server issued an error response."

}

############################################################

headers = {

"User-Agent": "Mozilla/5.0 (X11; Linux x86_64) " \

"AppleWebKit/537.36 (KHTML, like Gecko) " \

"Chrome/40.0.2214.91 Safari/537.36",

"Accept": "text/html,application/xhtml+xml,application/xml;" \

"q=0.9,image/webp,*/*;q=0.8",

"Accept-Encoding": "gzip, deflate, sdch",

"Accept-Language": "en-US,en;q=0.8",

"Referer": "http://flvgo.com/download"

}

ss = requests.session()

ss.headers.update(headers)

def download(info):

if not os.path.exists(info['dir_']):

os.mkdir(info['dir_'])

#else:

#if os.path.exists(info['filename']):

#return 0

num = random.randint(0, 7) % 8

col = s % (2, num + 90, os.path.basename(info['filename']))

print '\n ++ 正在下载:', '#', \

s % (1, 97, info['n']), '/', \

s % (1, 97, info['amount']), \

'#', col

print info['durl']

cmd = 'wget -c -nv --user-agent "%s" -O "%s" "%s"' \

% (headers['User-Agent'], info['filename'], info['durl'])

status = os.system(cmd)

if status != 0: # other http-errors, such as 302.

wget_exit_status_info = wget_es[status]

print('\n\n ----### \x1b[1;91mERROR\x1b[0m ==> \

\x1b[1;91m%d (%s)\x1b[0m ###--- \n\n' \

% (status, wget_exit_status_info))

print s % (1, 91, ' ===> '), cmd

sys.exit(1)

def play(info):

num = random.randint(0, 7) % 8

col = s % (2, num + 90, os.path.basename(info['filename']))

print '\n ++ play:', '#', \

s % (1, 97, info['n']), '/', \

s % (1, 97, info['amount']), \

'#', col

cmd = 'mpv --really-quiet --cache 8140 --cache-default 8140 ' \

'--http-header-fields "User-Agent:%s" ' \

'"%s"' % (headers['User-Agent'], info['durl'])

#'"%s"' % parser.unescape(info['durl'])

os.system(cmd)

timeout = 1

ii, _, _ = select.select([sys.stdin], [], [], timeout)

if ii:

sys.exit(0)

else:

pass

def flvxz_parser(cn):

blocks = cn.split('playerContainer')[1:]

infos = {}

title = re.search(r'class="name">(.+?)<', cn).group(1)

infos['title'] = title

infos['data'] = {}

for bc in blocks:

quality = re.search(r'视频格式：(\w+)', bc).group(1)

size = sum([float(s) for s in re.findall(r'>([\d.]+) MB<', bc)])

durls = re.findall(r'<td><a href="(.+?)">', bc)

infos['data'][quality] = {

'size': size,

'durls': durls

}

return infos

def pickup(infos):

print s % (1, 97, infos['title'])

print s % (1, 97, ' ++ pick a quality:')

sizes = [(infos['data'][q]['size'], q) for q in infos['data']]

sizes.sort()

sizes.reverse()

for i in xrange(len(sizes)):

print s % (1, 91, ' %s' % (i+1)), \

str(sizes[i][0]) + 'MB\t', sizes[i][1]

p = raw_input(s % (1, 92, ' Enter') + ' (1): ')

if p == '':

return sizes[0][1]

if not p.isdigit():

print s % (1, 91, ' !! enter error')

sys.exit()

p = int(p)

if p <= len(infos['data']):

print s % (2, 92, ' -- %s' % sizes[p-1][1])

return sizes[p-1][1]

else:

print s % (1, 91, ' !! enter error')

sys.exit()

def getext(durl):

if durl.find('flv'):

return '.flv'

elif durl.find('mp4'):

return '.mp4'

elif durl.find('m3u8'):

return '.m3u8'

else:

return '.flv'

def main(purl):

apiurl = 'http://flvgo.com/download?url=%s' % urllib.quote(purl)

ss.get('http://flvgo.com')

cn = ss.get(apiurl).content

infos = flvxz_parser(cn)

title = infos['title']

quality = pickup(infos)

durls = infos['data'][quality]['durls']

yes = True if len(durls) > 1 else False

dir_ = os.path.join(os.getcwd(), infos['title']) if yes else os.getcwd()

n = args.from_ - 1

amount = len(durls)

for i in xrange(n, amount):

info = {

'title': title,

'filename': os.path.join(dir_, str(i+1) + getext(durls[i])),

'durl': durls[i],

'dir_': dir_,

'amount': amount,

'n': n

}

if args.play:

play(info)

else:

download(info)

n += 1

if __name__ == '__main__':

p = argparse.ArgumentParser(description='flvxz')

p.add_argument('url', help='site url')

p.add_argument('-p', '--play', action='store_true', \

help='play with mpv')

p.add_argument('-f', '--from_', action='store', \

default=1, type=int, \

help='从第几个开始下载，eg: -f 42')

args = p.parse_args()

main(args.url)

iScript/flv_cmd.py at master · fengqiaoye/iScript · GitHub

Repository navigation

Expand file tree

Breadcrumbs

flv_cmd.py

Latest commit

History

Breadcrumbs

flv_cmd.py

File metadata and controls