Forráskód Böngészése

Build a clean metadata record when renaming muxed files

ZiB 10 hónapja
szülő
commit
f417d89e59
1 módosított fájl, 20 hozzáadás és 5 törlés
  1. 20 5
      loki/remux.py

+ 20 - 5
loki/remux.py

@@ -17,6 +17,17 @@ subs_ext      = [ r'srt' ]
 subs_sdh      = [ r'sdh' ]
 subs_lang     = [ r'\w\w' ]
 
+# Default metadata record
+meta_default = {
+  'path':      None,
+  'name':      '',
+  'ext':       '',
+  'lang':      None,
+  'sdh':       False,
+  'mux_vers':  None,
+  'lang_vers': None,
+}
+
 # ENV config
 if 'DEBUG' in os.environ:
   debug = bool(os.environ['DEBUG'])
@@ -71,7 +82,9 @@ def scan(files):
 
   # Scan this directory for media
   for file in files:
-    info = { 'name': file, 'path': parent.joinpath(file), 'lang': None, 'sdh': False }
+    info = meta_default.copy()
+    info['path'] = parent.joinpath(file)
+    info['name'] = file
 
     # Parse media paths
     try:
@@ -183,7 +196,6 @@ def mux(media):
     # Set xattr
     if not write_xattr(temp, mux_xattr, mux_vers):
       continue
-    videos[0]['mux_vers'] = mux_vers
 
     # Remove sources
     try:
@@ -200,9 +212,12 @@ def mux(media):
     try:
       logger.debug(f"Renaming: {temp.name} -> {final.name}")
       temp.rename(final)
-      videos[0]['path'] = final
-      videos[0]['name'] = final.name
-      media.append(videos[0])
+      info = meta_default.copy()
+      info['path']     = final
+      info['name']     = final.name
+      info['ext']      = 'mkv'
+      info['mux_vers'] = mux_vers
+      media.append(info)
     except BaseException as e:
       err = e
     if not final.exists():