#-----------------------------------------------------------------------------
# Copyright (c) 2012 - 2021, Anaconda, Inc., and Bokeh Contributors.
# All rights reserved.
#
# The full license is in the file LICENSE.txt, distributed with this software.
#-----------------------------------------------------------------------------
''' Abstract base class for subcommands that output to a file (or stdout).
'''
#-----------------------------------------------------------------------------
# Boilerplate
#-----------------------------------------------------------------------------
from __future__ import annotations
import logging # isort:skip
log = logging.getLogger(__name__)
#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------
# Standard library imports
import argparse
import sys
from abc import abstractmethod
from os.path import splitext
from typing import List, Union
# Bokeh imports
from ...document import Document
from ..subcommand import (
Arg,
Args,
Argument,
Subcommand,
)
from ..util import build_single_handler_applications, die
#-----------------------------------------------------------------------------
# Globals and constants
#-----------------------------------------------------------------------------
__all__ = (
'FileOutputSubcommand',
)
#-----------------------------------------------------------------------------
# General API
#-----------------------------------------------------------------------------
#-----------------------------------------------------------------------------
# Dev API
#-----------------------------------------------------------------------------
[docs]class FileOutputSubcommand(Subcommand):
''' Abstract subcommand to output applications as some type of file.
'''
# subtype must set this instance attribute to file extension
extension: str
[docs] @classmethod
def files_arg(cls, output_type_name: str) -> Arg:
''' Returns a positional arg for ``files`` to specify file inputs to
the command.
Subclasses should include this to their class ``args``.
Example:
.. code-block:: python
class Foo(FileOutputSubcommand):
args = (
FileOutputSubcommand.files_arg("FOO"),
# more args for Foo
) + FileOutputSubcommand.other_args()
'''
return ('files', Argument(
metavar='DIRECTORY-OR-SCRIPT',
nargs='+',
help=("The app directories or scripts to generate %s for" % (output_type_name)),
default=None
))
[docs] @classmethod
def other_args(cls) -> Args:
''' Return args for ``-o`` / ``--output`` to specify where output
should be written, and for a ``--args`` to pass on any additional
command line args to the subcommand.
Subclasses should append these to their class ``args``.
Example:
.. code-block:: python
class Foo(FileOutputSubcommand):
args = (
FileOutputSubcommand.files_arg("FOO"),
# more args for Foo
) + FileOutputSubcommand.other_args()
'''
return (
(('-o', '--output'), Argument(
metavar='FILENAME',
action='append',
type=str,
help="Name of the output file or - for standard output."
)),
('--args', Argument(
metavar='COMMAND-LINE-ARGS',
nargs="...",
help="Any command line arguments remaining are passed on to the application handler",
)),
)
[docs] def filename_from_route(self, route: str, ext: str) -> str:
'''
'''
if route == "/":
base = "index"
else:
base = route[1:]
return "%s.%s" % (base, ext)
[docs] def invoke(self, args: argparse.Namespace) -> None:
'''
'''
argvs = { f : args.args for f in args.files}
applications = build_single_handler_applications(args.files, argvs)
if args.output is None:
outputs: List[str] = []
else:
outputs = list(args.output) # copy so we can pop from it
if len(outputs) > len(applications):
die("--output/-o was given too many times (%d times for %d applications)" %
(len(outputs), len(applications)))
for (route, app) in applications.items():
doc = app.create_document()
if len(outputs) > 0:
filename = outputs.pop(0)
else:
filename = self.filename_from_route(route, self.extension)
self.write_file(args, filename, doc)
[docs] def write_file(self, args: argparse.Namespace, filename: str, doc: Document) -> None:
'''
'''
def write_str(content: str, filename: str) -> None:
if filename == "-":
print(content)
else:
with open(filename, "w", encoding="utf-8") as file:
file.write(content)
self.after_write_file(args, filename, doc)
def write_bytes(content: bytes, filename: str) -> None:
if filename == "-":
sys.stdout.buffer.write(content)
else:
with open(filename, "wb") as f:
f.write(content)
self.after_write_file(args, filename, doc)
contents = self.file_contents(args, doc)
if isinstance(contents, str):
write_str(contents, filename)
elif isinstance(contents, bytes):
write_bytes(contents, filename)
else:
if filename == "-" or len(contents) <= 1:
def indexed(i: int) -> str:
return filename
else:
def indexed(i: int) -> str:
root, ext = splitext(filename)
return f"{root}_{i}{ext}"
for i, content in enumerate(contents):
if isinstance(content, str):
write_str(content, indexed(i))
elif isinstance(content, bytes):
write_bytes(content, indexed(i))
# can be overridden optionally
[docs] def after_write_file(self, args: argparse.Namespace, filename: str, doc: Document) -> None:
'''
'''
pass
[docs] @abstractmethod
def file_contents(self, args: argparse.Namespace, doc: Document) -> Union[str, bytes, List[str], List[bytes]]:
''' Subclasses must override this method to return the contents of the output file for the given doc.
subclassed methods return different types:
str: html, json
bytes: SVG, png
Raises:
NotImplementedError
'''
raise NotImplementedError()
#-----------------------------------------------------------------------------
# Private API
#-----------------------------------------------------------------------------
#-----------------------------------------------------------------------------
# Code
#-----------------------------------------------------------------------------