main.py

"""
This add-on allows you to bulk export notes from DocumentCloud
"""

import zipfile
from documentcloud.addon import AddOn


class NoteExport(AddOn):
    """Export all of the selected documents' notes as text in a zip file"""

    def main(self):
        """Export all notes as a zip file"""
        with zipfile.ZipFile("export.zip", mode="w") as archive:
            for document in self.client.documents.list(id__in=self.documents):
                with archive.open(
                    f"{document.slug} - {document.id} - notes.txt", "w"
                ) as notes_file:
                    for note in document.notes:
                        notes_file.write(f"Page: {note.page+1}\n".encode("utf8"))
                        notes_file.write(f"{note.title}\n".encode("utf8"))
                        notes_file.write(f"{note.content}\n".encode("utf8"))
                        notes_file.write(f"Access level: {note.access}\n\n".encode(
                            "utf8"
                        ))

        self.upload_file(open("export.zip", encoding="utf-8"))


if __name__ == "__main__":
    NoteExport().main()