import time
import requests
import config
import os
import string
import random


def get_current_time(time_format):
    current_struct_time = time.localtime(time.time())
    current_time = time.strftime(time_format, current_struct_time)
    return current_time


def generate_filename(room_id):
    data = dict()
    data['c_time'] = get_current_time('%Y%m%d_%H%M%S_' + ''.join(random.sample(string.ascii_letters + string.digits,4)))
    data['room_id'] = room_id
    baseDir = os.path.join(os.getcwd(), 'files')
    checkRecordDirExisted(baseDir)
    return os.path.join(baseDir, '_'.join(data.values()) + '.flv')


def inform(room_id, desp=''):
    if config.enable_inform:
        param = {
            'text': '直播间：{} 开始直播啦！'.format(room_id),
            'desp': desp,
        }
        resp = requests.get(url=config.inform_url, params=param)
        print_log(room_id=room_id,
                  content='通知完成！') if resp.status_code == 200 else None
    else:
        pass


def print_log(room_id='None', content='None'):
    brackets = '[{}]'
    time_part = brackets.format(get_current_time('%Y-%m-%d %H:%M:%S'))
    room_part = brackets.format(room_id)
    print(time_part, room_part, content)


def checkRecordDirExisted(dirs):
    if not os.path.exists(dirs):
        os.makedirs(dirs)


if __name__ == '__main__':
    print(generate_filename('1075'))
    print_log(content='开始录制')