70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
"""
|
|
Contains shared functions for the various CMM subcommands.
|
|
"""
|
|
|
|
import argparse
|
|
from pathlib import Path
|
|
from ..message import Message, MessageError, source_code
|
|
|
|
|
|
def read_text_file(file: Path) -> str:
|
|
with open(file) as r:
|
|
content = r.read().strip()
|
|
return content
|
|
|
|
|
|
def add_file_as_text(question_parts: list[str], file: str) -> None:
|
|
"""
|
|
Add the given file as plain text to the question part list.
|
|
If the file is a Message, add the answer.
|
|
"""
|
|
file_path = Path(file)
|
|
content: str
|
|
try:
|
|
message = Message.from_file(file_path)
|
|
if message and message.answer:
|
|
content = message.answer
|
|
except MessageError:
|
|
content = read_text_file(Path(file))
|
|
if len(content) > 0:
|
|
question_parts.append(content)
|
|
|
|
|
|
def add_file_as_code(question_parts: list[str], file: str) -> None:
|
|
"""
|
|
Add all source code from the given file. If no code segments can be extracted,
|
|
the whole content is added as source code segment. If the file is a Message,
|
|
extract the source code from the answer.
|
|
"""
|
|
file_path = Path(file)
|
|
content: str
|
|
try:
|
|
message = Message.from_file(file_path)
|
|
if message and message.answer:
|
|
content = message.answer
|
|
except MessageError:
|
|
with open(file) as r:
|
|
content = r.read().strip()
|
|
# extract and add source code
|
|
code_parts = source_code(content, include_delims=True)
|
|
if len(code_parts) > 0:
|
|
question_parts += code_parts
|
|
else:
|
|
question_parts.append(f"```\n{content}\n```")
|
|
|
|
|
|
def invert_input_tag_args(args: argparse.Namespace) -> None:
|
|
"""
|
|
Changes the semantics of the INPUT tags for this command:
|
|
* not tags specified on the CLI -> no tags are selected
|
|
* empty tags specified on the CLI -> all tags are selected
|
|
"""
|
|
if args.or_tags is None:
|
|
args.or_tags = set()
|
|
elif len(args.or_tags) == 0:
|
|
args.or_tags = None
|
|
if args.and_tags is None:
|
|
args.and_tags = set()
|
|
elif len(args.and_tags) == 0:
|
|
args.and_tags = None
|