mirror/poke
1
0
Fork 0
mirror of https://codeberg.org/ashley/poke.git synced 2024-11-23 06:37:49 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

stop writing bundler output to disk

Browse source
This commit is contained in:
nin0dev 2024-06-23 23:28:08 -04:00
parent 08271eb2b4
commit 6dafa572a4
1 changed files with 56 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

96
videobundler/main.py
View file

@ -11,51 +11,67 @@ app = web.Application()
app.router._frozen = False app.router._frozen = False
def get_random_string(length): def get_random_string(length):
# choose from all lowercase letter # choose from all lowercase letter
letters = string.ascii_lowercase letters = string.ascii_lowercase
result_str = "".join(random.choice(letters) for i in range(length)) result_str = "".join(random.choice(letters) for i in range(length))
return result_str return result_str
async def run_command(cmd):
# Create subprocess
process = await asyncio.create_subprocess_shell(
cmd,
stdout=asyncio.subprocess.PIPE,
)
# Wait for the subprocess to finish
stdout, stderr = await process.communicate()
# Check for errors
if process.returncode!= 0:
# Log or handle the error
print(f"Command '{args}' failed with return code {process.returncode}")
return None
# Decode stdout and return
return stdout
async def merge(request): async def merge(request):
# register params # register params
job_id = request.rel_url.query["id"] job_id = request.rel_url.query["id"]
video_id: str = request.rel_url.query["id"] video_id: str = request.rel_url.query["id"]
audio_itag: str = request.rel_url.query["audio_itag"] audio_itag: str = request.rel_url.query["audio_itag"]
video_itag: str = request.rel_url.query["video_itag"] video_itag: str = request.rel_url.query["video_itag"]
# validate # validate
if " " in video_id or len(video_id) > 11: if " " in video_id or len(video_id) > 11:
print(f"Video {video_id} flagged as invalid, dropping request") print(f"Video {video_id} flagged as invalid, dropping request")
return return
if not audio_itag.isdigit(): if not audio_itag.isdigit():
print(f"Audio itag {audio_itag} flagged as invalid, dropping request") print(f"Audio itag {audio_itag} flagged as invalid, dropping request")
return return
if not video_itag.isdigit(): if not video_itag.isdigit():
print(f"Video itag {video_itag} flagged as invalid, dropping request") print(f"Video itag {video_itag} flagged as invalid, dropping request")
return return
if os.path.isfile(f"{job_id}.mp4"): if os.path.isfile(f"{job_id}.mp4"):
return web.FileResponse( return web.FileResponse(
path=f"{job_id}.mp4" path=f"{job_id}.mp4"
) )
cmdline = f"ffmpeg -i \"https://eu-proxy.poketube.fun/latest_version?id={video_id}&itag={audio_itag}&local=true\" -i \"https://eu-proxy.poketube.fun/latest_version?id={video_id}&itag={video_itag}&local=true\" -c copy -f mp4 -movflags frag_keyframe+empty_moov {video_id}.mp4" cmdline = f"ffmpeg -i \"https://eu-proxy.poketube.fun/latest_version?id={video_id}&itag={audio_itag}&local=true\" -i \"https://eu-proxy.poketube.fun/latest_version?id={video_id}&itag={video_itag}&local=true\" -c copy -f mp4 -movflags frag_keyframe+empty_moov -"
proc_ffmpeg = await asyncio.create_subprocess_shell( #proc_ffmpeg = await asyncio.create_subprocess_shell(
cmdline, # cmdline,
stdout=asyncio.subprocess.PIPE, # stdout=asyncio.subprocess.PIPE
stderr=asyncio.subprocess.PIPE #)
) #print(f"ffmpeg -i \"https://eu-proxy.poketube.fun/latest_version?id={video_id}&itag={audio_itag}&local=true\" -i \"https://eu-proxy.poketube.fun/latest_version?id={video_id}&itag={video_itag}&local=true\" -c copy -f mp4 -movflags frag_keyframe+empty_moov -")
print(f"ffmpeg -i \"https://eu-proxy.poketube.fun/latest_version?id={video_id}&itag={audio_itag}&local=true\" -i \"https://eu-proxy.poketube.fun/latest_version?id={video_id}&itag={video_itag}&local=true\" -c copy -f mp4 -movflags frag_keyframe+empty_moov -") #stdout, _ = await proc_ffmpeg.communicate()
stdout, stderr = await proc_ffmpeg.communicate() iwannakillmyself = await run_command(cmdline)
response = FileResponse(f"{video_id}.mp4") response = Response(body=iwannakillmyself, content_type="video/mp4", headers="")
return response return response
async def ping(request): async def ping(request):
return web.Response(body='{"success": true}', content_type="application/json") return web.Response(body='{"success": true}', content_type="application/json")
async def init_app(): async def init_app():
app.router.add_get("/{id:.+}", merge) app.router.add_get("/{id:.+}", merge)
app.router.add_get("/", ping) app.router.add_get("/", ping)
return app return app
if __name__ == '__main__': if __name__ == '__main__':
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
app = loop.run_until_complete(init_app()) app = loop.run_until_complete(init_app())
web.run_app(app, port=3030) web.run_app(app, port=3030)