feature: add coloured console output

This commit is contained in:
thomasabishop 2024-11-11 16:14:01 +00:00
parent 9f9b29898d
commit d3e2abe11b
3 changed files with 7 additions and 5 deletions

View file

@ -5,7 +5,7 @@ setup(
version="0.1", version="0.1",
packages=find_packages(where="src"), packages=find_packages(where="src"),
package_dir={"": "src"}, package_dir={"": "src"},
install_requires=["python-frontmatter, "hurry.filesize"], install_requires=["python-frontmatter", "termcolor"],
entry_points={ entry_points={
"console_scripts": [ "console_scripts": [
"run=app:main", "run=app:main",

View file

@ -1,6 +1,8 @@
import sqlite3 import sqlite3
from typing import Optional from typing import Optional
from termcolor import colored
class DatabaseService: class DatabaseService:
def __init__(self, db_name): def __init__(self, db_name):
@ -14,11 +16,11 @@ class DatabaseService:
try: try:
self.connection = sqlite3.connect(f"{self.db_path}/{self.db_name}.db") self.connection = sqlite3.connect(f"{self.db_path}/{self.db_name}.db")
print("INFO Database connection established") print(colored("INFO Database connection established", "light_blue"))
return self.connection return self.connection
except Exception as e: except Exception as e:
raise Exception(f"Problem connecting to database: {e}") raise Exception(f"ERROR Problem connecting to database: {e}")
def disconnect(self) -> None: def disconnect(self) -> None:
try: try:

View file

@ -2,7 +2,7 @@ import os
from datetime import datetime from datetime import datetime
from pathlib import Path from pathlib import Path
from hurry.filesize import size from termcolor import colored
from models.entry import Entry from models.entry import Entry
from services.parse_markdown_service import ParseMarkdownService from services.parse_markdown_service import ParseMarkdownService
@ -30,7 +30,7 @@ class ParseFileService:
} }
def parse_source_directory(self): def parse_source_directory(self):
print("INFO Indexing entries in source directory") print(colored("INFO Indexing entries in source directory", "light_blue"))
parsed_entries = [] parsed_entries = []
with os.scandir(self.source_directory) as dir: with os.scandir(self.source_directory) as dir:
for file in dir: for file in dir: