Uploader.py

import configparser
import logging
import time

from bilibiliuploader.bilibiliuploader import BilibiliUploader
from bilibiliuploader.core import VideoPart
import os
import requests

#log 初始化
LOG_FORMAT = "%(asctime)s - %(levelname)s - %(message)s"
DATE_FORMAT = "%m/%d/%Y %H:%M:%S %p"

logging.basicConfig(filename='Uploader.log', level=logging.INFO, format=LOG_FORMAT, datefmt=DATE_FORMAT)

#读取config文件
def getConfig(section, key):
    config = configparser.ConfigParser()
    configpath =  './Uploader.conf'
    config.read(configpath,'utf-8')
    return config.get(section, key)
#上传
def UploadFile(VPlist):
    la, lb = uploader.upload(parts=VPlist,copyright=1,title=title+time.strftime(" %Y-%m-%d",time.localtime()),tid=tid,tag=tag,desc=desc,open_elec=1,max_retry=20)
    return la,lb
#修改分P
def AppendFile(VPlist):
    uploader.edit(bvid=BV,parts=VPlist,max_retry=20)


VideoPartList = []
UploadingFile = []
LastUploadDate = -1;
BV = 0;
AV = 0;
#读取相对路径和录像文件名词
path = getConfig('Common','path')
type = getConfig('Common','type')
#读取监听的B站直播间
room_id = int(getConfig('BilibiliLive','room_id'))
#读取用户名和密码,以及B站上传设置
username = getConfig('Bilibili','username')
password = getConfig('Bilibili','password')
title = getConfig('Bilibili','title')
tid = int(getConfig('Bilibili','tid'))
tag = getConfig('Bilibili','tag')
desc = getConfig('Bilibili','desc')
dynamtic = getConfig('Bilibili','dynamtic')
#log 读取的数据
logging.debug("Readed"+"path: "+path+' type: '+type+' room_id: '+str(room_id)+' username: '+username+' password: '+ password +' title: '+title)
logging.debug("Readed"+"tid: "+str(tid)+' tag: '+str(tag)+' desc: '+desc+' dynamtic: '+dynamtic+' From the config file')
uploader = BilibiliUploader()
try:
    uploader.login(username,password)
except:
    logging.debug("Login Failed")
else:
    logging.info("Login successful")

#暴力死循环
while(True):
    if LastUploadDate != time.strftime("%d",time.localtime()):
        print("创建新分P尝试")
        #判断今天是否创建过分P
        files = os.listdir(path)
        #遍历录像目录
        for file in files:
            #寻找指定类型的录像文件
            if file.endswith(type):
                #检测录像文件体积编号
                tmpsizeA = os.path.getsize(path+'/'+file)
                time.sleep(10)
                tmpsizeB = os.path.getsize(path+'/'+file)
                #如果录像体积无变化,开始上传
                if tmpsizeA == tmpsizeB:
                    VideoPartList.append(VideoPart(
                        path=path+'/'+file,
                        title=file,
                    ))
                    UploadingFile.append(file)
        if len(VideoPartList) > 0:
            print("创建新分P中")
            AV, BV = UploadFile(VideoPartList)
            # 更新今天分P的BV状态
            LastUploadDate = time.strftime("%d", time.localtime())
            # 更新今天的上传状态
            logging.info("Upload "+LastUploadDate+" steam successful, list of Uploaded file:")
            for file in UploadingFile:
                logging.info("Uploaded "+file+"delete it")
                # 移除额外文件
                os.remove(path + '/' + file)
            VideoPartList = []
            UploadingFile = []
    else:
        print("合并已有分P尝试")
        files = os.listdir(path)
        # 遍历录像目录
        for file in files:
            # 寻找指定类型的录像文件
            if file.endswith(type):
                # 检测录像文件体积编号
                tmpsizeA = os.path.getsize(path + '/' + file)
                time.sleep(10)
                tmpsizeB = os.path.getsize(path + '/' + file)
                # 如果录像体积无变化,开始上传
                if tmpsizeA == tmpsizeB:
                    VideoPartList.append(VideoPart(
                        path=path + '/' + file,
                        title=file,
                    ))
                    UploadingFile.append(file)
        if len(VideoPartList) > 0:
            #如果有可以上传的内容
            print("合并已有分P中")
            AppendFile(VideoPartList)
            logging.info("Upload " + LastUploadDate + " steam successful, list of Uploaded file:")
            for file in UploadingFile:
                # 移除额外文件
                logging.info("Uploaded " + file + "delete it")
                os.remove(path + '/' + file)
            VideoPartList = []
            UploadingFile = []
    time.sleep(60)









"""
def getVcount(UID):
  
      UID: UP主的ID
      :return: 返回视频数量
      
    url = "http://space.bilibili.com/ajax/member/getSubmitVideos"
    p = {'mid':UID,'pagesize':1,'page':1}
    r = requests.get(url,p)
    data = json.load(r)
    return data['count']
def getVList(UID):
  
    UID: UP主的ID
    :return: 返回视频列表
    
    url = "http://space.bilibili.com/ajax/member/getSubmitVideos"
    videocount = getVcount(UID)
    pages = math.ceil(videocount/100)
    vList = []
    for page in range(0,pages):
        p = {'mid': UID, 'pagesize': 100, 'page': page+1}
        r = requests.get(url, p)
        data = json.load(r)
        vList += data['vlist']
    return vList
#监听直播状态, 0下播 1开播
def GetLiveStates(id):
    Room = {'room_id': id}
    RoomInfo = requests.get("https://api.live.bilibili.com/room/v1/Room/get_info", params=Room)
    states =  RoomInfo.json()['data']['live_status']
    return states
"""