diff options
-rw-r--r-- | pypaste/__init__.py | 22 | ||||
-rw-r--r-- | pypaste/__main__.py | 21 |
2 files changed, 23 insertions, 20 deletions
diff --git a/pypaste/__init__.py b/pypaste/__init__.py index e69de29..ea50c40 100644 --- a/pypaste/__init__.py +++ b/pypaste/__init__.py @@ -0,0 +1,22 @@ +import sys +from datetime import datetime + +RESET = "\x1b[0m" +RED = "\x1b[31m" +GREEN = "\x1b[32m" +YELLOW = "\x1b[33m" + + +def log_info(msg: str) -> None: + now = datetime.now().isoformat() + print(f"{GREEN}[info]{RESET} {now} {msg}", file=sys.stderr) + + +def log_warning(msg: str) -> None: + now = datetime.now().isoformat() + print(f"{YELLOW}[warning]{RESET} {now} {msg}", file=sys.stderr) + + +def log_error(msg: str) -> None: + now = datetime.now().isoformat() + print(f"{RED}[warning]{RESET} {now} {msg}", file=sys.stderr) diff --git a/pypaste/__main__.py b/pypaste/__main__.py index c9c5cff..bd098c6 100644 --- a/pypaste/__main__.py +++ b/pypaste/__main__.py @@ -21,6 +21,7 @@ import aiohttp import zstandard from pypaste import s3 from pypaste.database import Database, PasteRow +from pypaste import log_error, log_warning, log_info from socket import socket, AF_UNIX, SOCK_STREAM from hashlib import sha256 from argparse import ArgumentParser @@ -34,26 +35,6 @@ from pygments.lexers import guess_lexer, get_lexer_by_name from pygments.formatters import HtmlFormatter from pygments.styles import get_style_by_name -RESET = "\x1b[0m" -RED = "\x1b[31m" -GREEN = "\x1b[32m" -YELLOW = "\x1b[33m" - - -def log_info(msg: str) -> None: - now = datetime.now().isoformat() - print(f"{GREEN}[info]{RESET} {now} {msg}", file=sys.stderr) - - -def log_warning(msg: str) -> None: - now = datetime.now().isoformat() - print(f"{YELLOW}[warning]{RESET} {now} {msg}", file=sys.stderr) - - -def log_error(msg: str) -> None: - now = datetime.now().isoformat() - print(f"{RED}[warning]{RESET} {now} {msg}", file=sys.stderr) - def pygmentize( content: str, syntax: Optional[str], style: str, line_numbers: str | bool |