mirror of
https://github.com/jrnl-org/jrnl.git
synced 2025-05-12 09:28:31 +02:00
3ddfb4d594
- Implement mock editor fixture - Add fixture to keep track of editor state - Implement various steps to check editor state Co-authored-by: Micah Jerome Ellison <micah.jerome.ellison@gmail.com>
62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
import logging
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import textwrap
|
|
from pathlib import Path
|
|
|
|
from .color import ERROR_COLOR
|
|
from .color import RESET_COLOR
|
|
from .os_compat import on_windows
|
|
from .os_compat import split_args
|
|
|
|
|
|
def get_text_from_editor(config, template=""):
|
|
suffix = ".jrnl"
|
|
if config["template"]:
|
|
template_filename = Path(config["template"]).name
|
|
suffix = "-" + template_filename
|
|
filehandle, tmpfile = tempfile.mkstemp(prefix="jrnl", text=True, suffix=suffix)
|
|
os.close(filehandle)
|
|
|
|
with open(tmpfile, "w", encoding="utf-8") as f:
|
|
if template:
|
|
f.write(template)
|
|
|
|
try:
|
|
subprocess.call(split_args(config["editor"]) + [tmpfile])
|
|
except FileNotFoundError as e:
|
|
error_msg = f"""
|
|
{ERROR_COLOR}{str(e)}{RESET_COLOR}
|
|
|
|
Please check the 'editor' key in your config file for errors:
|
|
{repr(config['editor'])}
|
|
"""
|
|
print(textwrap.dedent(error_msg).strip(), file=sys.stderr)
|
|
exit(1)
|
|
|
|
with open(tmpfile, "r", encoding="utf-8") as f:
|
|
raw = f.read()
|
|
os.remove(tmpfile)
|
|
|
|
if not raw:
|
|
print("[Nothing saved to file]", file=sys.stderr)
|
|
|
|
return raw
|
|
|
|
|
|
def get_text_from_stdin():
|
|
_how_to_quit = "Ctrl+z and then Enter" if on_windows() else "Ctrl+d"
|
|
print(
|
|
f"[Writing Entry; on a blank line, press {_how_to_quit} to finish writing]\n",
|
|
file=sys.stderr,
|
|
)
|
|
try:
|
|
raw = sys.stdin.read()
|
|
except KeyboardInterrupt:
|
|
logging.error("Write mode: keyboard interrupt")
|
|
print("[Entry NOT saved to journal]", file=sys.stderr)
|
|
sys.exit(0)
|
|
|
|
return raw
|