move logging functions into __init__.py

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