add save for different types of files
This commit is contained in:
@@ -82,7 +82,6 @@ class Transcript:
|
|||||||
speaker = seq["speaker"]
|
speaker = seq["speaker"]
|
||||||
|
|
||||||
fstring += f"{speaker}: {seq['text']}\n"
|
fstring += f"{speaker}: {seq['text']}\n"
|
||||||
|
|
||||||
return fstring
|
return fstring
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
@@ -183,6 +182,29 @@ class Transcript:
|
|||||||
def to_pdf(self, path: str) -> None:
|
def to_pdf(self, path: str) -> None:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def save(self, path: str, *args, **kwargs) -> None:
|
||||||
|
"""
|
||||||
|
Save transcript to file with given path and file format
|
||||||
|
|
||||||
|
:param path: path to save file
|
||||||
|
:type path: str
|
||||||
|
:raises ValueError: if file format is unknown
|
||||||
|
"""
|
||||||
|
if path.endswith(".json"):
|
||||||
|
self.to_json(path, *args, **kwargs)
|
||||||
|
elif path.endswith(".txt"):
|
||||||
|
self.to_txt(path, *args, **kwargs)
|
||||||
|
elif path.endswith(".md"):
|
||||||
|
self.to_md(path, *args, **kwargs)
|
||||||
|
elif path.endswith(".html"):
|
||||||
|
self.to_html(path, *args, **kwargs)
|
||||||
|
elif path.endswith(".tex"):
|
||||||
|
self.to_tex(path, *args, **kwargs)
|
||||||
|
elif path.endswith(".pdf"):
|
||||||
|
self.to_pdf(path, *args, **kwargs)
|
||||||
|
else:
|
||||||
|
raise ValueError("Unknown file format")
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
test = Transcript(json.load(open("tests/test.json", "r")))
|
test = Transcript(json.load(open("tests/test.json", "r")))
|
||||||
print(repr(test))
|
print(repr(test))
|
||||||
|
|||||||
Reference in New Issue
Block a user