main.py

import requests, json, time, os, datetime
from threading import Thread
from sys import stdout
from livestreamer import Livestreamer


global resolution # In what resolution should stream be recorded (will choose closest under this parameter if specified is not present)
resolution = 1080 # 1080p 



model = input(f'INPUT	| Enter model name or link: ')

if 'bongacams' in model:
	model = os.path.basename(os.path.normpath(model))


def sizeConvert(size):
    in_KB = size/1024
    in_MB = size/1024/1024
    in_GB = size/1024/1024/1024

    if in_GB > 0.8:
        return f'{round(in_GB, 2)} GB'
    elif in_MB > 0.8:
        return f'{round(in_MB, 2)} MB'
    else:
        return f'{round(in_KB, 2)} KB'
global sizePrintStop
sizePrintStop = False
def print_file_details(path):
	startTime = time.perf_counter()
	while not sizePrintStop:
		time.sleep(1)
		timeNow = time.perf_counter()
		stdout.write('\r'+f'INFO	| Recording...  {datetime.timedelta(seconds=(round(timeNow - startTime, 0)))}	{sizeConvert(os.path.getsize(path))}'.ljust(100))
		stdout.flush()


def get_data(model):
	headers = {
		'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.101 Safari/537.36',
		'X-Requested-With': 'XMLHttpRequest',
	}
	data = [('method', 'getRoomData'), ('args[]', model)]
	r = requests.post('https://bongacams.com/tools/amf.php', headers=headers, data=data)
	return json.loads(r.text)


def stream(videoServerUrl, model):
	session = Livestreamer()
	session.set_option('http-headers', 'referer=https://bongacams.com/%s' % model)

	url = 'hlsvariant://https:%s/hls/stream_%s/playlist.m3u8' % (videoServerUrl, model)

	streams = session.streams(url)

	resolutions = []
	for x in streams:
	    try:
	        x = int(x[:-1])
	        if x <= resolution:
	            resolutions.append(x)
	    except Exception as e:
	        pass
	# print(f'{resolutions[-1:][0]}p')

	if not resolutions:
	    stream = streams[f'worst']
	else:
	    stream = streams[f'{resolutions[-1:][0]}p']

	fd = stream.open()

	now = datetime.datetime.now()

	filePath = '%s/%s.mp4' % (model, model+now.strftime('%Y-%m-%d-%H-%M'))
	print(f'INFO	| Stream recording started {filePath}')

	if not os.path.exists(model):
		os.makedirs(model)

	sizePrint = Thread(target=print_file_details, args=(filePath,))
	sizePrint.start()

	with open(filePath, 'wb') as f:
		while True:
			try:
				data = fd.read(1024)
				f.write(data)
			except:
				sizePrintStop = True
				print(f'ERROR | Unable to write to a file {filePath}')
				f.close()
				return


if __name__ == '__main__':
	data = get_data(model)
	if 'videoServerUrl' in data['localData']:
		stream(data['localData']['videoServerUrl'], model)
	else:
		sizePrintStop = True
		print('ERROR	| This model is not online')