diff --git a/utils/Sync/kazimir/Marker.py b/utils/Sync/kazimir/Marker.py index bd0d40f1..14cc3982 100644 --- a/utils/Sync/kazimir/Marker.py +++ b/utils/Sync/kazimir/Marker.py @@ -4,7 +4,7 @@ from markdown import markdown from kazimir.InstagramToken import InstagramToken from kazimir.YoutubeToken import YoutubeToken from kazimir.Token import SplitToken, ImageToken, UrlToken, FileToken, TextToken, DocumentToken, TokenFactory, \ - DocumentTokenFactory + BuildTokenFactory class Marker: @@ -182,7 +182,7 @@ More text m.add_token_factory(TokenFactory(InstagramToken)) m.add_token_factory(TokenFactory(UrlToken)) m.add_token_factory(TokenFactory(ImageToken)) - m.add_token_factory(DocumentTokenFactory(build=build_document)) + m.add_token_factory(BuildTokenFactory(build=build_document)) m.add_token_factory(TokenFactory(FileToken)) m.add_token_factory(TokenFactory(TextToken)) m.add_tree_middleware(fix_links_quotes) diff --git a/utils/Sync/kazimir/Token.py b/utils/Sync/kazimir/Token.py index 1ecd33e1..ed54d6da 100644 --- a/utils/Sync/kazimir/Token.py +++ b/utils/Sync/kazimir/Token.py @@ -16,9 +16,9 @@ class TokenFactory: return self.token(data=data) -class DocumentTokenFactory(TokenFactory): - def __init__(self, build) -> None: - super().__init__(DocumentToken) +class BuildTokenFactory(TokenFactory): + def __init__(self, token, build) -> None: + super().__init__(token) self.build = build def create(self, data):