Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
A simple script to get lots of viewers of Bilibili Live
#!/usr/bin/env python
#coding:utf-8
# Author: Beining --<cnbeining#gmail.com>
# Purpose: A simple script to get lots of viewers of Bilibili Live
# Created: 08/11/2015
# Error report: http://www.cnbeining.com/?p=952
# https://github.com/cnbeining somewhere within my gists
import sys
import time
import getopt
from multiprocessing import Process
from websocket import create_connection, WebSocketConnectionClosedException, WebSocketProxyException, WebSocketException
import random
import re
import traceback
import socket
global proxy_list
#----------------------------------------------------------------------
def fake_connector(cid, is_proxy = False):
""""""
try:
if is_proxy:
proxy_server = random.choice(proxy_list)
host, port = proxy_server.split(':')
ws = create_connection('ws://livecmt.bilibili.com:88/{cid}'.format(cid = cid), http_proxy_host = host, http_proxy_port = int(port))
else:
ws = create_connection('ws://livecmt.bilibili.com:88/{cid}'.format(cid = cid))
while 1:
time.sleep(5)
a = ws.recv()
except (WebSocketProxyException, socket.error, TypeError): #proxy server died
#/usr/local/lib/python2.7/site-packages/websocket/_http.py L186 WTF????
print(proxy_server + ' died!')
try:
proxy_list.remove(proxy_server)
except Exception, err:
print(traceback.format_exc())
except (WebSocketConnectionClosedException, WebSocketException):
return -1
except Exception, err:
print(traceback.format_exc())
return 0
finally:
return
#----------------------------------------------------------------------
def main(cid, thread_number, is_proxy = False):
""""""
#Get a list of threads
process_list = [Process(target=fake_connector, args=((cid, is_proxy, ))) for i in xrange(int(thread_number))] #Queue? Your ass
[i.start() for i in process_list] #ignite every one
try:
while 1:
alive_list = [i.is_alive() for i in process_list]
print('Active thread: ' + str(len(alive_list)))
death_position_list = [i for i, x in enumerate(alive_list) if x == False]
if len(death_position_list) > 0: #someone died
print('Some died, adding {COUNT} new threads'.format(COUNT = len(death_position_list)))
for i in death_position_list:
del process_list[i] #remove body
process_list.append(Process(target=fake_connector, args=((cid, is_proxy, ))))
process_list[-1].start() #ignite
time.sleep(3)
except Exception as e:
print(e)
for i in process_list:
try:
i.terminate() #ensure safe exit
except:
pass
exit()
#----------------------------------------------------------------------
def proxy_file_to_list(proxy_file):
"""fileIO->list
file:
112.20.190.135:80
223.151.86.8:9000
171.36.67.213:8090
124.202.168.26:8118
219.133.31.120:8888
list:
['112.20.190.135:80', '223.151.86.8:9000', '171.36.67.213:8090',...
"""
final_list = []
pattern = re.compile(r'\d+.\d+.\d+.\d+:\d+') #117.149.219.176:8123
with open(proxy_file, 'r') as file_this:
final_list = [line.strip() for line in file_this if pattern.match(line)]
return final_list
#----------------------------------------------------------------------
def usage():
""""""
print('''Use as:
-c: cid, room number
-t: thread number
-p: proxy file
Press Ctrl+C to exit.
''')
if __name__=='__main__':
is_proxy = False
argv_list = sys.argv[1:]
try:
opts, args = getopt.getopt(argv_list, "hc:t:p:",
['help', "cid=", 'thread_number=', 'proxy_file='])
except getopt.GetoptError:
usage()
exit()
for o, a in opts:
if o in ('-h', '--help'):
usage()
exit()
if o in ('-c', '--cid'):
cid = a
if o in ('-t', '--thread_number'):
thread_number = int(a)
if o in ('-p', '--proxy_file'):
is_proxy = True
proxy_file = a
if is_proxy:
proxy_list = proxy_file_to_list(proxy_file)
print('Getting room {cid} {thread_number} viewers...'.format(cid = cid, thread_number = thread_number))
main(cid, thread_number, is_proxy)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.