diff --git a/README.md b/README.md index 3725a5b..301f8fb 100644 --- a/README.md +++ b/README.md @@ -59,7 +59,7 @@ zt_rw = Zotero2Readwise( include_annotations=True, # Include Zotero annotations -> Default: True include_notes=False, # Include Zotero notes -> Default: False ) -zt_rw.run_all() +zt_rw.run() ``` ## Approach 2: diff --git a/zotero2readwise/zt2rw.py b/zotero2readwise/zt2rw.py index 791df8c..d0b4daa 100644 --- a/zotero2readwise/zt2rw.py +++ b/zotero2readwise/zt2rw.py @@ -40,8 +40,9 @@ def get_all_zotero_items(self) -> List[Dict]: print(f"{len(all_zotero_items)} Zotero items are retrieved.") return all_zotero_items - def run_all(self) -> None: - zot_annots_notes = self.get_all_zotero_items() + def run(self, zot_annots_notes: List[Dict] = None) -> None: + if zot_annots_notes is None: + zot_annots_notes = self.get_all_zotero_items() formatted_items = self.zotero.format_items(zot_annots_notes) if self.zotero.failed_items: self.zotero.save_failed_items_to_json("failed_zotero_items.json")