forked from rDrama/rDrama
fix bugs
parent
42344cb76f
commit
3a4aa3f3af
|
@ -16,4 +16,4 @@ class Media(Base):
|
|||
super().__init__(*args, **kwargs)
|
||||
|
||||
def __repr__(self):
|
||||
return f"<Streamer(id={self.id})>"
|
||||
return f"<Media(kind={self.kind}, filename={self.filename})>"
|
||||
|
|
|
@ -11,6 +11,7 @@ import gevent
|
|||
import imagehash
|
||||
from shutil import copyfile
|
||||
from files.classes.media import *
|
||||
from files.__main__ import db_session
|
||||
|
||||
def process_files():
|
||||
body = ''
|
||||
|
@ -57,7 +58,7 @@ def process_audio(file):
|
|||
return f'{SITE_FULL}{name}'
|
||||
|
||||
|
||||
def webm_to_mp4(old, new):
|
||||
def webm_to_mp4(old, new, vid):
|
||||
tmp = new.replace('.mp4', '-t.mp4')
|
||||
subprocess.run(["ffmpeg", "-y", "-loglevel", "warning", "-i", old, "-map_metadata", "-1", tmp, "-threads:v", "1"], check=True, stderr=subprocess.STDOUT)
|
||||
os.replace(tmp, new)
|
||||
|
@ -65,6 +66,18 @@ def webm_to_mp4(old, new):
|
|||
requests.post(f'https://api.cloudflare.com/client/v4/zones/{CF_ZONE}/purge_cache', headers=CF_HEADERS,
|
||||
data=f'{{"files": ["{SITE_FULL}{new}"]}}', timeout=5)
|
||||
|
||||
db = db_session()
|
||||
media = Media(
|
||||
kind='video',
|
||||
filename=new.split('/')[-1],
|
||||
user_id=vid,
|
||||
size=os.stat(new).st_size
|
||||
)
|
||||
db.add(media)
|
||||
db.commit()
|
||||
db.close()
|
||||
|
||||
|
||||
def process_video(file):
|
||||
old = f'/videos/{time.time()}'.replace('.','')
|
||||
file.save(old)
|
||||
|
@ -82,18 +95,18 @@ def process_video(file):
|
|||
if extension == 'webm':
|
||||
new = new.replace('.webm', '.mp4')
|
||||
copyfile(old, new)
|
||||
gevent.spawn(webm_to_mp4, old, new)
|
||||
gevent.spawn(webm_to_mp4, old, new, g.v.id)
|
||||
else:
|
||||
subprocess.run(["ffmpeg", "-y", "-loglevel", "warning", "-i", old, "-map_metadata", "-1", "-c:v", "copy", "-c:a", "copy", new], check=True)
|
||||
os.remove(old)
|
||||
|
||||
media = Media(
|
||||
kind='video',
|
||||
filename=new.split('/')[-1],
|
||||
user_id=g.v.id,
|
||||
size=os.stat(new).st_size
|
||||
)
|
||||
g.db.add(media)
|
||||
media = Media(
|
||||
kind='video',
|
||||
filename=new.split('/')[-1],
|
||||
user_id=g.v.id,
|
||||
size=os.stat(new).st_size
|
||||
)
|
||||
g.db.add(media)
|
||||
|
||||
return f'{SITE_FULL}{new}'
|
||||
|
||||
|
|
Loading…
Reference in New Issue