Monkeytype anki/rsbackend.py

This commit is contained in:
Alan Du 2020-02-27 19:05:06 -05:00
parent 02dd30f2a0
commit b451f4e3f2

View file

@ -178,7 +178,9 @@ def proto_progress_to_native(progress: pb.Progress) -> Progress:
class RustBackend: class RustBackend:
def __init__(self, col_path: str, media_folder_path: str, media_db_path: str): def __init__(
self, col_path: str, media_folder_path: str, media_db_path: str
) -> None:
ftl_folder = os.path.join(anki.lang.locale_folder, "fluent") ftl_folder = os.path.join(anki.lang.locale_folder, "fluent")
init_msg = pb.BackendInit( init_msg = pb.BackendInit(
collection_path=col_path, collection_path=col_path,
@ -340,7 +342,7 @@ class RustBackend:
) )
).format_time_span ).format_time_span
def studied_today(self, cards: int, seconds: float,) -> str: def studied_today(self, cards: int, seconds: float) -> str:
return self._run_command( return self._run_command(
pb.BackendInput( pb.BackendInput(
studied_today=pb.StudiedTodayIn(cards=cards, seconds=seconds) studied_today=pb.StudiedTodayIn(cards=cards, seconds=seconds)
@ -376,7 +378,7 @@ class I18nBackend:
) )
self._backend = ankirspy.open_i18n(init_msg.SerializeToString()) self._backend = ankirspy.open_i18n(init_msg.SerializeToString())
def translate(self, key: TR, **kwargs: Union[str, int, float]): def translate(self, key: TR, **kwargs: Union[str, int, float]) -> str:
return self._backend.translate( return self._backend.translate(
translate_string_in(key, **kwargs).SerializeToString() translate_string_in(key, **kwargs).SerializeToString()
) )