forked from LBRYCommunity/lbry-sdk
59 lines
1.7 KiB
Python
Executable file
59 lines
1.7 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import asyncio
|
|
import logging
|
|
import sys
|
|
|
|
# noinspection PyUnresolvedReferences
|
|
import lbry.wallet # needed to make the following line work (it's a bug):
|
|
from lbry.conf import TranscodeConfig
|
|
from lbry.file_analysis import VideoFileAnalyzer
|
|
|
|
|
|
def enable_logging():
|
|
root = logging.getLogger()
|
|
root.setLevel(logging.DEBUG)
|
|
|
|
handler = logging.StreamHandler(sys.stdout)
|
|
handler.setLevel(logging.DEBUG)
|
|
formatter = logging.Formatter('%(message)s') # %(asctime)s - %(levelname)s -
|
|
handler.setFormatter(formatter)
|
|
root.addHandler(handler)
|
|
|
|
|
|
async def process_video(analyzer, video_file):
|
|
try:
|
|
await analyzer.verify_or_repair(True, False, video_file)
|
|
print("No concerns. Ship it!")
|
|
except FileNotFoundError as e:
|
|
print(str(e))
|
|
except Exception as e:
|
|
print(str(e))
|
|
transcode = input("Would you like to make a repaired clone now? [y/N] ")
|
|
if transcode == "y":
|
|
try:
|
|
new_video_file = await analyzer.verify_or_repair(True, True, video_file)
|
|
print("Successfully created ", new_video_file)
|
|
except Exception as e:
|
|
print("Unable to complete the transcode. Message: ", str(e))
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) < 2:
|
|
print("Usage: check_video.py <path to video file>", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
enable_logging()
|
|
|
|
video_file = sys.argv[1]
|
|
conf = TranscodeConfig()
|
|
analyzer = VideoFileAnalyzer(conf)
|
|
loop = asyncio.ProactorEventLoop() if conf.needs_proactor else asyncio.get_event_loop()
|
|
try:
|
|
loop.run_until_complete(process_video(analyzer, video_file))
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|