nvkdeamon.py

#! python3
# -*- coding: utf-8 -*-
'''
自动监测生肉文件夹变化，如果有新文件进入自动启动压制
'''

import sys
def show_exception_and_exit(exc_type, exc_value, tb):
	import traceback
	traceback.print_exception(exc_type, exc_value, tb)
	input('''脚本遇到错误，请截图此画面发送给Kilo19。5秒后继续
	Error encountered, please send a screenshot of this error to Kilo19
	Press Enter to Exit
	''')
	import time
	time.sleep(5)

sys.excepthook = show_exception_and_exit

import subprocess
import codecs
import time
import sys
import os
import nvksupport
from settings import cloudRoot, cloudDir, customDownDir

scriptDir = os.path.dirname(os.path.realpath(sys.argv[0]))

def GrabArgs(lines, argLen):
	returnVal = [''] * argLen
	for x in range(len(returnVal)):
		if len(lines) > x:
			returnVal[x] = lines[x].strip()
	return returnVal

def ListIngredients(dirPath, name, ext):
	myIngredients = set()
	if os.path.isdir(dirPath):
		for root, dirs, files in os.walk(dirPath):
			if root == dirPath:
				for file in files:
					if not name in file and file[-3:] == ext:
						fileStr = os.path.join(root, file) + '\n'
						myIngredients.add(fileStr)
	return myIngredients

def ListFile(dirPath, name, ext):
	candidates = set()
	if os.path.isdir(dirPath):
		for root, dirs, files in os.walk(dirPath):
			for file in files:
				if file[-3:] == ext and name in file:
					fileStr = os.path.join(root, file) + '\n'
					candidates.add(fileStr)
	return candidates

def SafeReadandWriteHead(inName, fileHead = None):
	lines = []
	if not os.path.isfile(inName):
		codecs.open(inName, 'w', 'utf-8').close()
	else:
		inf = codecs.open(inName, 'r', 'utf-8')
		lines = inf.readlines()
		inf.close()

	lines = [line.strip() + '\n' for line in lines]
	if fileHead:
		fileHead = [l.strip() + '\n' for l in fileHead]
		if len(lines) < len(fileHead) or lines[:len(fileHead)] != fileHead:
			print("writehead: " + time.ctime())
			inf = codecs.open(inName, 'w', 'utf-8')
			inf.writelines(fileHead)
			inf.close()
	x = 0
	while x < len(lines):
		if lines[x][0] == '#':
			lines.pop(x)
		else:
			x += 1
	return lines

def DeamonHelper(inName, argLen, scriptName, fileHead):
	inName = os.path.join(customDownDir, inName + '.txt')
	lines = SafeReadandWriteHead(inName, fileHead)
	if len(lines):
		inf = codecs.open(inName, 'w+', 'utf-8')
		if fileHead:
			fileHead = [l.strip() + '\n' for l in fileHead]
			inf.writelines(fileHead)
		else:
			inf.write('')
		inf.close()

		while lines:
			args = GrabArgs(lines, argLen)
			lines = lines[len(args):]
			print(inName + ' command: ' + time.ctime())
			for arg in args:
				print(arg)
			# For safety
			time.sleep(0.5)
			subprocess.Popen(
				[
					'py', '-3', os.path.join(scriptDir, scriptName + '.py')
				] + args,
				creationflags = subprocess.CREATE_NEW_CONSOLE
			 )

if __name__ == "__main__":
	print("NVK Deamon online: " + time.ctime())
	ingHead = ['#网盘高压下令文件',
			'#不要删除开头带#的行，这些都是注释',
			'#不要使用OC网页版编辑此文件',
			'#此文件为UTF-8编码，用来下令压生肉',
			'#下令前请先用everything搜索（任务栏那个放大镜）',
			'#优先下载1080p',
			'#搜索时按Alt+3然后Alt+4可以缩放everything窗口',
			'#生肉速度大约视频1/2时长',
			'#同步到网盘速度不等']

	beHead = ['#压熟肉下令文件，带#的行都是注释',
			'#请在BUG检测完成后再压制',
			'#不要使用OC网页版编辑此文件',
			'#教程https://docs.google.com/document/d/1CdyqEP7eanL6J6xUpbcOuq-uIBQkSEqhhHgeGFT94ik/edit',
			'#此文件为UTF-8编码，用来下令压熟肉',
			'#熟肉速度大约视频0.8-0.9倍',
			'#请同时提供生肉路径与字幕所在网盘文件夹，然后保存',
			'#同步到网盘速度不等']

	cookedHead = ['#此文件专门记录熟肉路径，自动更新',
			   '#请优先从已有熟肉中选择上传',
			   '#上传完成后尽快删除']

	while True:
		DeamonHelper("ing", 1, "NixieCloud_Enc", ingHead)
		DeamonHelper("BE", 3, "Bili_Enc", beHead)

		ingredientsListPath = os.path.join(
			scriptDir, 'ingredientsList' + '.txt'
		)
		ingredients = ListIngredients(customDownDir, '[BE_', 'mp4')

		cookedPath = os.path.join(customDownDir, 'cookedList' + '.txt')
		cooked = ListFile(os.path.join(cloudRoot, cloudDir), '[BE_', 'mp4')

		oldIngList = set(SafeReadandWriteHead(ingredientsListPath, None))
		oldcookedList = set(SafeReadandWriteHead(cookedPath, None))

		if ingredients != oldIngList:
			#set difference finds new ingredients to transcode
			newIngs = list(ingredients - oldIngList)
			print('Auto NCE: ' + time.ctime())
			for newIng in newIngs:
				print(newIng)
				newIngStripped = newIng.strip()
				newIngStripped = nvksupport.ReplaceNaughtyCharacters(
					newIngStripped, nvksupport.tl
				)
				subprocess.Popen(
					[
						'py', '-3',
						os.path.join(scriptDir, "NixieCloud_Enc" + '.py'),
						newIngStripped
					],
					creationflags=subprocess.CREATE_NEW_CONSOLE
				)
			ingredients = ListIngredients(customDownDir, '[BE_', 'mp4')
			print("writeIng: " + time.ctime())
			outFile = codecs.open(ingredientsListPath, 'w', 'utf-8')
			outFile.writelines(list(ingredients))
			outFile.close()

		if cooked != oldcookedList:
			print("writeCooked: " + time.ctime())
			codecs.open(cookedPath, 'w', 'utf-8').close()
			SafeReadandWriteHead(cookedPath, cookedHead)
			outFile = codecs.open(cookedPath, 'a+', 'utf-8')
			outFile.writelines(list(cooked))
			outFile.close()

		time.sleep(5)