format render_utils

render_utils.py

@@ -5,9 +5,10 @@ import lxml.etree
 
 from .render_kdenlive import melt_xml
 
+
 def parse_lang(lang):
     if lang and "," in lang:
-        lang = lang.split(',')
+        lang = lang.split(",")
     if isinstance(lang, list):
         tlang = lang[1:]
         lang = lang[0]
@@ -17,6 +18,7 @@ def parse_lang(lang):
         lang = None
     return lang, tlang
 
+
 def random_int(seq, length):
     n = n_ = length - 1
     # print('len', n)
@@ -31,25 +33,29 @@ def random_int(seq, length):
     r = int(round(n_ * r / base))
     return r
 
+
 def random_choice(seq, items, pop=False):
     n = random_int(seq, len(items))
     if pop:
         return items.pop(n)
     return items[n]
 
+
 def chance(seq, chance):
     return (seq() / 10) < chance
 
+
 def get_clip_by_seqid(clips, seqid):
     selected = None
     for i, clip in enumerate(clips):
-        if clip['seqid'] == seqid:
+        if clip["seqid"] == seqid:
             selected = i
             break
     if selected is not None:
         return clips.pop(i)
     return None
 
+
 def get_scene_duration(scene):
     if isinstance(scene, str):
         with open(scene) as fd:
@@ -58,17 +64,19 @@ def get_scene_duration(scene):
     for key, value in scene.items():
         for name, clips in value.items():
             for clip in clips:
-                duration += int(clip['duration'] * 24)
+                duration += int(clip["duration"] * 24)
             return duration / 24
 
+
 def get_offset_duration(prefix):
     duration = 0
     for root, folders, files in os.walk(prefix):
         for f in files:
-            if f == 'scene.json':
+            if f == "scene.json":
                 duration += get_scene_duration(scene)
     return duration
 
+
 def get_track_duration(scene, k, n):
     duration = 0
     for key, value in scene.items():
@@ -76,41 +84,45 @@ def get_track_duration(scene, k, n):
             for name, clips in value.items():
                 if name == n:
                     for clip in clips:
-                        duration += int(clip['duration'] * 24)
+                        duration += int(clip["duration"] * 24)
                     return duration / 24
 
+
 def get_project_duration(file):
     out = melt_xml(file)
-    chain = lxml.etree.fromstring(out).xpath('producer')[0]
-    duration = int(chain.attrib['out']) + 1
+    chain = lxml.etree.fromstring(out).xpath("producer")[0]
+    duration = int(chain.attrib["out"]) + 1
     return duration
 
+
 def fix_overlaps(data):
     previous = None
     for sub in data:
         if previous is None:
             previous = sub
         else:
-            if sub['in'] < previous['out']:
-                previous['out'] = sub['in'] - 0.001
+            if sub["in"] < previous["out"]:
+                previous["out"] = sub["in"] - 0.001
             previous = sub
     return data
 
+
 def shift_clips(data, offset):
     for clip in data:
-        clip['in'] += offset
-        clip['out'] += offset
+        clip["in"] += offset
+        clip["out"] += offset
     return data
 
+
 def ass_encode(subs, options):
     if "lang" in options:
-        langs = options["lang"].split(',')
+        langs = options["lang"].split(",")
     else:
         langs = list(subs[0]["values"])
     # print('ass_encode', langs, options)
     # print(subs)
 
-    header = '''[Script Info]
+    header = """[Script Info]
 ScriptType: v4.00+
 PlayResX: 1920
 PlayResY: 1080
@@ -119,7 +131,7 @@ YCbCr Matrix: None
 
 [V4+ Styles]
 Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding
-'''
+"""
     ass = header
     offset = options.get("sub_margin", 10)
     spacing = options.get("sub_spacing", 20)
@@ -129,26 +141,26 @@ Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour,
         if isinstance(options.get("font"), list) and lang in options["font"]:
             font = options["font"][lang]
         else:
-            font = 'SimHei' if lang in ('zh', 'jp') else 'Menlo'
+            font = "SimHei" if lang in ("zh", "jp") else "Menlo"
         if isinstance(options.get("font_size"), list) and lang in options["font_size"]:
             size = options["font_size"][lang]
         else:
-            size = 46 if font == 'SimHei' else 42
+            size = 46 if font == "SimHei" else 42
 
         styles.append(
-            f'Style: {lang},{font},{size},&Hffffff,&Hffffff,&H0,&H0,0,0,0,0,100,100,0,0,1,1,0,2,10,10,{offset},1'
+            f"Style: {lang},{font},{size},&Hffffff,&Hffffff,&H0,&H0,0,0,0,0,100,100,0,0,1,1,0,2,10,10,{offset},1"
         )
         offset += size + spacing
-    ass += '\n'.join(reversed(styles)) + '\n'
+    ass += "\n".join(reversed(styles)) + "\n"
     events = [
-        'Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text'
+        "Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text"
     ]
     for sub in subs:
-        start = ox.format_timecode(sub['in']).rstrip('0')
-        stop = ox.format_timecode(sub['out']).rstrip('0')
+        start = ox.format_timecode(sub["in"]).rstrip("0")
+        stop = ox.format_timecode(sub["out"]).rstrip("0")
         for lang in reversed(langs):
-            value = sub['values'][lang]
-            event = f'Dialogue: 0,{start},{stop},{lang},,0,0,0,,{value}'
+            value = sub["values"][lang]
+            event = f"Dialogue: 0,{start},{stop},{lang},,0,0,0,,{value}"
             events.append(event)
-    ass += '\n\n[Events]\n' + '\n'.join(events) + '\n'
+    ass += "\n\n[Events]\n" + "\n".join(events) + "\n"
     return ass