create_jupyter_service.py

import argparse
import getpass
import os


JUPYTER_SERVICE_TEMPLATE = """
[Unit]
Description=Jupyter Notebook Service

[Service]
Type=simple
User=%s
ExecStart=/bin/sh -c "jupyter lab --ip=0.0.0.0 --no-browser"
WorkingDirectory=%s
Restart=always

[Install]
WantedBy=multi-user.target
"""


JUPYTER_SERVICE_NAME = 'ugv_jupyter'


def get_jupyter_service(working_directory):
    assert(os.path.isdir(working_directory))
    service_str = JUPYTER_SERVICE_TEMPLATE % (getpass.getuser(), working_directory)
    return service_str


if __name__ == '__main__':

    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--working_directory',
        type=str,
        help='The directory for Jupyter Lab',
        default=os.path.expanduser('~'))
    parser.add_argument('--output', default='ugv_jupyter.service')
    args = parser.parse_args()

    with open(args.output, 'w') as f:
        f.write(get_jupyter_service(args.working_directory))