Simple image server in Python, for use in Jupyter Notebook, markdown, local website. Can also be adapted for viewing PDF with pdf.js

image_server.py

import subprocess
from pathlib import Path
from datetime import datetime
from PIL import ImageGrab
import atexit
from urllib.parse import quote


class Server:
    def __init__(self, root, port=14074):
        self.root = root
        self.port = port

        self.server = subprocess.Popen([
            'python', '-m',
            'http.server',
            '-d', self.root,
            str(self.port)
        ])

        atexit.register(self.close)

    def close(self):
        self.server.kill()

    def paste(self, filename=None):
        if not filename:
            filename = 'blob/' + datetime.now().strftime('%Y%m%d_%H.%M.%S') + '.png'

        filename = Path(filename).with_suffix('.png')

        im = ImageGrab.grabclipboard()
        if im:
            true_output = Path(self.root).joinpath(filename)
            true_output.parent.mkdir(parents=True, exist_ok=True)
            im.save(true_output)

            print('Pasted as "{}"'
                  .format(f'http://localhost:{self.port}/{quote(str(filename))}'))