hotfolder.py

#!/usr/bin/python

# -*- coding: utf-8 -*-

import click
import sys
import os
import time
from joplin import joplinapi


@click.command()
@click.option(
    "-n",
    "--notebook",
    "notebook",
    default="Import",
    show_default=True,
    help="""Specify the notebook in which to place newly created notes."""
    """ Specified notebook must exist or program will exit.""",
)
@click.option(
    "-t",
    "--token",
    "token",
    required=False,
    help="""Specify the Joplin API token.""",
)
@click.option(
    "--tag",
    "add_tag",
    required=False,
    help="""Specify Tags to add to the note. Comma separated for multiple tags.""",
)
@click.option(
    "-p",
    "--path",
    "path",
    required=True,
    help="""Specify the folder for monitoring.""",
)
@click.option(
    "--as-plain",
    "plain",
    required=False,
    help="""Specify additional file extensions comma separated for input as text.""",
)
@click.option(
    "-u",
    "--url",
    "url",
    required=False,
    default="http://localhost:41184",
    show_default=True,
    help="""Specify the Joplin web clipper URL.""",
)
@click.option(
    "--preview/--no-preview",
    "preview",
    required=False,
    default=False,
    show_default=True,
    help="""Create a preview of the first site from an PDF file.""",
)
def Main(path, notebook, token, url, plain, add_tag, preview):
    if not os.path.exists(path):
        print("Path does not exist")

    if token is not None:
        joplinapi.SetEndpoint(url, token)
    elif joplinapi.LoadEndpoint() == False:
        joplinapi.SetEndpoint(url, token)

    while joplinapi.Ping() == False:
        print("Wait for Joplin")
        time.sleep(10)
    
    if plain is not None:
        plain = plain.replace(", ", ",")
        plain = plain.split(",")

    notebook_id = joplinapi.GetNotebookID(notebook)

    if notebook_id == False:
        print("Notebook not found")
        sys.exit(1)

    if add_tag is not None:
        add_tag = add_tag.replace(", ", ",")
        add_tag = add_tag.split(",")

    WatchFolder(path, notebook_id, plain, add_tag, preview)


def WatchFolder(path, notebook_id, plain, add_tags, preview):
    files = dict()
    while 1:
        while joplinapi.Ping() == False:
            print("Wait for Joplin")
            time.sleep(10)

        # Add files and process
        try:
            file_list = os.listdir(path)
        except:
            print("Path not found")
            time.sleep(10)
            continue
        
        for file in file_list:
            file_path = os.path.join(path, file)
            
            if file.find(".lock") > 0:
                continue
            
            if not file in files:
                files[file] = os.path.getsize(file_path)
            elif os.path.getsize(file_path) == files[file] and not os.path.exists(file_path + ".lock"):
                f = open(file_path + ".lock", 'w')
                f.close()

                print("Add to Joplin: " + file)
                note_id = joplinapi.CreateNoteWithFile(
                    os.path.join(path, file), notebook_id, plain, preview)
                if note_id != False:
                    if add_tags is not None:
                        for tag in add_tags:
                            joplinapi.AddTagToNote(tag, note_id, True)
                    print("Joplin upload completed")
                    try:
                        os.remove(file_path)
                        os.remove(file_path + ".lock")
                        files.pop(file)
                    except:
                        print("File remove failed: " + file)
                        files[file] = -2
                else:
                    files[file] = -1
                print("")
            elif files[file] >= 0:
                files[file] = os.path.getsize(file_path)

        # Remove orphan entrys
        check = files.copy()
        for file in check:
            if not os.path.exists(os.path.join(path, file)):
                print("Removed orphan: " + file)
                files.pop(file)

        time.sleep(10)


if __name__ == "__main__":
    Main()