formatters.py python 2 compliant and misc.

Remove use of ABC, not part of Python 2 and implementing ABCMeta changes between Python 2 and 3 so left it out entirely. Base class now raises NotImplementedError manually.

Fix parse_timecode issue with start and end times being identical

Replaced uses of F-strings with .format() also for compatibility.
This commit is contained in:
Chris Howell 2020-07-09 20:13:58 -07:00
parent 74d36a821e
commit c4b8b5b18d
1 changed files with 22 additions and 21 deletions

View File

@ -1,5 +1,3 @@
from abc import ABC
from abc import abstractclassmethod
from collections import defaultdict
import json
import re
@ -17,17 +15,18 @@ def parse_timecode(time):
>>> parse_timecode(6.93)
'00:00:06,930'
"""
time = float(time)
hours, mins, secs = (
str(int(time)//3600).rjust(2, '0'),
str(int(time)//60).rjust(2, '0'),
str(int(time)%60).rjust(2, '0'),
)
ms = str(int(round((time - int(time))*1000, 2))).rjust(3, '0')
return f"{hours}:{mins}:{secs},{ms}"
times = {
'hours': str(int(time) // 3600).rjust(2, '0'),
'mins': str(int(time) // 60).rjust(2, '0'),
'secs': str(int(time) % 60).rjust(2, '0'),
'ms': str(int(round((time - int(time))*1000, 2))).rjust(3, '0')
}
return "{hours}:{mins}:{secs},{ms}".format(**times)
class TranscriptFormatter(ABC):
class TranscriptFormatter(object):
"""Abstract Base TranscriptFormatter class
This class should be inherited from to create additional
@ -51,7 +50,7 @@ class TranscriptFormatter(ABC):
return cls.DELIMITER.join(
str(transcript) for transcript in transcripts)
@abstractclassmethod
@classmethod
def format(cls, transcript_data):
"""Any subclass must implement this format class method.
@ -61,7 +60,9 @@ class TranscriptFormatter(ABC):
as a string.
:rtype: list[str]
"""
pass
raise NotImplementedError(
cls.__name__ + '.format'
)
class JSONTranscriptFormatter(TranscriptFormatter):
@ -103,22 +104,20 @@ class SRTTranscriptFormatter(TranscriptFormatter):
output = []
for frame, item in enumerate(transcript_data, start=1):
start_time = float(item.get('start'))
duration = float(item.get('dur', '0.0'))
end_time = parse_timecode(start_time + duration)
start_time = parse_timecode(start_time)
duration = float(item.get('duration', '0.0'))
output.append("{frame}\n".format(frame=frame))
output.append("{start_time} --> {end_time}\n".format(
start_time=start_time, end_time=end_time))
start_time=parse_timecode(start_time),
end_time=parse_timecode(start_time + duration)
))
output.append("{text}".format(text=item.get('text')))
if frame < len(transcript_data):
output.append('\n\n')
return '{}\n'.format(''.join(output))
class TranscriptFormatterFactory:
class TranscriptFormatterFactory(object):
"""A Transcript Class Factory
Allows for adding additional custom Transcript classes for the API
@ -139,8 +138,10 @@ class TranscriptFormatterFactory:
:rtype None
"""
if not issubclass(formatter_class, TranscriptFormatter):
raise TypeError(
f'{formatter_class} must be a subclass of TranscriptFormatter')
raise TypeError((
'{0} must be a subclass of TranscriptFormatter'
).format(formatter_class)
)
self._formatters.update({name: formatter_class})
def add_formatters(self, formatters_dict):