diff --git a/sijapi/__init__.py b/sijapi/__init__.py index 0aeb2e8..26ed05f 100644 --- a/sijapi/__init__.py +++ b/sijapi/__init__.py @@ -10,32 +10,32 @@ from dateutil import tz from pathlib import Path from pydantic import BaseModel from typing import List, Optional -from .logs import Logger from .classes import AutoResponder, IMAPConfig, SMTPConfig, EmailAccount, EmailContact, IncomingEmail, Database, Geocoder, APIConfig, Configuration +from .logs import Logger -### Initial initialization +# INITIALization BASE_DIR = Path(__file__).resolve().parent CONFIG_DIR = BASE_DIR / "config" ENV_PATH = CONFIG_DIR / ".env" -LOGS_DIR = BASE_DIR / "logs" - -L = Logger("Central", LOGS_DIR) -os.makedirs(LOGS_DIR, exist_ok=True) load_dotenv(ENV_PATH) +LOGS_DIR = BASE_DIR / "logs" +os.makedirs(LOGS_DIR, exist_ok=True) +L = Logger("Central", LOGS_DIR) -### API essentials +# API essentials API = APIConfig.load('api', 'secrets') Dir = Configuration.load('dirs') HOST = f"{API.BIND}:{API.PORT}" LOCAL_HOSTS = [ipaddress.ip_address(localhost.strip()) for localhost in os.getenv('LOCAL_HOSTS', '127.0.0.1').split(',')] + ['localhost'] SUBNET_BROADCAST = os.getenv("SUBNET_BROADCAST", '10.255.255.255') MAX_CPU_CORES = min(int(os.getenv("MAX_CPU_CORES", int(multiprocessing.cpu_count()/2))), multiprocessing.cpu_count()) + DB = Database.from_env() News = Configuration.load('news', 'secrets') IMG = Configuration.load('img', 'secrets') -### Directories & general paths +# Directories & general paths ROUTER_DIR = BASE_DIR / "routers" DATA_DIR = BASE_DIR / "data" os.makedirs(DATA_DIR, exist_ok=True) @@ -45,12 +45,10 @@ REQUESTS_DIR = LOGS_DIR / "requests" os.makedirs(REQUESTS_DIR, exist_ok=True) REQUESTS_LOG_PATH = LOGS_DIR / "requests.log" -### LOCATE AND WEATHER LOCALIZATIONS +# LOCATE AND WEATHER LOCALIZATIONS USER_FULLNAME = os.getenv('USER_FULLNAME') USER_BIO = os.getenv('USER_BIO') -HOME_ZIP = os.getenv("HOME_ZIP") # unimplemented NAMED_LOCATIONS = CONFIG_DIR / "named-locations.yaml" -# DB = DATA_DIR / "weatherlocate.db" # deprecated VISUALCROSSING_BASE_URL = os.getenv("VISUALCROSSING_BASE_URL", "https://weather.visualcrossing.com/VisualCrossingWebServices/rest/services/timeline") VISUALCROSSING_API_KEY = os.getenv("VISUALCROSSING_API_KEY") GEONAMES_TXT = DATA_DIR / "geonames.txt" @@ -59,7 +57,7 @@ TZ = tz.gettz(os.getenv("TZ", "America/Los_Angeles")) TZ_CACHE = DATA_DIR / "tzcache.json" GEO = Geocoder(NAMED_LOCATIONS, TZ_CACHE) -### Obsidian & notes +# Obsidian & notes ALLOWED_FILENAME_CHARS = r'[^\w \.-]' MAX_PATH_LENGTH = 254 OBSIDIAN_VAULT_DIR = Path(os.getenv("OBSIDIAN_BASE_DIR") or Path(Dir.HOME) / "Nextcloud" / "notes") @@ -74,13 +72,13 @@ os.makedirs(ARCHIVE_DIR, exist_ok=True) DOC_DIR = DATA_DIR / "docs" os.makedirs(DOC_DIR, exist_ok=True) -### DATETIME SCHEMA FOR DAILY NOTE FOLDER HIERARCHY FORMATTING ### +# DATETIME SCHEMA FOR DAILY NOTE FOLDER HIERARCHY FORMATTING ### YEAR_FMT = os.getenv("YEAR_FMT") MONTH_FMT = os.getenv("MONTH_FMT") DAY_FMT = os.getenv("DAY_FMT") DAY_SHORT_FMT = os.getenv("DAY_SHORT_FMT") -### Large language model +# Large language model LLM_URL = os.getenv("LLM_URL", "http://localhost:11434") LLM_SYS_MSG = os.getenv("SYSTEM_MSG", "You are a helpful AI assistant.") DEFAULT_LLM = os.getenv("DEFAULT_LLM", "llama3") @@ -88,7 +86,8 @@ DEFAULT_VISION = os.getenv("DEFAULT_VISION", "llava") DEFAULT_VOICE = os.getenv("DEFAULT_VOICE", "Luna") DEFAULT_11L_VOICE = os.getenv("DEFAULT_11L_VOICE", "Victoria") OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") -### Summarization + +# Summarization SUMMARY_CHUNK_SIZE = int(os.getenv("SUMMARY_CHUNK_SIZE", 16384)) # measured in tokens SUMMARY_CHUNK_OVERLAP = int(os.getenv("SUMMARY_CHUNK_OVERLAP", 256)) # measured in tokens SUMMARY_TPW = float(os.getenv("SUMMARY_TPW", 1.3)) # measured in tokens @@ -100,7 +99,7 @@ SUMMARY_INSTRUCT = os.getenv('SUMMARY_INSTRUCT', "You are an AI assistant that p SUMMARY_INSTRUCT_TTS = os.getenv('SUMMARY_INSTRUCT_TTS', "You are an AI assistant that provides email summaries for Sanjay. Your response will undergo Text-To-Speech conversion and added to Sanjay's private podcast. Providing adequate context (Sanjay did not send this question to you, he will only hear your response) but aiming for conciseness and precision, and bearing in mind the Text-To-Speech conversion (avoiding acronyms and formalities), summarize the following email.") -### Stable diffusion +# Stable diffusion IMG_DIR = DATA_DIR / "img" / "images" os.makedirs(IMG_DIR, exist_ok=True) IMG_WORKFLOWS_DIR = DATA_DIR / "img" / "workflows" @@ -111,13 +110,13 @@ COMFYUI_OUTPUT_DIR = COMFYUI_DIR / 'output' COMFYUI_LAUNCH_CMD = os.getenv('COMFYUI_LAUNCH_CMD', 'mamba activate comfyui && python main.py') IMG_CONFIG_PATH = CONFIG_DIR / 'img.yaml' -### ASR +# ASR ASR_DIR = DATA_DIR / "asr" os.makedirs(ASR_DIR, exist_ok=True) WHISPER_CPP_DIR = Path(Dir.HOME) / str(os.getenv("WHISPER_CPP_DIR")) WHISPER_CPP_MODELS = os.getenv('WHISPER_CPP_MODELS', 'NULL,VOID').split(',') -### TTS +# TTS PREFERRED_TTS = os.getenv("PREFERRED_TTS", "None") TTS_DIR = DATA_DIR / "tts" os.makedirs(TTS_DIR, exist_ok=True) @@ -132,7 +131,7 @@ os.makedirs(TTS_SEGMENTS_DIR, exist_ok=True) ELEVENLABS_API_KEY = os.getenv("ELEVENLABS_API_KEY") -### Calendar & email account +# Calendar & email account MS365_TOGGLE = True if os.getenv("MS365_TOGGLE") == "True" else False ICAL_TOGGLE = True if os.getenv("ICAL_TOGGLE") == "True" else False ICS_PATH = DATA_DIR / 'calendar.ics' # deprecated now, but maybe revive? @@ -142,7 +141,7 @@ EMAIL_CONFIG = CONFIG_DIR / "email.yaml" EMAIL_LOGS = LOGS_DIR / "email" os.makedirs(EMAIL_LOGS, exist_ok = True) -### Courtlistener & other webhooks +# Courtlistener & other webhooks COURTLISTENER_DOCKETS_DIR = DATA_DIR / "courtlistener" / "dockets" os.makedirs(COURTLISTENER_DOCKETS_DIR, exist_ok=True) COURTLISTENER_SEARCH_DIR = DATA_DIR / "courtlistener" / "cases" @@ -152,7 +151,7 @@ COURTLISTENER_API_KEY = os.getenv("COURTLISTENER_API_KEY") COURTLISTENER_BASE_URL = os.getenv("COURTLISTENER_BASE_URL", "https://www.courtlistener.com") COURTLISTENER_DOCKETS_URL = "https://www.courtlistener.com/api/rest/v3/dockets/" -### Keys & passwords +# Keys & passwords PUBLIC_KEY_FILE = os.getenv("PUBLIC_KEY_FILE", 'you_public_key.asc') PUBLIC_KEY = (BASE_DIR.parent / PUBLIC_KEY_FILE).read_text() MAC_ID = os.getenv("MAC_ID") @@ -164,27 +163,26 @@ PHOTOPRISM_URL = os.getenv("PHOTOPRISM_URL") PHOTOPRISM_USER = os.getenv("PHOTOPRISM_USER") PHOTOPRISM_PASS = os.getenv("PHOTOPRISM_PASS") -### Tailscale +# Tailscale TS_IP = ipaddress.ip_address(os.getenv("TS_IP", "NULL")) TS_SUBNET = ipaddress.ip_network(os.getenv("TS_SUBNET")) if os.getenv("TS_SUBNET") else None TS_ID = os.getenv("TS_ID", "NULL") TS_TAILNET = os.getenv("TS_TAILNET", "NULL") TS_ADDRESS = f"http://{TS_ID}.{TS_TAILNET}.ts.net" -### Cloudflare +# Cloudflare CF_API_BASE_URL = os.getenv("CF_API_BASE_URL") CF_TOKEN = os.getenv("CF_TOKEN") CF_IP = DATA_DIR / "cf_ip.txt" # to be deprecated soon CF_DOMAINS_PATH = DATA_DIR / "cf_domains.json" # to be deprecated soon -### Caddy - not fully implemented +# Caddy - not fully implemented API.URL = os.getenv("API.URL") CADDY_SERVER = os.getenv('CADDY_SERVER', None) CADDYFILE_PATH = os.getenv("CADDYFILE_PATH", "") if CADDY_SERVER is not None else None CADDY_API_KEY = os.getenv("CADDY_API_KEY") - -### Microsoft Graph +# Microsoft Graph MS365_CLIENT_ID = os.getenv('MS365_CLIENT_ID') MS365_SECRET = os.getenv('MS365_SECRET') MS365_TENANT_ID = os.getenv('MS365_TENANT_ID') @@ -199,6 +197,7 @@ MS365_AUTHORITY_URL = f"{MS365_LOGIN_URL}/{MS365_TENANT_ID}" MS365_REDIRECT_PATH = os.getenv("MS365_REDIRECT_PATH", "https://api.sij.ai/o365/oauth_redirect") MS365_SCOPE = os.getenv("MS365_SCOPE", 'Calendars.Read,Calendars.ReadWrite,offline_access').split(',') -### Maintenance + +# Maintenance GARBAGE_COLLECTION_INTERVAL = 60 * 60 # Run cleanup every hour GARBAGE_TTL = 60 * 60 * 24 # Delete files older than 24 hours \ No newline at end of file diff --git a/sijapi/__main__.py b/sijapi/__main__.py index c5ef1bb..3f41e30 100755 --- a/sijapi/__main__.py +++ b/sijapi/__main__.py @@ -19,23 +19,20 @@ from dotenv import load_dotenv from pathlib import Path from datetime import datetime import argparse +from . import L, API, ROUTER_DIR parser = argparse.ArgumentParser(description='Personal API.') -parser.add_argument('--debug', action='store_true', help='Set log level to L.INFO') +parser.add_argument('--log', type=str, default='INFO', help='Set overall log level (e.g., DEBUG, INFO, WARNING)') +parser.add_argument('--debug', nargs='+', default=[], help='Set DEBUG log level for specific modules') parser.add_argument('--test', type=str, help='Load only the specified module.') args = parser.parse_args() -from . import L, API, ROUTER_DIR L.setup_from_args(args) - -from sijapi import ROUTER_DIR - -# Initialize a FastAPI application -api = FastAPI() +print(f"Debug modules after setup: {L.debug_modules}") # Debug print -# CORSMiddleware -api.add_middleware( +app = FastAPI() +app.add_middleware( CORSMiddleware, allow_origins=['*'], allow_credentials=True, @@ -67,31 +64,16 @@ class SimpleAPIKeyMiddleware(BaseHTTPMiddleware): return response # Add the middleware to your FastAPI app -api.add_middleware(SimpleAPIKeyMiddleware) +app.add_middleware(SimpleAPIKeyMiddleware) - -canceled_middleware = """ -@api.middleware("http") -async def log_requests(request: Request, call_next): - L.DEBUG(f"Incoming request: {request.method} {request.url}") - L.DEBUG(f"Request headers: {request.headers}") - L.DEBUG(f"Request body: {await request.body()}") - response = await call_next(request) - return response - -async def log_outgoing_request(request): - L.INFO(f"Outgoing request: {request.method} {request.url}") - L.DEBUG(f"Request headers: {request.headers}") - L.DEBUG(f"Request body: {request.content}") -""" - -@api.exception_handler(HTTPException) +@app.exception_handler(HTTPException) async def http_exception_handler(request: Request, exc: HTTPException): L.ERR(f"HTTP Exception: {exc.status_code} - {exc.detail}") L.ERR(f"Request: {request.method} {request.url}") return JSONResponse(status_code=exc.status_code, content={"detail": exc.detail}) -@api.middleware("http") + +@app.middleware("http") async def handle_exception_middleware(request: Request, call_next): try: response = await call_next(request) @@ -104,37 +86,35 @@ async def handle_exception_middleware(request: Request, call_next): return response - - def load_router(router_name): router_file = ROUTER_DIR / f'{router_name}.py' - L.DEBUG(f"Attempting to load {router_name.capitalize()}...") + module_logger = L.get_module_logger(router_name) + module_logger.debug(f"Attempting to load {router_name.capitalize()}...") if router_file.exists(): module_path = f'sijapi.routers.{router_name}' try: module = importlib.import_module(module_path) router = getattr(module, router_name) - api.include_router(router) - L.INFO(f"{router_name.capitalize()} router loaded.") + app.include_router(router) + # module_logger.info(f"{router_name.capitalize()} router loaded.") except (ImportError, AttributeError) as e: - L.CRIT(f"Failed to load router {router_name}: {e}") + module_logger.critical(f"Failed to load router {router_name}: {e}") else: - L.ERR(f"Router file for {router_name} does not exist.") + module_logger.error(f"Router file for {router_name} does not exist.") def main(argv): if args.test: load_router(args.test) else: - L.CRIT(f"sijapi launched") - L.CRIT(f"{args._get_args}") + L.logger.critical(f"sijapi launched") + L.logger.critical(f"Arguments: {args}") for module_name in API.MODULES.__fields__: if getattr(API.MODULES, module_name): load_router(module_name) - config = HypercornConfig() - config.bind = [API.BIND] # Use the resolved BIND value - asyncio.run(serve(api, config)) + config.bind = [API.BIND] + asyncio.run(serve(app, config)) if __name__ == "__main__": main(sys.argv[1:]) \ No newline at end of file diff --git a/sijapi/classes.py b/sijapi/classes.py index 1dd34c3..e634a45 100644 --- a/sijapi/classes.py +++ b/sijapi/classes.py @@ -23,7 +23,6 @@ from timezonefinder import TimezoneFinder T = TypeVar('T', bound='Configuration') - import os from pathlib import Path from typing import Union, Optional, Any, Dict, List diff --git a/sijapi/logs.py b/sijapi/logs.py index 4c97f1a..e852c5c 100644 --- a/sijapi/logs.py +++ b/sijapi/logs.py @@ -1,99 +1,42 @@ -# logs.py import os import sys from loguru import logger -import traceback class Logger: def __init__(self, name, logs_dir): self.logs_dir = logs_dir self.name = name - self.logger = logger.bind(name=name) + self.logger = logger + self.debug_modules = set() def setup_from_args(self, args): if not os.path.exists(self.logs_dir): os.makedirs(self.logs_dir) - # Remove default logger - logger.remove() - + self.logger.remove() + + log_format = "{time:YYYY-MM-DD HH:mm:ss} - {name} - {level: <8} - {message}" + # File handler - handler_path = os.path.join(self.logs_dir, 'app.log') - logger.add(handler_path, rotation="2 MB", compression="zip", level="DEBUG", format="{time:YYYY-MM-DD HH:mm:ss} - {name} - {level} - {message}") + self.logger.add(os.path.join(self.logs_dir, 'app.log'), rotation="2 MB", level="DEBUG", format=log_format) - # Console handler - log_format = ( - "{time:YYYY-MM-DD HH:mm:ss} - " - "{name} - " - "{level: <8} - " - "" - "{message}" - "" - ) - console_level = "DEBUG" if args.debug else "INFO" - logger.add( - sys.stdout, - format=log_format, - level=console_level, - colorize=True, - filter=lambda record: record["level"].name != "INFO", # Apply colors to all levels except INFO - ) + # Set debug modules + self.debug_modules = set(args.debug) + + # Console handler with custom filter + def module_filter(record): + return (record["level"].no >= logger.level(args.log.upper()).no or + record["name"] in self.debug_modules) + + self.logger.add(sys.stdout, level="DEBUG", format=log_format, filter=module_filter, colorize=True) - # Add a separate handler for INFO level without colors - logger.add( - sys.stdout, - format="{time:YYYY-MM-DD HH:mm:ss} - {name} - {level: <8} - {message}", - level="INFO", - filter=lambda record: record["level"].name == "INFO", - ) - - # Custom color and style mappings - logger.level("CRITICAL", color="") - logger.level("ERROR", color="") - logger.level("WARNING", color="") - logger.level("DEBUG", color="") + self.logger.level("CRITICAL", color="") + self.logger.level("ERROR", color="") + self.logger.level("WARNING", color="") + self.logger.level("DEBUG", color="") + + self.logger.info(f"Debug modules: {self.debug_modules}") - # Test color output - self.logger.debug("Debug message (should be italic green)") - self.logger.info("Info message (should be uncolored)") - self.logger.warning("Warning message (should be bold orange/yellow)") - self.logger.error("Error message (should be bold red)") - self.logger.critical("Critical message (should be bold yellow on magenta)") - - - def DEBUG(self, log_message): self.logger.debug(log_message) - def INFO(self, log_message): self.logger.info(log_message) - def WARN(self, log_message): self.logger.warning(log_message) - def ERR(self, log_message): - self.logger.error(log_message) - self.logger.error(traceback.format_exc()) - def CRIT(self, log_message): - self.logger.critical(log_message) - self.logger.critical(traceback.format_exc()) - - def get_logger(self): - return self - - -if __name__ == "__main__": - import argparse - parser = argparse.ArgumentParser() - parser.add_argument('--debug', action='store_true') - args = parser.parse_args() - - logger_instance = Logger("test", "logs") - logger_instance.setup_from_args(args) - test_logger = logger_instance.get_logger() - - print("FORCE_COLOR:", os.environ.get('FORCE_COLOR')) - print("NO_COLOR:", os.environ.get('NO_COLOR')) - print("TERM:", os.environ.get('TERM')) - print("PYCHARM_HOSTED:", os.environ.get('PYCHARM_HOSTED')) - print("PYTHONIOENCODING:", os.environ.get('PYTHONIOENCODING')) - - test_logger.debug("This is a debug message") - test_logger.info("This is an info message") - test_logger.warning("This is a warning message") - test_logger.error("This is an error message") - test_logger.critical("This is a critical message") + def get_module_logger(self, module_name): + return self.logger.bind(name=module_name) diff --git a/sijapi/routers/asr.py b/sijapi/routers/asr.py index 399ab15..ea4d23c 100644 --- a/sijapi/routers/asr.py +++ b/sijapi/routers/asr.py @@ -14,10 +14,10 @@ from fastapi import APIRouter, HTTPException, Form, UploadFile, File, Background from fastapi.responses import JSONResponse from pydantic import BaseModel, Field from typing import Optional - from sijapi import L, ASR_DIR, WHISPER_CPP_MODELS, GARBAGE_COLLECTION_INTERVAL, GARBAGE_TTL, WHISPER_CPP_DIR, MAX_CPU_CORES asr = APIRouter() +logger = L.get_module_logger("asr") class TranscribeParams(BaseModel): model: str = Field(default="small") @@ -81,7 +81,7 @@ async def transcribe_endpoint( return JSONResponse(content={"status": "timeout", "message": "Transcription is taking longer than expected. Please check back later."}, status_code=202) async def transcribe_audio(file_path, params: TranscribeParams): - L.DEBUG(f"Transcribing audio file from {file_path}...") + logger.debug(f"Transcribing audio file from {file_path}...") file_path = await convert_to_wav(file_path) model = params.model if params.model in WHISPER_CPP_MODELS else 'small' model_path = WHISPER_CPP_DIR / 'models' / f'ggml-{model}.bin' @@ -119,11 +119,11 @@ async def transcribe_audio(file_path, params: TranscribeParams): command.extend(['--dtw', params.dtw]) command.extend(['-f', file_path]) - L.DEBUG(f"Command: {command}") + logger.debug(f"Command: {command}") # Create a unique ID for this transcription job job_id = str(uuid.uuid4()) - L.DEBUG(f"Created job ID: {job_id}") + logger.debug(f"Created job ID: {job_id}") # Store the job status transcription_results[job_id] = {"status": "processing", "result": None} @@ -135,20 +135,20 @@ async def transcribe_audio(file_path, params: TranscribeParams): poll_interval = 1 # 1 second start_time = asyncio.get_event_loop().time() - L.DEBUG(f"Starting to poll for job {job_id}") + logger.debug(f"Starting to poll for job {job_id}") try: while asyncio.get_event_loop().time() - start_time < max_wait_time: job_status = transcription_results.get(job_id, {}) - L.DEBUG(f"Current status for job {job_id}: {job_status['status']}") + logger.debug(f"Current status for job {job_id}: {job_status['status']}") if job_status["status"] == "completed": - L.INFO(f"Transcription completed for job {job_id}") + logger.info(f"Transcription completed for job {job_id}") return job_status["result"] elif job_status["status"] == "failed": - L.ERR(f"Transcription failed for job {job_id}: {job_status.get('error', 'Unknown error')}") + logger.error(f"Transcription failed for job {job_id}: {job_status.get('error', 'Unknown error')}") raise Exception(f"Transcription failed: {job_status.get('error', 'Unknown error')}") await asyncio.sleep(poll_interval) - L.ERR(f"Transcription timed out for job {job_id}") + logger.error(f"Transcription timed out for job {job_id}") raise TimeoutError("Transcription timed out") finally: # Ensure the task is cancelled if we exit the loop @@ -159,20 +159,20 @@ async def transcribe_audio(file_path, params: TranscribeParams): async def process_transcription(command, file_path, job_id): try: - L.DEBUG(f"Starting transcription process for job {job_id}") + logger.debug(f"Starting transcription process for job {job_id}") result = await run_transcription(command, file_path) transcription_results[job_id] = {"status": "completed", "result": result} - L.DEBUG(f"Transcription completed for job {job_id}") + logger.debug(f"Transcription completed for job {job_id}") except Exception as e: - L.ERR(f"Transcription failed for job {job_id}: {str(e)}") + logger.error(f"Transcription failed for job {job_id}: {str(e)}") transcription_results[job_id] = {"status": "failed", "error": str(e)} finally: # Clean up the temporary file os.remove(file_path) - L.DEBUG(f"Cleaned up temporary file for job {job_id}") + logger.debug(f"Cleaned up temporary file for job {job_id}") async def run_transcription(command, file_path): - L.DEBUG(f"Running transcription command: {' '.join(command)}") + logger.debug(f"Running transcription command: {' '.join(command)}") proc = await asyncio.create_subprocess_exec( *command, stdout=asyncio.subprocess.PIPE, @@ -181,9 +181,9 @@ async def run_transcription(command, file_path): stdout, stderr = await proc.communicate() if proc.returncode != 0: error_message = f"Error running command: {stderr.decode()}" - L.ERR(error_message) + logger.error(error_message) raise Exception(error_message) - L.DEBUG("Transcription command completed successfully") + logger.debug("Transcription command completed successfully") return stdout.decode().strip() async def convert_to_wav(file_path: str): diff --git a/sijapi/routers/cal.py b/sijapi/routers/cal.py index 7cca17a..f1a3399 100644 --- a/sijapi/routers/cal.py +++ b/sijapi/routers/cal.py @@ -22,33 +22,34 @@ from sijapi.routers import loc cal = APIRouter() oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/token") timeout = httpx.Timeout(12) +logger = L.get_module_logger("cal") if MS365_TOGGLE is True: - L.CRIT(f"Visit https://api.sij.ai/o365/login to obtain your Microsoft 365 authentication token.") + logger.critical(f"Visit https://api.sij.ai/o365/login to obtain your Microsoft 365 authentication token.") @cal.get("/o365/login") async def login(): - L.DEBUG(f"Received request to /o365/login") - L.DEBUG(f"SCOPE: {MS365_SCOPE}") + logger.debug(f"Received request to /o365/login") + logger.debug(f"SCOPE: {MS365_SCOPE}") if not MS365_SCOPE: - L.ERR("No scopes defined for authorization.") + logger.error("No scopes defined for authorization.") raise HTTPException( status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="No scopes defined for authorization." ) authorization_url = f"{MS365_AUTHORITY_URL}/oauth2/v2.0/authorize?client_id={MS365_CLIENT_ID}&response_type=code&redirect_uri={MS365_REDIRECT_PATH}&scope={'+'.join(MS365_SCOPE)}" - L.INFO(f"Redirecting to authorization URL: {authorization_url}") + logger.info(f"Redirecting to authorization URL: {authorization_url}") return RedirectResponse(authorization_url) @cal.get("/o365/oauth_redirect") async def oauth_redirect(code: str = None, error: str = None): - L.DEBUG(f"Received request to /o365/oauth_redirect") + logger.debug(f"Received request to /o365/oauth_redirect") if error: - L.ERR(f"OAuth2 Error: {error}") + logger.error(f"OAuth2 Error: {error}") raise HTTPException( status_code=status.HTTP_400_BAD_REQUEST, detail="OAuth2 Error" ) - L.INFO(f"Requesting token with authorization code: {code}") + logger.info(f"Requesting token with authorization code: {code}") token_url = f"{MS365_AUTHORITY_URL}/oauth2/v2.0/token" data = { "client_id": MS365_CLIENT_ID, @@ -59,15 +60,15 @@ if MS365_TOGGLE is True: } async with httpx.AsyncClient(timeout=timeout) as client: response = await client.post(token_url, data=data) - L.DEBUG(f"Token endpoint response status code: {response.status_code}") - L.INFO(f"Token endpoint response text: {response.text}") + logger.debug(f"Token endpoint response status code: {response.status_code}") + logger.info(f"Token endpoint response text: {response.text}") result = response.json() if 'access_token' in result: await save_token(result) - L.INFO("Access token obtained successfully") + logger.info("Access token obtained successfully") return {"message": "Access token stored successfully"} else: - L.CRIT(f"Failed to obtain access token. Response: {result}") + logger.critical(f"Failed to obtain access token. Response: {result}") raise HTTPException( status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="Failed to obtain access token" @@ -75,7 +76,7 @@ if MS365_TOGGLE is True: @cal.get("/o365/me") async def read_items(): - L.DEBUG(f"Received request to /o365/me") + logger.debug(f"Received request to /o365/me") token = await load_token() if not token: raise HTTPException( @@ -88,10 +89,10 @@ if MS365_TOGGLE is True: response = await client.get(graph_url, headers=headers) if response.status_code == 200: user = response.json() - L.INFO(f"User retrieved: {user}") + logger.info(f"User retrieved: {user}") return user else: - L.ERR("Invalid or expired token") + logger.error("Invalid or expired token") raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid or expired token", @@ -99,14 +100,14 @@ if MS365_TOGGLE is True: ) async def save_token(token): - L.DEBUG(f"Saving token: {token}") + logger.debug(f"Saving token: {token}") try: token["expires_at"] = int(time.time()) + token["expires_in"] with open(MS365_TOKEN_PATH, "w") as file: json.dump(token, file) - L.DEBUG(f"Saved token to {MS365_TOKEN_PATH}") + logger.debug(f"Saved token to {MS365_TOKEN_PATH}") except Exception as e: - L.ERR(f"Failed to save token: {e}") + logger.error(f"Failed to save token: {e}") async def load_token(): if os.path.exists(MS365_TOKEN_PATH): @@ -114,21 +115,21 @@ if MS365_TOGGLE is True: with open(MS365_TOKEN_PATH, "r") as file: token = json.load(file) except FileNotFoundError: - L.ERR("Token file not found.") + logger.error("Token file not found.") return None except json.JSONDecodeError: - L.ERR("Failed to decode token JSON") + logger.error("Failed to decode token JSON") return None if token: token["expires_at"] = int(time.time()) + token["expires_in"] - L.DEBUG(f"Loaded token: {token}") # Add this line to log the loaded token + logger.debug(f"Loaded token: {token}") # Add this line to log the loaded token return token else: - L.DEBUG("No token found.") + logger.debug("No token found.") return None else: - L.ERR(f"No file found at {MS365_TOKEN_PATH}") + logger.error(f"No file found at {MS365_TOKEN_PATH}") return None @@ -158,39 +159,39 @@ if MS365_TOGGLE is True: response = await client.post(token_url, data=data) result = response.json() if "access_token" in result: - L.INFO("Access token refreshed successfully") + logger.info("Access token refreshed successfully") return result else: - L.ERR("Failed to refresh access token") + logger.error("Failed to refresh access token") return None async def refresh_token(): token = await load_token() if not token: - L.ERR("No token found in storage") + logger.error("No token found in storage") raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail="No token found", ) if 'refresh_token' not in token: - L.ERR("Refresh token not found in the loaded token") + logger.error("Refresh token not found in the loaded token") raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail="Refresh token not found", ) refresh_token = token['refresh_token'] - L.DEBUG("Found refresh token, attempting to refresh access token") + logger.debug("Found refresh token, attempting to refresh access token") new_token = await get_new_token_with_refresh_token(refresh_token) if new_token: await save_token(new_token) - L.INFO("Token refreshed and saved successfully") + logger.info("Token refreshed and saved successfully") else: - L.ERR("Failed to refresh token") + logger.error("Failed to refresh token") raise HTTPException( status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="Failed to refresh token", @@ -204,7 +205,7 @@ def get_calendar_ids() -> Dict[str, str]: calendar_identifiers = { calendar.title() : calendar.calendarIdentifier() for calendar in all_calendars } - L.DEBUG(f"{calendar_identifiers}") + logger.debug(f"{calendar_identifiers}") return calendar_identifiers # Helper to convert datetime to NSDate @@ -245,7 +246,7 @@ def get_macos_calendar_events(start_date: datetime, end_date: datetime, calendar def completion_handler(granted, error): if error is not None: - L.ERR(f"Error: {error}") + logger.error(f"Error: {error}") access_granted.append(granted) # Notify the main thread that the completion handler has executed with access_granted_condition: @@ -260,11 +261,11 @@ def get_macos_calendar_events(start_date: datetime, end_date: datetime, calendar if access_granted: return access_granted[0] else: - L.ERR("Request access timed out or failed") + logger.error("Request access timed out or failed") return False if not request_access(): - L.ERR("Access to calendar data was not granted") + logger.error("Access to calendar data was not granted") return [] ns_start_date = datetime_to_nsdate(start_date) @@ -330,7 +331,7 @@ async def get_ms365_events(start_date: datetime, end_date: datetime): response = await client.get(graph_url, headers=headers) if response.status_code != 200: - L.ERR("Failed to retrieve events from Microsoft 365") + logger.error("Failed to retrieve events from Microsoft 365") raise HTTPException( status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="Failed to retrieve events", @@ -346,33 +347,33 @@ async def parse_calendar_for_day(range_start: datetime, range_end: datetime, eve event_list = [] for event in events: - L.INFO(f"Event: {event}") + logger.info(f"Event: {event}") start_str = event.get('start') end_str = event.get('end') if isinstance(start_str, dict): start_str = start_str.get('dateTime') else: - L.INFO(f"Start date string not a dict") + logger.info(f"Start date string not a dict") if isinstance(end_str, dict): end_str = end_str.get('dateTime') else: - L.INFO(f"End date string not a dict") + logger.info(f"End date string not a dict") try: start_date = await loc.dt(start_str) if start_str else None except (ValueError, TypeError) as e: - L.ERR(f"Invalid start date format: {start_str}, error: {e}") + logger.error(f"Invalid start date format: {start_str}, error: {e}") continue try: end_date = await loc.dt(end_str) if end_str else None except (ValueError, TypeError) as e: - L.ERR(f"Invalid end date format: {end_str}, error: {e}") + logger.error(f"Invalid end date format: {end_str}, error: {e}") continue - L.DEBUG(f"Comparing {start_date} with range {range_start} to {range_end}") + logger.debug(f"Comparing {start_date} with range {range_start} to {range_end}") if start_date: # Ensure start_date is timezone-aware @@ -404,11 +405,11 @@ async def parse_calendar_for_day(range_start: datetime, range_end: datetime, eve "busy": event.get('showAs', '') in ['busy', 'tentative'], "all_day": event.get('isAllDay', False) } - L.INFO(f"Event_data: {event_data}") + logger.info(f"Event_data: {event_data}") event_list.append(event_data) else: - L.DEBUG(f"Event outside of specified range: {start_date} to {end_date}") + logger.debug(f"Event outside of specified range: {start_date} to {end_date}") else: - L.ERR(f"Invalid or missing start date for event: {event.get('id', 'Unknown ID')}") + logger.error(f"Invalid or missing start date for event: {event.get('id', 'Unknown ID')}") return event_list \ No newline at end of file diff --git a/sijapi/routers/cf.py b/sijapi/routers/cf.py index 1067b44..97b1934 100644 --- a/sijapi/routers/cf.py +++ b/sijapi/routers/cf.py @@ -12,6 +12,7 @@ from asyncio import sleep import os cf = APIRouter() +logger = L.get_module_logger("cal") class DNSRecordRequest(BaseModel): full_domain: str @@ -69,7 +70,7 @@ async def retry_request(url, headers, max_retries=5, backoff_factor=1): response.raise_for_status() return response except (httpx.HTTPError, httpx.ConnectTimeout) as e: - L.ERR(f"Request failed: {e}. Retrying {retry + 1}/{max_retries}...") + logger.error(f"Request failed: {e}. Retrying {retry + 1}/{max_retries}...") await sleep(backoff_factor * (2 ** retry)) raise HTTPException(status_code=500, detail="Max retries exceeded for Cloudflare API request") diff --git a/sijapi/routers/dist.py b/sijapi/routers/dist.py index a21a3a8..0f1b7f4 100644 --- a/sijapi/routers/dist.py +++ b/sijapi/routers/dist.py @@ -5,10 +5,10 @@ from fastapi import APIRouter, HTTPException import asyncio import logging from sijapi.utilities import run_ssh_command -from sijapi import REBOOT_SCRIPT_PATH, HOST_CONFIG, API_CONFIG +from sijapi import L, REBOOT_SCRIPT_PATH, HOST_CONFIG, API_CONFIG dist = APIRouter() -logger = logging.getLogger(__name__) +logger = L.get_module_logger("dist") @dist.get("/update-restart-others") async def update_and_restart_others(): diff --git a/sijapi/routers/email.py b/sijapi/routers/email.py index d7c8b3b..6fd968f 100644 --- a/sijapi/routers/email.py +++ b/sijapi/routers/email.py @@ -24,9 +24,14 @@ from sijapi import L, PODCAST_DIR, DEFAULT_VOICE, EMAIL_CONFIG, EMAIL_LOGS from sijapi.routers import img, loc, tts, llm from sijapi.utilities import clean_text, assemble_journal_path, extract_text, prefix_lines from sijapi.classes import EmailAccount, IMAPConfig, SMTPConfig, IncomingEmail, EmailContact, AutoResponder -from sijapi.classes import EmailAccount -email = APIRouter(tags=["private"]) +email = APIRouter() + +logger = L.get_module_logger("email") +print(f"Email logger level: {logger.level}") # Debug print + +logger.debug("This is a debug message from email module") +logger.info("This is an info message from email module") def load_email_accounts(yaml_path: str) -> List[EmailAccount]: @@ -44,46 +49,46 @@ def get_imap_connection(account: EmailAccount): starttls=account.imap.encryption == 'STARTTLS') - -def get_smtp_connection(autoresponder): +def get_smtp_connection(autoresponder: AutoResponder): # Create an SSL context that doesn't verify certificates context = ssl.create_default_context() context.check_hostname = False context.verify_mode = ssl.CERT_NONE - if autoresponder.smtp.encryption == 'SSL': + smtp_config = autoresponder.smtp + + if smtp_config.encryption == 'SSL': try: - L.DEBUG(f"Attempting SSL connection to {autoresponder.smtp.host}:{autoresponder.smtp.port}") - return SMTP_SSL(autoresponder.smtp.host, autoresponder.smtp.port, context=context) + logger.debug(f"Attempting SSL connection to {smtp_config.host}:{smtp_config.port}") + return SMTP_SSL(smtp_config.host, smtp_config.port, context=context) except ssl.SSLError as e: - L.ERR(f"SSL connection failed: {str(e)}") + logger.error(f"SSL connection failed: {str(e)}") # If SSL fails, try TLS try: - L.DEBUG(f"Attempting STARTTLS connection to {autoresponder.smtp.host}:{autoresponder.smtp.port}") - smtp = SMTP(autoresponder.smtp.host, autoresponder.smtp.port) + logger.debug(f"Attempting STARTTLS connection to {smtp_config.host}:{smtp_config.port}") + smtp = SMTP(smtp_config.host, smtp_config.port) smtp.starttls(context=context) return smtp except Exception as e: - L.ERR(f"STARTTLS connection failed: {str(e)}") + logger.error(f"STARTTLS connection failed: {str(e)}") raise - elif autoresponder.smtp.encryption == 'STARTTLS': + elif smtp_config.encryption == 'STARTTLS': try: - L.DEBUG(f"Attempting STARTTLS connection to {autoresponder.smtp.host}:{autoresponder.smtp.port}") - smtp = SMTP(autoresponder.smtp.host, autoresponder.smtp.port) + logger.debug(f"Attempting STARTTLS connection to {smtp_config.host}:{smtp_config.port}") + smtp = SMTP(smtp_config.host, smtp_config.port) smtp.starttls(context=context) return smtp except Exception as e: - L.ERR(f"STARTTLS connection failed: {str(e)}") + logger.error(f"STARTTLS connection failed: {str(e)}") raise else: try: - L.DEBUG(f"Attempting unencrypted connection to {autoresponder.smtp.host}:{autoresponder.smtp.port}") - return SMTP(autoresponder.smtp.host, autoresponder.smtp.port) + logger.debug(f"Attempting unencrypted connection to {smtp_config.host}:{smtp_config.port}") + return SMTP(smtp_config.host, smtp_config.port) except Exception as e: - L.ERR(f"Unencrypted connection failed: {str(e)}") + logger.error(f"Unencrypted connection failed: {str(e)}") raise - async def send_response(to_email: str, subject: str, body: str, profile: AutoResponder, image_attachment: Path = None) -> bool: server = None try: @@ -98,20 +103,20 @@ async def send_response(to_email: str, subject: str, body: str, profile: AutoRes img = MIMEImage(img_file.read(), name=os.path.basename(image_attachment)) message.attach(img) - L.DEBUG(f"Sending auto-response to {to_email} concerning {subject} from account {profile.name}...") + logger.debug(f"Sending auto-response to {to_email} concerning {subject} from account {profile.name}...") - server = get_smtp_connection(profile.smtp) - L.DEBUG(f"SMTP connection established: {type(server)}") + server = get_smtp_connection(profile) + logger.debug(f"SMTP connection established: {type(server)}") server.login(profile.smtp.username, profile.smtp.password) server.send_message(message) - L.INFO(f"Auto-response sent to {to_email} concerning {subject} from account {profile.name}!") + logger.info(f"Auto-response sent to {to_email} concerning {subject} from account {profile.name}!") return True except Exception as e: - L.ERR(f"Error in preparing/sending auto-response from account {profile.name}: {str(e)}") - L.ERR(f"SMTP details - Host: {profile.smtp.host}, Port: {profile.smtp.port}, Encryption: {profile.smtp.encryption}") - L.ERR(traceback.format_exc()) + logger.error(f"Error in preparing/sending auto-response from account {profile.name}: {str(e)}") + logger.error(f"SMTP details - Host: {profile.smtp.host}, Port: {profile.smtp.port}, Encryption: {profile.smtp.encryption}") + logger.error(traceback.format_exc()) return False finally: @@ -119,7 +124,8 @@ async def send_response(to_email: str, subject: str, body: str, profile: AutoRes try: server.quit() except Exception as e: - L.ERR(f"Error closing SMTP connection: {str(e)}") + logger.error(f"Error closing SMTP connection: {str(e)}") + @@ -155,10 +161,10 @@ async def process_account_archival(account: EmailAccount): while True: try: processed_uids = await load_processed_uids(summarized_log) - L.DEBUG(f"{len(processed_uids)} emails marked as already summarized are being ignored.") + logger.debug(f"{len(processed_uids)} emails marked as already summarized are being ignored.") with get_imap_connection(account) as inbox: unread_messages = inbox.messages(unread=True) - L.DEBUG(f"There are {len(unread_messages)} unread messages.") + logger.debug(f"There are {len(unread_messages)} unread messages.") for uid, message in unread_messages: uid_str = uid.decode() if isinstance(uid, bytes) else str(uid) if uid_str not in processed_uids: @@ -178,13 +184,13 @@ async def process_account_archival(account: EmailAccount): save_success = await save_email(md_path, md_content) if save_success: await save_processed_uid(summarized_log, account.name, uid_str) - L.INFO(f"Summarized email: {uid_str}") + logger.info(f"Summarized email: {uid_str}") else: - L.WARN(f"Failed to summarize {this_email.subject}") + logger.warn(f"Failed to summarize {this_email.subject}") else: - L.DEBUG(f"Skipping {uid_str} because it was already processed.") + logger.debug(f"Skipping {uid_str} because it was already processed.") except Exception as e: - L.ERR(f"An error occurred during summarization for account {account.name}: {e}") + logger.error(f"An error occurred during summarization for account {account.name}: {e}") await asyncio.sleep(account.refresh) @@ -230,7 +236,7 @@ tags: return markdown_content except Exception as e: - L.ERR(f"Exception: {e}") + logger.error(f"Exception: {e}") return False async def save_email(md_path, md_content): @@ -238,14 +244,14 @@ async def save_email(md_path, md_content): with open(md_path, 'w', encoding='utf-8') as md_file: md_file.write(md_content) - L.DEBUG(f"Saved markdown to {md_path}") + logger.debug(f"Saved markdown to {md_path}") return True except Exception as e: - L.ERR(f"Failed to save email: {e}") + logger.error(f"Failed to save email: {e}") return False def get_matching_autoresponders(this_email: IncomingEmail, account: EmailAccount) -> List[AutoResponder]: - L.DEBUG(f"Called get_matching_autoresponders for email \"{this_email.subject},\" account name \"{account.name}\"") + logger.debug(f"Called get_matching_autoresponders for email \"{this_email.subject},\" account name \"{account.name}\"") def matches_list(item: str, this_email: IncomingEmail) -> bool: if '@' in item: return item in this_email.sender @@ -256,12 +262,12 @@ def get_matching_autoresponders(this_email: IncomingEmail, account: EmailAccount whitelist_match = not profile.whitelist or any(matches_list(item, this_email) for item in profile.whitelist) blacklist_match = any(matches_list(item, this_email) for item in profile.blacklist) if whitelist_match and not blacklist_match: - L.DEBUG(f"We have a match for {whitelist_match} and no blacklist matches.") + logger.debug(f"We have a match for {whitelist_match} and no blacklist matches.") matching_profiles.append(profile) elif whitelist_match and blacklist_match: - L.DEBUG(f"Matched whitelist for {whitelist_match}, but also matched blacklist for {blacklist_match}") + logger.debug(f"Matched whitelist for {whitelist_match}, but also matched blacklist for {blacklist_match}") else: - L.DEBUG(f"No whitelist or blacklist matches.") + logger.debug(f"No whitelist or blacklist matches.") return matching_profiles @@ -272,30 +278,30 @@ async def process_account_autoresponding(account: EmailAccount): while True: try: processed_uids = await load_processed_uids(EMAIL_AUTORESPONSE_LOG) - L.DEBUG(f"{len(processed_uids)} emails marked as already responded to are being ignored.") + logger.debug(f"{len(processed_uids)} emails marked as already responded to are being ignored.") with get_imap_connection(account) as inbox: unread_messages = inbox.messages(unread=True) - L.DEBUG(f"There are {len(unread_messages)} unread messages.") + logger.debug(f"There are {len(unread_messages)} unread messages.") for uid, message in unread_messages: uid_str = uid.decode() if isinstance(uid, bytes) else str(uid) if uid_str not in processed_uids: await autorespond_single_email(message, uid_str, account, EMAIL_AUTORESPONSE_LOG) else: - L.DEBUG(f"Skipping {uid_str} because it was already processed.") + logger.debug(f"Skipping {uid_str} because it was already processed.") except Exception as e: - L.ERR(f"An error occurred during auto-responding for account {account.name}: {e}") + logger.error(f"An error occurred during auto-responding for account {account.name}: {e}") await asyncio.sleep(account.refresh) async def autorespond_single_email(message, uid_str: str, account: EmailAccount, log_file: Path): this_email = await create_incoming_email(message) - L.DEBUG(f"Evaluating {this_email.subject} for autoresponse-worthiness...") + logger.debug(f"Evaluating {this_email.subject} for autoresponse-worthiness...") matching_profiles = get_matching_autoresponders(this_email, account) - L.DEBUG(f"Matching profiles: {matching_profiles}") + logger.debug(f"Matching profiles: {matching_profiles}") for profile in matching_profiles: response_body = await generate_response(this_email, profile, account) @@ -305,15 +311,15 @@ async def autorespond_single_email(message, uid_str: str, account: EmailAccount, jpg_path = await img.workflow(profile.image_prompt, earlyout=False, downscale_to_fit=True) if profile.image_prompt else None success = await send_response(this_email.sender, subject, response_body, profile, jpg_path) if success: - L.WARN(f"Auto-responded to email: {this_email.subject}") + logger.warn(f"Auto-responded to email: {this_email.subject}") await save_processed_uid(log_file, account.name, uid_str) else: - L.WARN(f"Failed to send auto-response to {this_email.subject}") + logger.warn(f"Failed to send auto-response to {this_email.subject}") else: - L.WARN(f"Unable to generate auto-response for {this_email.subject}") + logger.warn(f"Unable to generate auto-response for {this_email.subject}") async def generate_response(this_email: IncomingEmail, profile: AutoResponder, account: EmailAccount) -> Optional[str]: - L.INFO(f"Generating auto-response to {this_email.subject} with profile: {profile.name}") + logger.info(f"Generating auto-response to {this_email.subject} with profile: {profile.name}") now = await loc.dt(dt_datetime.now()) then = await loc.dt(this_email.datetime_received) @@ -331,7 +337,7 @@ Respond on behalf of {account.fullname}, who is unable to respond personally bec try: response = await llm.query_ollama(usr_prompt, sys_prompt, profile.ollama_model, 400) - L.DEBUG(f"query_ollama response: {response}") + logger.debug(f"query_ollama response: {response}") if isinstance(response, dict) and "message" in response and "content" in response["message"]: response = response["message"]["content"] @@ -339,7 +345,7 @@ Respond on behalf of {account.fullname}, who is unable to respond personally bec return response + "\n\n" except Exception as e: - L.ERR(f"Error generating auto-response: {str(e)}") + logger.error(f"Error generating auto-response: {str(e)}") return None diff --git a/sijapi/routers/ghost.py b/sijapi/routers/ghost.py new file mode 100644 index 0000000..f328302 --- /dev/null +++ b/sijapi/routers/ghost.py @@ -0,0 +1,50 @@ +def generate_jwt_token(): + key_id, key_secret = GHOST_API_KEY.split(':') + iat = int(date.now().timestamp()) + exp = iat + 5 * 60 # Token expiration time set to 5 minutes from now for consistency with the working script + payload = { + 'iat': iat, + 'exp': exp, + 'aud': '/admin/' # Adjusted to match the working script + } + token = jwt.encode(payload, bytes.fromhex(key_secret), algorithm='HS256', headers={'kid': key_id}) + return token.decode('utf-8') if isinstance(token, bytes) else token # Ensure the token is decoded to UTF-8 string + + +def post_to_ghost(title, image_path, html_content, ghost_tags): + jwt_token = generate_jwt_token() + ghost_headers = {'Authorization': f'Ghost {jwt_token}'} + + # Upload the image to Ghost + with open(image_path, 'rb') as f: + files = {'file': (os.path.basename(image_path), f, 'image/jpg')} + image_response = requests.post(f"{GHOST_API_URL}/images/upload/", headers=ghost_headers, files=files) + image_response.raise_for_status() # Ensure the request was successful + image_url = image_response.json()['images'][0]['url'] + + # Prepare the post content + updated_html_content = f'Image
{html_content}' + mobiledoc = { + "version": "0.3.1", + "atoms": [], + "cards": [["html", {"cardName": "html", "html": updated_html_content}]], + "markups": [], + "sections": [[10, 0]] + } + mobiledoc = json.dumps(mobiledoc) + + post_data = { + 'posts': [{ + 'title': title, + 'mobiledoc': mobiledoc, + 'status': 'published', + 'tags': ghost_tags + }] + } + + # Create a new post + post_response = requests.post(f"{GHOST_API_URL}/posts/", json=post_data, headers=ghost_headers) + post_response.raise_for_status() + post_url = post_response.json()['posts'][0]['url'] + + return post_url \ No newline at end of file diff --git a/sijapi/routers/health.py b/sijapi/routers/health.py index bd2cfbd..ebca96f 100644 --- a/sijapi/routers/health.py +++ b/sijapi/routers/health.py @@ -11,6 +11,7 @@ from tailscale import Tailscale from sijapi import L, API, TS_ID, SUBNET_BROADCAST health = APIRouter(tags=["public", "trusted", "private"]) +logger = L.get_module_logger("health") @health.get("/health") def get_health(): @@ -48,7 +49,7 @@ async def get_wan_ip(): wan_info = response.json() return wan_info.get('ip', 'Unavailable') except Exception as e: - L.ERR(f"Error fetching WAN IP: {e}") + logger.error(f"Error fetching WAN IP: {e}") return "Unavailable" @health.get("/ts_ip") diff --git a/sijapi/routers/ig.py b/sijapi/routers/ig.py index 6ab6775..bd67336 100644 --- a/sijapi/routers/ig.py +++ b/sijapi/routers/ig.py @@ -41,6 +41,7 @@ from io import BytesIO import base64 ig = APIRouter() +logger = L.get_module_logger("ig") class IG_Request(BaseModel): file: Optional[UploadFile] = None # upload a particular file to Instagram @@ -186,7 +187,7 @@ def get_user_media(username, amount=30): Fetch recent media for a given username, return List of medias """ - L.DEBUG(f"Fetching recent media for {username}...") + logger.debug(f"Fetching recent media for {username}...") user_id = cl.user_id_from_username(username) medias = cl.user_medias(user_id, amount) final_medias = [] @@ -201,7 +202,7 @@ def get_user_image_urls(username, amount=30) -> List[str]: """ Fetch recent media URLs for a given username, return List of media URLs """ - L.DEBUG(f"Fetching recent media URLs for {username}...") + logger.debug(f"Fetching recent media URLs for {username}...") user_id = cl.user_id_from_username(username) medias = cl.user_medias(user_id, amount) @@ -229,7 +230,7 @@ def get_random_follower(): def get_medias_by_hashtag(hashtag: str, days_ago_max:int = 14, ht_type:str = None, amount:int = args.count): if not ht_type: ht_type = args.commentmode - L.DEBUG(f"Fetching {ht_type} media for hashtag: {hashtag}") + logger.debug(f"Fetching {ht_type} media for hashtag: {hashtag}") ht_medias = [] while True: sleep(SHORT) @@ -239,10 +240,10 @@ def get_medias_by_hashtag(hashtag: str, days_ago_max:int = 14, ht_type:str = Non ht_medias.extend(cl.hashtag_medias_recent(name=hashtag, amount=amount*10)) filtered_medias = filter_medias(ht_medias, days_ago_max=days_ago_max) - L.DEBUG(f"Filtered {ht_type} media count obtained for '#{hashtag}': {len(filtered_medias)}") + logger.debug(f"Filtered {ht_type} media count obtained for '#{hashtag}': {len(filtered_medias)}") if len(filtered_medias) >= amount: - L.DEBUG(f"Desired amount of {amount} filtered media reached.") + logger.debug(f"Desired amount of {amount} filtered media reached.") break return filtered_medias @@ -250,11 +251,11 @@ def get_medias_by_hashtag(hashtag: str, days_ago_max:int = 14, ht_type:str = Non def get_medias_from_all_hashtags(days_ago_max=14, ht_type:str = None, amount:int = args.count): if not ht_type: ht_type = args.commentmode - L.DEBUG(f"Fetching {ht_type} media.") + logger.debug(f"Fetching {ht_type} media.") filtered_medias = [] while len(filtered_medias) < amount: hashtag = random.choice(HASHTAGS) - L.DEBUG(f"Using hashtag: {hashtag}") + logger.debug(f"Using hashtag: {hashtag}") fetched_medias = [] sleep(SHORT) if ht_type == "top": @@ -264,15 +265,15 @@ def get_medias_from_all_hashtags(days_ago_max=14, ht_type:str = None, amount:int current_filtered_medias = filter_medias(fetched_medias, days_ago_max=days_ago_max) filtered_medias.extend(current_filtered_medias) - L.DEBUG(f"Filtered {ht_type} media count obtained for '#{hashtag}': {len(current_filtered_medias)}") + logger.debug(f"Filtered {ht_type} media count obtained for '#{hashtag}': {len(current_filtered_medias)}") # Trim the list if we've collected more than needed if len(filtered_medias) > amount: filtered_medias = filtered_medias[:amount] - L.DEBUG(f"Desired amount of {amount} filtered media reached.") + logger.debug(f"Desired amount of {amount} filtered media reached.") break else: - L.DEBUG(f"Total filtered media count so far: {len(filtered_medias)}") + logger.debug(f"Total filtered media count so far: {len(filtered_medias)}") return filtered_medias @@ -345,7 +346,7 @@ def download_and_resize_image(url: str, download_path: str = None, max_dimension return download_path except Exception as e: # Handle or log the error as needed - L.DEBUG(f"Error downloading or resizing image: {e}") + logger.debug(f"Error downloading or resizing image: {e}") return None @@ -365,17 +366,17 @@ def comment_on_user_media(user: str, comment_type: str = "default", amount=5): comment_text = llava(encoded_media, COMMENT_PROMPT_SYS, comment_prompt_usr) if args.llava or not args.openai else gpt4v(encoded_media, COMMENT_PROMPT_SYS, comment_prompt_usr) if comment_text: cl.media_comment(media.pk, comment_text) - L.DEBUG(f"Commented on media: {media.pk}") + logger.debug(f"Commented on media: {media.pk}") else: - L.DEBUG(f"Failed to generate comment for media: {media.pk}") + logger.debug(f"Failed to generate comment for media: {media.pk}") add_media_to_completed_lists(media) sleep(SHORT) else: - L.DEBUG(f"We received a nonetype! {media_path}") + logger.debug(f"We received a nonetype! {media_path}") else: - L.DEBUG(f"URL for {media.pk} disappeared it seems...") + logger.debug(f"URL for {media.pk} disappeared it seems...") else: - L.DEBUG(f"Media already interacted with: {media.pk}") + logger.debug(f"Media already interacted with: {media.pk}") def comment_on_hashtagged_media(comment_type: str = args.commenttype, amount=3, hashtag: str = None): """ @@ -400,22 +401,22 @@ def comment_on_hashtagged_media(comment_type: str = args.commenttype, amount=3, try: like_result = cl.media_like(media) if like_result: - L.DEBUG(f"Liked media: https://instagram.com/p/{media.pk}/") + logger.debug(f"Liked media: https://instagram.com/p/{media.pk}/") except instagrapi.exceptions.FeedbackRequired as e: - L.DEBUG(f"Cannot like media {media.pk}: {str(e)}") + logger.debug(f"Cannot like media {media.pk}: {str(e)}") if comment_text: try: cl.media_comment(media.pk, comment_text) - L.DEBUG(f"Commented on media: https://instagram.com/p/{media.pk}/") + logger.debug(f"Commented on media: https://instagram.com/p/{media.pk}/") except instagrapi.exceptions.FeedbackRequired as e: - L.DEBUG(f"Cannot comment on media {media.pk}: {str(e)}") + logger.debug(f"Cannot comment on media {media.pk}: {str(e)}") else: - L.DEBUG(f"Failed to generate comment for media: https://instagram.com/p/{media.pk}") + logger.debug(f"Failed to generate comment for media: https://instagram.com/p/{media.pk}") add_media_to_completed_lists(media) sleep(SHORT) else: - L.DEBUG(f"Media already interacted with: {media.pk}") + logger.debug(f"Media already interacted with: {media.pk}") def comment_on_specific_media(media_url, comment_type: str = "default"): @@ -436,11 +437,11 @@ def comment_on_specific_media(media_url, comment_type: str = "default"): if comment_text: try: cl.media_comment(media.pk, comment_text) - L.DEBUG(f"Commented on specific media: https://instagram.com/p/{media.pk}/") + logger.debug(f"Commented on specific media: https://instagram.com/p/{media.pk}/") except instagrapi.exceptions.FeedbackRequired as e: - L.DEBUG(f"Failed to comment on specific media: https://instagram.com/p/{media.pk}/ due to error: {str(e)}") + logger.debug(f"Failed to comment on specific media: https://instagram.com/p/{media.pk}/ due to error: {str(e)}") else: - L.DEBUG(f"Failed to generate comment for specific media: https://instagram.com/p/{media.pk}/") + logger.debug(f"Failed to generate comment for specific media: https://instagram.com/p/{media.pk}/") @@ -485,16 +486,16 @@ def update_prompt(workflow: dict, post: dict, positive: str, found_key=[None], p if value == "API_PrePrompt": workflow[key] = post.get(value, "") + positive - L.DEBUG(f"Updated API_PrePrompt to: {workflow[key]}") + logger.debug(f"Updated API_PrePrompt to: {workflow[key]}") elif value == "API_StylePrompt": workflow[key] = post.get(value, "") - L.DEBUG(f"Updated API_StylePrompt to: {workflow[key]}") + logger.debug(f"Updated API_StylePrompt to: {workflow[key]}") elif value == "API_NegativePrompt": workflow[key] = post.get(value, "") - L.DEBUG(f"Updated API_NegativePrompt to: {workflow[key]}") + logger.debug(f"Updated API_NegativePrompt to: {workflow[key]}") elif key == "seed" or key == "noise_seed": workflow[key] = random.randint(1000000000000, 9999999999999) - L.DEBUG(f"Updated seed to: {workflow[key]}") + logger.debug(f"Updated seed to: {workflow[key]}") elif (key == "width" or key == "max_width" or key == "scaled_width" or key == "side_length") and (value == 1023 or value == 1025): # workflow[key] = post.get(value, "") workflow[key] = post.get("width", 1024) @@ -502,7 +503,7 @@ def update_prompt(workflow: dict, post: dict, positive: str, found_key=[None], p # workflow[key] = post.get(value, "") workflow[key] = post.get("height", 1024) except Exception as e: - L.DEBUG(f"Error in update_prompt at path {' -> '.join(path)}: {e}") + logger.debug(f"Error in update_prompt at path {' -> '.join(path)}: {e}") raise return found_key[0] @@ -527,22 +528,22 @@ def update_prompt_custom(workflow: dict, API_PrePrompt: str, API_StylePrompt: st if value == "API_PrePrompt": workflow[key] = API_PrePrompt - L.DEBUG(f"Updated API_PrePrompt to: {workflow[key]}") + logger.debug(f"Updated API_PrePrompt to: {workflow[key]}") elif value == "API_StylePrompt": workflow[key] = API_StylePrompt - L.DEBUG(f"Updated API_StylePrompt to: {workflow[key]}") + logger.debug(f"Updated API_StylePrompt to: {workflow[key]}") elif value == "API_NegativePrompt": workflow[key] = API_NegativePrompt - L.DEBUG(f"Updated API_NegativePrompt to: {workflow[key]}") + logger.debug(f"Updated API_NegativePrompt to: {workflow[key]}") elif key == "seed" or key == "noise_seed": workflow[key] = random.randint(1000000000000, 9999999999999) - L.DEBUG(f"Updated seed to: {workflow[key]}") + logger.debug(f"Updated seed to: {workflow[key]}") elif (key == "width" or key == "max_width" or key == "scaled_width") and (value == 1023 or value == 1025): workflow[key] = 1024 elif (key == "dimension" or key == "height" or key == "max_height" or key == "scaled_height") and (value == 1023 or value == 1025): workflow[key] = 1024 except Exception as e: - L.DEBUG(f"Error in update_prompt_custom at path {' -> '.join(path)}: {e}") + logger.debug(f"Error in update_prompt_custom at path {' -> '.join(path)}: {e}") raise return found_key[0] @@ -582,14 +583,14 @@ def poll_status(prompt_id): elapsed_time = int(time.time() - start_time) # Calculate elapsed time in seconds status_response = requests.get(f"{COMFYUI_URL}/history/{prompt_id}") # Use \r to return to the start of the line, and end='' to prevent newline - L.DEBUG(f"\rGenerating {prompt_id}. Elapsed time: {elapsed_time} seconds", end='') + logger.debug(f"\rGenerating {prompt_id}. Elapsed time: {elapsed_time} seconds", end='') if status_response.status_code != 200: raise Exception("Failed to get job status") status_data = status_response.json() job_data = status_data.get(prompt_id, {}) if job_data.get("status", {}).get("completed", False): - L.DEBUG() - L.DEBUG(f"{prompt_id} completed in {elapsed_time} seconds.") + logger.debug() + logger.debug(f"{prompt_id} completed in {elapsed_time} seconds.") return job_data time.sleep(1) @@ -600,14 +601,14 @@ def poll_status(prompt_id): elapsed_time = int(time.time() - start_time) # Calculate elapsed time in seconds status_response = requests.get(f"{COMFYUI_URL}/history/{prompt_id}") # Use \r to return to the start of the line, and end='' to prevent newline - L.DEBUG(f"\rGenerating {prompt_id}. Elapsed time: {elapsed_time} seconds", end='') + logger.debug(f"\rGenerating {prompt_id}. Elapsed time: {elapsed_time} seconds", end='') if status_response.status_code != 200: raise Exception("Failed to get job status") status_data = status_response.json() job_data = status_data.get(prompt_id, {}) if job_data.get("status", {}).get("completed", False): - L.DEBUG() - L.DEBUG(f"{prompt_id} completed in {elapsed_time} seconds.") + logger.debug() + logger.debug(f"{prompt_id} completed in {elapsed_time} seconds.") return job_data time.sleep(1) @@ -618,12 +619,12 @@ def poll_status(prompt_id): def load_post(chosen_post: str = "default"): if chosen_post in PROFILE_CONFIG['posts']: post = PROFILE_CONFIG['posts'][chosen_post] - L.DEBUG(f"Loaded post for {chosen_post}") + logger.debug(f"Loaded post for {chosen_post}") else: - L.DEBUG(f"Unable to load post for {chosen_post}. Choosing a default post.") - chosen_post = choose_post(PROFILE_CONFIG['posts']) + logger.debug(f"Unable to load post for {chosen_post}. Choosing a default post.") + chosen_post = choose_random_post(PROFILE_CONFIG['posts']) post = PROFILE_CONFIG['posts'][chosen_post] - L.DEBUG(f"Defaulted to {chosen_post}") + logger.debug(f"Defaulted to {chosen_post}") return post @@ -633,18 +634,18 @@ def handle_image_workflow(chosen_post=None): or posting to Instagram based on the local flag. """ if chosen_post is None: - chosen_post = choose_post(PROFILE_CONFIG['posts']) + chosen_post = choose_random_post(PROFILE_CONFIG['posts']) post = load_post(chosen_post) workflow_name = args.workflow if args.workflow else random.choice(post['workflows']) - L.DEBUG(f"Workflow name: {workflow_name}") + logger.debug(f"Workflow name: {workflow_name}") - L.DEBUG(f"Generating image concept for {chosen_post} and {workflow_name} now.") + logger.debug(f"Generating image concept for {chosen_post} and {workflow_name} now.") image_concept = query_ollama(llmPrompt = post['llmPrompt'], max_tokens = 180) if args.local or not args.openai else query_gpt4(llmPrompt = post['llmPrompt'], max_tokens = 180) - L.DEBUG(f"Image concept for {chosen_post}: {image_concept}") + logger.debug(f"Image concept for {chosen_post}: {image_concept}") workflow_data = None @@ -658,9 +659,9 @@ def handle_image_workflow(chosen_post=None): jpg_file_path = image_gen(image_concept, "dall-e-3") else: saved_file_key = update_prompt(workflow=workflow_data, post=post, positive=image_concept) - L.DEBUG(f"Saved file key: {saved_file_key}") + logger.debug(f"Saved file key: {saved_file_key}") prompt_id = queue_prompt(workflow_data) - L.DEBUG(f"Prompt ID: {prompt_id}") + logger.debug(f"Prompt ID: {prompt_id}") status_data = poll_status(prompt_id) image_data = get_image(status_data, saved_file_key) if chosen_post == "landscape": @@ -699,17 +700,17 @@ def handle_custom_image(custom_post: str): system_msg = "You are a helpful AI who assists in generating prompts that will be used to generate highly realistic images. Always use the most visually descriptive terms possible, and avoid any vague or abstract concepts. Do not include any words or descriptions based on other senses or emotions. Strive to show rather than tell. Space is limited, so be efficient with your words." image_concept = query_ollama(system_msg=system_msg, user_msg=custom_post, max_tokens = 180) if args.local or not args.openai else query_gpt4(system_msg=system_msg, user_msg=custom_post, max_tokens = 180) - L.DEBUG(f"Image concept: {image_concept}") + logger.debug(f"Image concept: {image_concept}") if args.dalle and not args.local: jpg_file_path = image_gen(image_concept, "dall-e-3") else: saved_file_key = update_prompt(workflow=workflow_data, post=post, positive=image_concept) - L.DEBUG(f"Saved file key: {saved_file_key}") + logger.debug(f"Saved file key: {saved_file_key}") prompt_id = queue_prompt(workflow_data) - L.DEBUG(f"Prompt ID: {prompt_id}") + logger.debug(f"Prompt ID: {prompt_id}") status_data = poll_status(prompt_id) image_data = get_image(status_data, saved_file_key) @@ -728,7 +729,7 @@ def image_aftergen(jpg_file_path: str, chosen_post: str = None, post: Dict = Non if chosen_post and not prompt: prompt = PROFILE_CONFIG['posts'][chosen_post]['Vision_Prompt'] encoded_string = encode_image_to_base64(jpg_file_path) - L.DEBUG(f"Image successfully encoded from {jpg_file_path}") + logger.debug(f"Image successfully encoded from {jpg_file_path}") instagram_description = llava(encoded_string, prompt) if args.local or args.llava or not args.openai else gpt4v(encoded_string, prompt, 150) instagram_description = re.sub(r'^["\'](.*)["\']$', r'\1', instagram_description) @@ -759,27 +760,27 @@ Tags: {', '.join(ghost_tags)} with open(markdown_filename, "w") as md_file: md_file.write(markdown_content) - L.DEBUG(f"Markdown file created at {markdown_filename}") + logger.debug(f"Markdown file created at {markdown_filename}") if args.wallpaper: change_wallpaper(jpg_file_path) - L.DEBUG(f"Wallpaper changed.") + logger.debug(f"Wallpaper changed.") if not args.local: ig_footer = "" if not args.noig: post_url = upload_photo(jpg_file_path, instagram_description) - L.DEBUG(f"Image posted at {post_url}") + logger.debug(f"Image posted at {post_url}") ig_footer = f"\nInstagram link" if not args.noghost: ghost_text = f"{instagram_description}" ghost_url = post_to_ghost(img_title, jpg_file_path, ghost_text, ghost_tags) - L.DEBUG(f"Ghost post: {ghost_url}\n{ig_footer}") + logger.debug(f"Ghost post: {ghost_url}\n{ig_footer}") -def choose_post(posts): +def choose_random_post(posts): total_frequency = sum(posts[post_type]['frequency'] for post_type in posts) random_choice = random.randint(1, total_frequency) current_sum = 0 @@ -800,8 +801,6 @@ def load_json(json_payload, workflow): raise ValueError("No valid input provided.") - - def save_as_jpg(image_data, prompt_id, chosen_post:str = None, max_size=2160, quality=80): chosen_post = chosen_post if chosen_post else "custom" filename_png = f"{prompt_id}.png" @@ -836,121 +835,19 @@ def save_as_jpg(image_data, prompt_id, chosen_post:str = None, max_size=2160, qu return new_file_path except Exception as e: - L.DEBUG(f"Error processing image: {e}") + logger.debug(f"Error processing image: {e}") return None def upload_photo(path, caption, title: str=None): - L.DEBUG(f"Uploading photo from {path}...") + logger.debug(f"Uploading photo from {path}...") media = cl.photo_upload(path, caption) post_url = f"https://www.instagram.com/p/{media.code}/" return post_url -def format_duration(seconds): - """Return a string representing the duration in a human-readable format.""" - if seconds < 120: - return f"{int(seconds)} sec" - elif seconds < 6400: - return f"{int(seconds // 60)} min" - else: - return f"{seconds / 3600:.2f} hr" - -######################## -### HELPER FUNCTIONS ### -######################## - -import subprocess - -def change_wallpaper(image_path): - command = """ - osascript -e 'tell application "Finder" to set desktop picture to POSIX file "{}"' - """.format(image_path) - subprocess.run(command, shell=True) - - -def sleep(seconds): - """Sleep for a random amount of time, approximately the given number of seconds.""" - sleepupto(seconds*0.66, seconds*1.5) - -def sleepupto(min_seconds, max_seconds=None): - interval = random.uniform(min_seconds if max_seconds is not None else 0, max_seconds if max_seconds is not None else min_seconds) - start_time = time.time() - end_time = start_time + interval - - with tqdm(total=interval, desc=f"Sleeping for {format_duration(interval)}", unit=" sec", ncols=75, bar_format='{desc}: {bar} {remaining}') as pbar: - while True: - current_time = time.time() - elapsed_time = current_time - start_time - remaining_time = end_time - current_time - if elapsed_time >= interval: - break - duration = min(1, interval - elapsed_time) # Adjust sleep time to not exceed interval - time.sleep(duration) - pbar.update(duration) - # Update remaining time display - pbar.set_postfix_str(f"{format_duration(remaining_time)} remaining") - - -######################## -### GHOST FUNCTIONS ### -######################## - - - -def generate_jwt_token(): - key_id, key_secret = GHOST_API_KEY.split(':') - iat = int(date.now().timestamp()) - exp = iat + 5 * 60 # Token expiration time set to 5 minutes from now for consistency with the working script - payload = { - 'iat': iat, - 'exp': exp, - 'aud': '/admin/' # Adjusted to match the working script - } - token = jwt.encode(payload, bytes.fromhex(key_secret), algorithm='HS256', headers={'kid': key_id}) - return token.decode('utf-8') if isinstance(token, bytes) else token # Ensure the token is decoded to UTF-8 string - - -def post_to_ghost(title, image_path, html_content, ghost_tags): - jwt_token = generate_jwt_token() - ghost_headers = {'Authorization': f'Ghost {jwt_token}'} - - # Upload the image to Ghost - with open(image_path, 'rb') as f: - files = {'file': (os.path.basename(image_path), f, 'image/jpg')} - image_response = requests.post(f"{GHOST_API_URL}/images/upload/", headers=ghost_headers, files=files) - image_response.raise_for_status() # Ensure the request was successful - image_url = image_response.json()['images'][0]['url'] - - # Prepare the post content - updated_html_content = f'Image
{html_content}' - mobiledoc = { - "version": "0.3.1", - "atoms": [], - "cards": [["html", {"cardName": "html", "html": updated_html_content}]], - "markups": [], - "sections": [[10, 0]] - } - mobiledoc = json.dumps(mobiledoc) - - post_data = { - 'posts': [{ - 'title': title, - 'mobiledoc': mobiledoc, - 'status': 'published', - 'tags': ghost_tags - }] - } - - # Create a new post - post_response = requests.post(f"{GHOST_API_URL}/posts/", json=post_data, headers=ghost_headers) - post_response.raise_for_status() - post_url = post_response.json()['posts'][0]['url'] - - return post_url - - ######################################################## + @ig.post("/ig/flow") async def ig_flow_endpoint(new_session: bool = False): current_unix_time = int(date.now().timestamp()) @@ -958,16 +855,16 @@ async def ig_flow_endpoint(new_session: bool = False): time_remaining = 30 - (time_since_rollover % 30) if time_remaining < 4: - L.DEBUG("Too close to end of TOTP counter. Waiting.") + logger.debug("Too close to end of TOTP counter. Waiting.") sleepupto(5, 5) if not new_session and os.path.exists(IG_SESSION_PATH): cl.load_settings(IG_SESSION_PATH) - L.DEBUG("Loaded past session.") + logger.debug("Loaded past session.") elif new_session and cl.login(IG_USERNAME, IG_PASSWORD, verification_code=TOTP.now()): cl.dump_settings(IG_SESSION_PATH) - L.DEBUG("Logged in and saved new session.") + logger.debug("Logged in and saved new session.") else: raise Exception(f"Failed to login as {IG_USERNAME}.") diff --git a/sijapi/routers/img.py b/sijapi/routers/img.py index 7748a44..a58c5f0 100644 --- a/sijapi/routers/img.py +++ b/sijapi/routers/img.py @@ -33,7 +33,7 @@ from sijapi.routers.llm import query_ollama from sijapi import API, L, COMFYUI_URL, COMFYUI_OUTPUT_DIR, IMG_CONFIG_PATH, IMG_DIR, IMG_WORKFLOWS_DIR img = APIRouter() - +logger = L.get_module_logger("img") CLIENT_ID = str(uuid.uuid4()) @img.post("/img") @@ -79,12 +79,12 @@ async def workflow(prompt: str, scene: str = None, size: str = None, earlyout: s scene_workflow = random.choice(scene_data['workflows']) if size: - L.DEBUG(f"Specified size: {size}") + logger.debug(f"Specified size: {size}") size = size if size else scene_workflow.get('size', '1024x1024') width, height = map(int, size.split('x')) - L.DEBUG(f"Parsed width: {width}; parsed height: {height}") + logger.debug(f"Parsed width: {width}; parsed height: {height}") workflow_path = Path(IMG_WORKFLOWS_DIR) / scene_workflow['workflow'] workflow_data = json.loads(workflow_path.read_text()) @@ -108,12 +108,12 @@ async def workflow(prompt: str, scene: str = None, size: str = None, earlyout: s if earlyout: asyncio.create_task(generate_and_save_image(prompt_id, saved_file_key, max_size, destination_path)) - L.DEBUG(f"Returning {destination_path}") + logger.debug(f"Returning {destination_path}") return destination_path else: await generate_and_save_image(prompt_id, saved_file_key, max_size, destination_path) - L.DEBUG(f"Returning {destination_path}") + logger.debug(f"Returning {destination_path}") return destination_path @@ -124,7 +124,7 @@ async def generate_and_save_image(prompt_id, saved_file_key, max_size, destinati jpg_file_path = await save_as_jpg(image_data, prompt_id, quality=90, max_size=max_size, destination_path=destination_path) if Path(jpg_file_path) != Path(destination_path): - L.ERR(f"Mismatch between jpg_file_path, {jpg_file_path}, and detination_path, {destination_path}") + logger.error(f"Mismatch between jpg_file_path, {jpg_file_path}, and detination_path, {destination_path}") except Exception as e: print(f"Error in generate_and_save_image: {e}") @@ -216,11 +216,11 @@ def set_presets(workflow_data, preset_values): if 'inputs' in workflow_data.get(preset_node, {}): workflow_data[preset_node]['inputs'][preset_key] = preset_value else: - L.DEBUG("Node not found in workflow_data") + logger.debug("Node not found in workflow_data") else: - L.DEBUG("Required data missing in preset_values") + logger.debug("Required data missing in preset_values") else: - L.DEBUG("No preset_values found") + logger.debug("No preset_values found") def get_return_path(destination_path): @@ -235,7 +235,7 @@ def get_scene(scene): IMG_CONFIG = yaml.safe_load(IMG_CONFIG_file) for scene_data in IMG_CONFIG['scenes']: if scene_data['scene'] == scene: - L.DEBUG(f"Found scene for \"{scene}\".") + logger.debug(f"Found scene for \"{scene}\".") return scene_data return None @@ -254,11 +254,11 @@ def get_matching_scene(prompt): max_count = count scene_data = sc if scene_data: - L.DEBUG(f"Found better-matching scene: the prompt contains {max_count} words that match triggers for {scene_data.get('name')}!") + logger.debug(f"Found better-matching scene: the prompt contains {max_count} words that match triggers for {scene_data.get('name')}!") if scene_data: return scene_data else: - L.DEBUG(f"No matching scenes found, falling back to default scene.") + logger.debug(f"No matching scenes found, falling back to default scene.") return IMG_CONFIG['scenes'][0] @@ -326,10 +326,10 @@ async def ensure_comfy(retries: int = 4, timeout: float = 6.0): # shareable_link = f"https://{PHOTOPRISM_URL}/p/{photo_uuid}" # return shareable_link # else: - # L.ERR("Could not find the uploaded photo details.") + # logger.error("Could not find the uploaded photo details.") # return None # except Exception as e: - # L.ERR(f"Error in upload_and_get_shareable_link: {e}") + # logger.error(f"Error in upload_and_get_shareable_link: {e}") # return None @@ -436,13 +436,13 @@ Even more important, it finds and returns the key to the filepath where the file workflow[key] = random.randint(1000000000000, 9999999999999) elif key in ["width", "max_width", "scaled_width", "height", "max_height", "scaled_height", "side_length", "size", "value", "dimension", "dimensions", "long", "long_side", "short", "short_side", "length"]: - L.DEBUG(f"Got a hit for a dimension: {key} {value}") + logger.debug(f"Got a hit for a dimension: {key} {value}") if value == 1023: workflow[key] = post.get("width", 1024) - L.DEBUG(f"Set {key} to {workflow[key]}.") + logger.debug(f"Set {key} to {workflow[key]}.") elif value == 1025: workflow[key] = post.get("height", 1024) - L.DEBUG(f"Set {key} to {workflow[key]}.") + logger.debug(f"Set {key} to {workflow[key]}.") update_recursive(workflow) return found_key[0] diff --git a/sijapi/routers/llm.py b/sijapi/routers/llm.py index 4a19e3b..db10a72 100644 --- a/sijapi/routers/llm.py +++ b/sijapi/routers/llm.py @@ -31,8 +31,8 @@ from sijapi.utilities import convert_to_unix_time, sanitize_filename, ocr_pdf, c from sijapi.routers import tts from sijapi.routers.asr import transcribe_audio - llm = APIRouter() +logger = L.get_module_logger("llm") # Initialize chromadb client client = chromadb.Client() @@ -89,13 +89,13 @@ async def query_ollama(usr: str, sys: str = LLM_SYS_MSG, model: str = DEFAULT_LL LLM = Ollama() response = await LLM.chat(model=model, messages=messages, options={"num_predict": max_tokens}) - L.DEBUG(response) + logger.debug(response) if "message" in response: if "content" in response["message"]: content = response["message"]["content"] return content else: - L.DEBUG("No choices found in response") + logger.debug("No choices found in response") return None async def query_ollama_multishot( @@ -116,12 +116,12 @@ async def query_ollama_multishot( LLM = Ollama() response = await LLM.chat(model=model, messages=messages, options={"num_predict": max_tokens}) - L.DEBUG(response) + logger.debug(response) if "message" in response and "content" in response["message"]: return response["message"]["content"] else: - L.DEBUG("No content found in response") + logger.debug("No content found in response") return None @@ -140,21 +140,21 @@ async def chat_completions(request: Request): raise HTTPException(status_code=400, detail="Message data is required in the request body.") requested_model = body.get('model', 'default-model') - L.DEBUG(f"Requested model: {requested_model}") + logger.debug(f"Requested model: {requested_model}") stream = body.get('stream') token_limit = body.get('max_tokens') or body.get('num_predict') # Check if the most recent message contains an image_url recent_message = messages[-1] if recent_message.get('role') == 'user' and is_vision_request(recent_message.get('content')): - L.DEBUG("Processing as a vision request") + logger.debug("Processing as a vision request") model = "llava" - L.DEBUG(f"Using model: {model}") + logger.debug(f"Using model: {model}") return StreamingResponse(stream_messages_with_vision(recent_message, model, token_limit), media_type="application/json") else: - L.DEBUG("Processing as a standard request") + logger.debug("Processing as a standard request") model = requested_model - L.DEBUG(f"Using model: {model}") + logger.debug(f"Using model: {model}") if stream: return StreamingResponse(stream_messages(messages, model, token_limit), media_type="application/json") else: @@ -279,17 +279,17 @@ async def generate_messages(messages: list, model: str = "llama3"): def is_model_available(model_name): model_data = OllamaList() available_models = [model['name'] for model in model_data['models']] - L.DEBUG(f"Available models: {available_models}") # Log using the configured LOGGER + logger.debug(f"Available models: {available_models}") # Log using the configured LOGGER matching_models = [model for model in available_models if model.startswith(model_name + ':') or model == model_name] if len(matching_models) == 1: - L.DEBUG(f"Unique match found: {matching_models[0]}") + logger.debug(f"Unique match found: {matching_models[0]}") return True elif len(matching_models) > 1: - L.ERR(f"Ambiguous match found, models: {matching_models}") + logger.error(f"Ambiguous match found, models: {matching_models}") return True else: - L.ERR(f"No match found for model: {model_name}") + logger.error(f"No match found for model: {model_name}") return False @@ -412,12 +412,12 @@ def query_gpt4(llmPrompt: List = [], system_msg: str = "", user_msg: str = "", m if hasattr(first_choice, "message") and hasattr(first_choice.message, "content"): return first_choice.message.content else: - L.DEBUG("No content attribute in the first choice's message") - L.DEBUG(f"No content found in message string: {response.choices}") - L.DEBUG("Trying again!") + logger.debug("No content attribute in the first choice's message") + logger.debug(f"No content found in message string: {response.choices}") + logger.debug("Trying again!") query_gpt4(messages, max_tokens) else: - L.DEBUG(f"No content found in message string: {response}") + logger.debug(f"No content found in message string: {response}") return "" def llava(image_base64, prompt): @@ -427,7 +427,7 @@ def llava(image_base64, prompt): prompt = f"This is a chat between a user and an assistant. The assistant is helping the user to describe an image. {prompt}", images = [image_base64] ) - L.DEBUG(response) + logger.debug(response) return "" if "pass" in response["response"].lower() else response["response"] def gpt4v(image_base64, prompt_sys: str, prompt_usr: str, max_tokens: int = 150): @@ -458,7 +458,7 @@ def gpt4v(image_base64, prompt_sys: str, prompt_usr: str, max_tokens: int = 150) comment_content = first_choice.message.content if "PASS" in comment_content: return "" - L.DEBUG(f"Generated comment: {comment_content}") + logger.debug(f"Generated comment: {comment_content}") response_2 = VISION_LLM.chat.completions.create( model="gpt-4-vision-preview", @@ -496,15 +496,15 @@ def gpt4v(image_base64, prompt_sys: str, prompt_usr: str, max_tokens: int = 150) first_choice = response_2.choices[0] if first_choice.message and first_choice.message.content: final_content = first_choice.message.content - L.DEBUG(f"Generated comment: {final_content}") + logger.debug(f"Generated comment: {final_content}") if "PASS" in final_content: return "" else: return final_content - L.DEBUG("Vision response did not contain expected data.") - L.DEBUG(f"Vision response: {response_1}") + logger.debug("Vision response did not contain expected data.") + logger.debug(f"Vision response: {response_1}") asyncio.sleep(15) try_again = gpt4v(image_base64, prompt_sys, prompt_usr, max_tokens) @@ -562,7 +562,7 @@ async def summarize_tts_endpoint( ) except Exception as e: - L.ERR(f"Error in summarize_tts_endpoint: {str(e)}") + logger.error(f"Error in summarize_tts_endpoint: {str(e)}") return JSONResponse( status_code=400, content={"error": str(e)} @@ -589,7 +589,7 @@ async def summarize_tts( bg_tasks = BackgroundTasks() model = await tts.get_model(voice) final_output_path = await tts.generate_speech(bg_tasks, summarized_text, voice, model=model, speed=speed, podcast=podcast, title=filename) - L.DEBUG(f"summary_tts completed with final_output_path: {final_output_path}") + logger.debug(f"summary_tts completed with final_output_path: {final_output_path}") return final_output_path @@ -605,10 +605,10 @@ def split_text_into_chunks(text: str) -> List[str]: sentences = re.split(r'(?<=[.!?])\s+', text) words = text.split() total_words = len(words) - L.DEBUG(f"Total words: {total_words}. SUMMARY_CHUNK_SIZE: {SUMMARY_CHUNK_SIZE}. SUMMARY_TPW: {SUMMARY_TPW}.") + logger.debug(f"Total words: {total_words}. SUMMARY_CHUNK_SIZE: {SUMMARY_CHUNK_SIZE}. SUMMARY_TPW: {SUMMARY_TPW}.") max_words_per_chunk = int(SUMMARY_CHUNK_SIZE / SUMMARY_TPW) - L.DEBUG(f"Maximum words per chunk: {max_words_per_chunk}") + logger.debug(f"Maximum words per chunk: {max_words_per_chunk}") chunks = [] current_chunk = [] @@ -628,7 +628,7 @@ def split_text_into_chunks(text: str) -> List[str]: if current_chunk: chunks.append(' '.join(current_chunk)) - L.DEBUG(f"Split text into {len(chunks)} chunks.") + logger.debug(f"Split text into {len(chunks)} chunks.") return chunks @@ -640,11 +640,11 @@ def calculate_max_tokens(text: str) -> int: async def extract_text(file: Union[UploadFile, bytes, bytearray, str, Path], bg_tasks: BackgroundTasks = None) -> str: - L.INFO(f"Attempting to extract text from file: {file}") + logger.info(f"Attempting to extract text from file: {file}") try: if isinstance(file, UploadFile): - L.INFO("File is an UploadFile object") + logger.info("File is an UploadFile object") file_extension = os.path.splitext(file.filename)[1] temp_file_path = tempfile.mktemp(suffix=file_extension) with open(temp_file_path, 'wb') as buffer: @@ -663,7 +663,7 @@ async def extract_text(file: Union[UploadFile, bytes, bytearray, str, Path], bg_ _, file_ext = os.path.splitext(file_path) file_ext = file_ext.lower() - L.INFO(f"File extension: {file_ext}") + logger.info(f"File extension: {file_ext}") if file_ext == '.pdf': text_content = await extract_text_from_pdf(file_path) @@ -690,7 +690,7 @@ async def extract_text(file: Union[UploadFile, bytes, bytearray, str, Path], bg_ return text_content except Exception as e: - L.ERR(f"Error extracting text: {str(e)}") + logger.error(f"Error extracting text: {str(e)}") raise ValueError(f"Error extracting text: {str(e)}") @@ -699,17 +699,17 @@ async def summarize_text(text: str, instruction: str = SUMMARY_INSTRUCT, length_ chunked_text = split_text_into_chunks(text) total_parts = len(chunked_text) - L.DEBUG(f"Total parts: {total_parts}. Length of chunked text: {len(chunked_text)}") + logger.debug(f"Total parts: {total_parts}. Length of chunked text: {len(chunked_text)}") total_words_count = sum(len(chunk.split()) for chunk in chunked_text) - L.DEBUG(f"Total words count: {total_words_count}") + logger.debug(f"Total words count: {total_words_count}") total_tokens_count = max(1, int(total_words_count * SUMMARY_TPW)) - L.DEBUG(f"Total tokens count: {total_tokens_count}") + logger.debug(f"Total tokens count: {total_tokens_count}") total_summary_length = length_override if length_override else total_tokens_count // length_quotient - L.DEBUG(f"Total summary length: {total_summary_length}") + logger.debug(f"Total summary length: {total_summary_length}") corrected_total_summary_length = min(total_summary_length, SUMMARY_TOKEN_LIMIT) - L.DEBUG(f"Corrected total summary length: {corrected_total_summary_length}") + logger.debug(f"Corrected total summary length: {corrected_total_summary_length}") summaries = await asyncio.gather(*[ process_chunk(instruction, chunk, i+1, total_parts, LLM=LLM) @@ -720,21 +720,21 @@ async def summarize_text(text: str, instruction: str = SUMMARY_INSTRUCT, length_ summaries = [f"\n\n\nPART {i+1} of {total_parts}:\n\n{summary}" for i, summary in enumerate(summaries)] concatenated_summary = ' '.join(summaries) - L.DEBUG(f"Concatenated summary: {concatenated_summary}") - L.DEBUG(f"Concatenated summary length: {len(concatenated_summary.split())}") + logger.debug(f"Concatenated summary: {concatenated_summary}") + logger.debug(f"Concatenated summary length: {len(concatenated_summary.split())}") if total_parts > 1: - L.DEBUG(f"Processing the concatenated_summary to smooth the edges...") + logger.debug(f"Processing the concatenated_summary to smooth the edges...") concatenated_instruct = f"The following text consists of the concatenated {total_parts} summaries of {total_parts} parts of a single document that had to be split for processing. Reword it for clarity and flow as a single cohesive summary, understanding that it all relates to a single document, but that document likely consists of multiple parts potentially from multiple authors. Do not shorten it and do not omit content, simply smooth out the edges between the parts." final_summary = await process_chunk(concatenated_instruct, concatenated_summary, 1, 1, length_ratio=1, LLM=LLM) - L.DEBUG(f"Final summary length: {len(final_summary.split())}") + logger.debug(f"Final summary length: {len(final_summary.split())}") return final_summary else: return concatenated_summary async def process_chunk(instruction: str, text: str, part: int, total_parts: int, length_ratio: float = None, LLM: Ollama = None) -> str: - # L.DEBUG(f"Processing chunk: {text}") + # logger.debug(f"Processing chunk: {text}") LLM = LLM if LLM else Ollama() words_count = len(text.split()) @@ -744,14 +744,14 @@ async def process_chunk(instruction: str, text: str, part: int, total_parts: int max_tokens = min(tokens_count // summary_length_ratio, SUMMARY_CHUNK_SIZE) max_tokens = max(max_tokens, SUMMARY_MIN_LENGTH) - L.DEBUG(f"Processing part {part} of {total_parts}: Words: {words_count}, Estimated tokens: {tokens_count}, Max output tokens: {max_tokens}") + logger.debug(f"Processing part {part} of {total_parts}: Words: {words_count}, Estimated tokens: {tokens_count}, Max output tokens: {max_tokens}") if part and total_parts > 1: prompt = f"{instruction}. Part {part} of {total_parts}:\n{text}" else: prompt = f"{instruction}:\n\n{text}" - L.DEBUG(f"Starting LLM.generate for part {part} of {total_parts}") + logger.debug(f"Starting LLM.generate for part {part} of {total_parts}") response = await LLM.generate( model=SUMMARY_MODEL, prompt=prompt, @@ -760,8 +760,8 @@ async def process_chunk(instruction: str, text: str, part: int, total_parts: int ) text_response = response['response'] - L.DEBUG(f"Completed LLM.generate for part {part} of {total_parts}") - L.DEBUG(f"Result: {text_response}") + logger.debug(f"Completed LLM.generate for part {part} of {total_parts}") + logger.debug(f"Result: {text_response}") return text_response async def title_and_summary(extracted_text: str): diff --git a/sijapi/routers/loc.py b/sijapi/routers/loc.py index 7b94ccc..78684c4 100644 --- a/sijapi/routers/loc.py +++ b/sijapi/routers/loc.py @@ -18,7 +18,7 @@ from sijapi.classes import Location from sijapi.utilities import haversine loc = APIRouter() - +logger = L.get_module_logger("loc") async def dt( date_time: Union[str, int, datetime], @@ -28,11 +28,11 @@ async def dt( # Convert integer (epoch time) to UTC datetime if isinstance(date_time, int): date_time = datetime.utcfromtimestamp(date_time).replace(tzinfo=timezone.utc) - L.DEBUG(f"Converted epoch time {date_time} to UTC datetime object.") + logger.debug(f"Converted epoch time {date_time} to UTC datetime object.") # Convert string to datetime if necessary elif isinstance(date_time, str): date_time = dateutil_parse(date_time) - L.DEBUG(f"Converted string '{date_time}' to datetime object.") + logger.debug(f"Converted string '{date_time}' to datetime object.") if not isinstance(date_time, datetime): raise ValueError(f"Input must be a string, integer (epoch time), or datetime object. What we received: {date_time}, type {type(date_time)}") @@ -40,7 +40,7 @@ async def dt( # Ensure the datetime is timezone-aware (UTC if not specified) if date_time.tzinfo is None: date_time = date_time.replace(tzinfo=timezone.utc) - L.DEBUG("Added UTC timezone to naive datetime.") + logger.debug("Added UTC timezone to naive datetime.") # Handle provided timezone if tz is not None: @@ -48,12 +48,12 @@ async def dt( if tz == "local": last_loc = await get_timezone_without_timezone(date_time) tz = await GEO.tz_at(last_loc.latitude, last_loc.longitude) - L.DEBUG(f"Using local timezone: {tz}") + logger.debug(f"Using local timezone: {tz}") else: try: tz = ZoneInfo(tz) except Exception as e: - L.ERR(f"Invalid timezone string '{tz}'. Error: {e}") + logger.error(f"Invalid timezone string '{tz}'. Error: {e}") raise ValueError(f"Invalid timezone string: {tz}") elif isinstance(tz, ZoneInfo): pass # tz is already a ZoneInfo object @@ -62,14 +62,14 @@ async def dt( # Convert to the provided or determined timezone date_time = date_time.astimezone(tz) - L.DEBUG(f"Converted datetime to timezone: {tz}") + logger.debug(f"Converted datetime to timezone: {tz}") return date_time except ValueError as e: - L.ERR(f"Error in dt: {e}") + logger.error(f"Error in dt: {e}") raise except Exception as e: - L.ERR(f"Unexpected error in dt: {e}") + logger.error(f"Unexpected error in dt: {e}") raise ValueError(f"Failed to process datetime: {e}") @@ -93,12 +93,12 @@ async def get_timezone_without_timezone(date_time): async def get_last_location() -> Optional[Location]: query_datetime = datetime.now(TZ) - L.DEBUG(f"Query_datetime: {query_datetime}") + logger.debug(f"Query_datetime: {query_datetime}") this_location = await fetch_last_location_before(query_datetime) if this_location: - L.DEBUG(f"location: {this_location}") + logger.debug(f"location: {this_location}") return this_location return None @@ -114,7 +114,7 @@ async def fetch_locations(start: datetime, end: datetime = None) -> List[Locatio if start_datetime.time() == datetime.min.time() and end_datetime.time() == datetime.min.time(): end_datetime = end_datetime.replace(hour=23, minute=59, second=59) - L.DEBUG(f"Fetching locations between {start_datetime} and {end_datetime}") + logger.debug(f"Fetching locations between {start_datetime} and {end_datetime}") async with DB.get_connection() as conn: locations = [] @@ -131,7 +131,7 @@ async def fetch_locations(start: datetime, end: datetime = None) -> List[Locatio ORDER BY datetime DESC ''', start_datetime.replace(tzinfo=None), end_datetime.replace(tzinfo=None)) - L.DEBUG(f"Range locations query returned: {range_locations}") + logger.debug(f"Range locations query returned: {range_locations}") locations.extend(range_locations) if not locations and (end is None or start_datetime.date() == end_datetime.date()): @@ -148,11 +148,11 @@ async def fetch_locations(start: datetime, end: datetime = None) -> List[Locatio LIMIT 1 ''', start_datetime.replace(tzinfo=None)) - L.DEBUG(f"Fallback query returned: {location_data}") + logger.debug(f"Fallback query returned: {location_data}") if location_data: locations.append(location_data) - L.DEBUG(f"Locations found: {locations}") + logger.debug(f"Locations found: {locations}") # Sort location_data based on the datetime field in descending order sorted_locations = sorted(locations, key=lambda x: x['datetime'], reverse=True) @@ -184,7 +184,7 @@ async def fetch_locations(start: datetime, end: datetime = None) -> List[Locatio async def fetch_last_location_before(datetime: datetime) -> Optional[Location]: datetime = await dt(datetime) - L.DEBUG(f"Fetching last location before {datetime}") + logger.debug(f"Fetching last location before {datetime}") async with DB.get_connection() as conn: @@ -204,10 +204,10 @@ async def fetch_last_location_before(datetime: datetime) -> Optional[Location]: await conn.close() if location_data: - L.DEBUG(f"Last location found: {location_data}") + logger.debug(f"Last location found: {location_data}") return Location(**location_data) else: - L.DEBUG("No location found before the specified datetime") + logger.debug("No location found before the specified datetime") return None @loc.get("/map/start_date={start_date_str}&end_date={end_date_str}", response_class=HTMLResponse) @@ -262,9 +262,9 @@ async def generate_map(start_date: datetime, end_date: datetime): async def post_location(location: Location): if not location.datetime: - L.DEBUG(f"location appears to be missing datetime: {location}") + logger.debug(f"location appears to be missing datetime: {location}") else: - L.DEBUG(f"post_location called with {location.datetime}") + logger.debug(f"post_location called with {location.datetime}") async with DB.get_connection() as conn: try: @@ -293,7 +293,7 @@ async def post_location(location: Location): location.suburb, location.county, location.country_code, location.country) await conn.close() - L.INFO(f"Successfully posted location: {location.latitude}, {location.longitude}, {location.elevation} on {localized_datetime}") + logger.info(f"Successfully posted location: {location.latitude}, {location.longitude}, {location.elevation} on {localized_datetime}") return { 'datetime': localized_datetime, 'latitude': location.latitude, @@ -323,8 +323,8 @@ async def post_location(location: Location): 'country': location.country } except Exception as e: - L.ERR(f"Error posting location {e}") - L.ERR(traceback.format_exc()) + logger.error(f"Error posting location {e}") + logger.error(traceback.format_exc()) return None @@ -347,26 +347,26 @@ async def post_locate_endpoint(locations: Union[Location, List[Location]]): "device_name": "Unknown", "device_os": "Unknown" } - L.DEBUG(f"Location received for processing: {lcn}") + logger.debug(f"Location received for processing: {lcn}") geocoded_locations = await GEO.code(locations) responses = [] if isinstance(geocoded_locations, List): for location in geocoded_locations: - L.DEBUG(f"Final location to be submitted to database: {location}") + logger.debug(f"Final location to be submitted to database: {location}") location_entry = await post_location(location) if location_entry: responses.append({"location_data": location_entry}) else: - L.WARN(f"Posting location to database appears to have failed.") + logger.warn(f"Posting location to database appears to have failed.") else: - L.DEBUG(f"Final location to be submitted to database: {geocoded_locations}") + logger.debug(f"Final location to be submitted to database: {geocoded_locations}") location_entry = await post_location(geocoded_locations) if location_entry: responses.append({"location_data": location_entry}) else: - L.WARN(f"Posting location to database appears to have failed.") + logger.warn(f"Posting location to database appears to have failed.") return {"message": "Locations and weather updated", "results": responses} @@ -387,7 +387,7 @@ async def get_locate(datetime_str: str, all: bool = False): try: date_time = await dt(datetime_str) except ValueError as e: - L.ERR(f"Invalid datetime string provided: {datetime_str}") + logger.error(f"Invalid datetime string provided: {datetime_str}") return ["ERROR: INVALID DATETIME PROVIDED. USE YYYYMMDDHHmmss or YYYYMMDD format."] locations = await fetch_locations(date_time) diff --git a/sijapi/routers/news.py b/sijapi/routers/news.py index 32097ca..0881e2a 100644 --- a/sijapi/routers/news.py +++ b/sijapi/routers/news.py @@ -34,9 +34,8 @@ from sijapi.routers import llm, tts, asr, loc from newspaper import Article - - news = APIRouter() +logger = L.get_module_logger("news") async def download_and_save_article(article, site_name, earliest_date, bg_tasks: BackgroundTasks, tts_mode: str = "summary", voice: str = DEFAULT_11L_VOICE): try: @@ -60,7 +59,7 @@ async def download_and_save_article(article, site_name, earliest_date, bg_tasks: try: article.publish_date = await loc.dt(article.publish_date, "UTC") except: - L.DEBUG(f"Failed to localize {article.publish_date}") + logger.debug(f"Failed to localize {article.publish_date}") article.publish_date = await loc.dt(dt_datetime.now(), "UTC") article.meta_description = traf.description if traf and traf.description else article.meta_description article.text = trafilatura.extract(source, output_format="markdown", include_comments=False) if source else article.text @@ -95,7 +94,7 @@ async def download_and_save_article(article, site_name, earliest_date, bg_tasks: if banner_image: banner_markdown = f"![[{OBSIDIAN_RESOURCES_DIR}/{banner_image}]]" except Exception as e: - L.ERR(f"No image found in article") + logger.error(f"No image found in article") authors = ', '.join(['[[{}]]'.format(author.strip()) for author in article.authors if author.strip()]) @@ -131,11 +130,11 @@ tags: obsidian_link = f"![[{audio_path.name}]]" body += f"{obsidian_link}\n\n" else: - L.WARN(f"Unexpected audio_path type: {type(audio_path)}. Value: {audio_path}") + logger.warn(f"Unexpected audio_path type: {type(audio_path)}. Value: {audio_path}") except Exception as e: - L.ERR(f"Failed to generate TTS for {audio_filename}. Error: {str(e)}") - L.ERR(f"TTS error details - voice: {voice}, model: eleven_turbo_v2, podcast: True") - L.ERR(f"Output directory: {Path(OBSIDIAN_VAULT_DIR) / OBSIDIAN_RESOURCES_DIR}") + logger.error(f"Failed to generate TTS for {audio_filename}. Error: {str(e)}") + logger.error(f"TTS error details - voice: {voice}, model: eleven_turbo_v2, podcast: True") + logger.error(f"Output directory: {Path(OBSIDIAN_VAULT_DIR) / OBSIDIAN_RESOURCES_DIR}") body += f"by {authors} in {article.source_url}\n\n" body += f"> [!summary]+\n" @@ -147,14 +146,14 @@ tags: with open(markdown_filename, 'w') as md_file: md_file.write(markdown_content) - L.INFO(f"Successfully saved to {markdown_filename}") + logger.info(f"Successfully saved to {markdown_filename}") add_to_daily_note(relative_path) print(f"Saved article: {relative_path}") return True except Exception as e: - L.ERR(f"Error processing article from {article.url}: {str(e)}") + logger.error(f"Error processing article from {article.url}: {str(e)}") return False # You'll need to update your is_article_within_date_range function: @@ -242,18 +241,18 @@ async def clip_get( voice: str = Query(DEFAULT_VOICE) ): parsed_content = await parse_article(url) - markdown_filename = await process_article(bg_tasks, parsed_content, tts, voice) + markdown_filename = await process_article2(bg_tasks, parsed_content, tts, voice) return {"message": "Clip saved successfully", "markdown_filename": markdown_filename} @news.post("/note/add") async def note_add_endpoint(file: Optional[UploadFile] = File(None), text: Optional[str] = Form(None), source: Optional[str] = Form(None), bg_tasks: BackgroundTasks = None): - L.DEBUG(f"Received request on /note/add...") + logger.debug(f"Received request on /note/add...") if not file and not text: - L.WARN(f"... without any file or text!") + logger.warn(f"... without any file or text!") raise HTTPException(status_code=400, detail="Either text or a file must be provided") else: result = await process_for_daily_note(file, text, source, bg_tasks) - L.INFO(f"Result on /note/add: {result}") + logger.info(f"Result on /note/add: {result}") return JSONResponse(result, status_code=204) async def process_for_daily_note(file: Optional[UploadFile] = File(None), text: Optional[str] = None, source: Optional[str] = None, bg_tasks: BackgroundTasks = None): @@ -261,7 +260,7 @@ async def process_for_daily_note(file: Optional[UploadFile] = File(None), text: transcription_entry = "" file_entry = "" if file: - L.DEBUG("File received...") + logger.debug("File received...") file_content = await file.read() audio_io = BytesIO(file_content) @@ -269,18 +268,18 @@ async def process_for_daily_note(file: Optional[UploadFile] = File(None), text: guessed_type = mimetypes.guess_type(file.filename) file_type = guessed_type[0] if guessed_type[0] else "application/octet-stream" - L.DEBUG(f"Processing as {file_type}...") + logger.debug(f"Processing as {file_type}...") # Extract the main type (e.g., 'audio', 'image', 'video') main_type = file_type.split('/')[0] subdir = main_type.title() if main_type else "Documents" absolute_path, relative_path = assemble_journal_path(now, subdir=subdir, filename=file.filename) - L.DEBUG(f"Destination path: {absolute_path}") + logger.debug(f"Destination path: {absolute_path}") with open(absolute_path, 'wb') as f: f.write(file_content) - L.DEBUG(f"Processing {f.name}...") + logger.debug(f"Processing {f.name}...") if main_type == 'audio': transcription = await asr.transcribe_audio(file_path=absolute_path, params=asr.TranscribeParams(model="small-en", language="en", threads=6)) @@ -291,7 +290,7 @@ async def process_for_daily_note(file: Optional[UploadFile] = File(None), text: file_entry = f"[Source]({relative_path})" text_entry = text if text else "" - L.DEBUG(f"transcription: {transcription_entry}\nfile_entry: {file_entry}\ntext_entry: {text_entry}") + logger.debug(f"transcription: {transcription_entry}\nfile_entry: {file_entry}\ntext_entry: {text_entry}") return await add_to_daily_note(transcription_entry, file_entry, text_entry, now) async def add_to_daily_note(transcription: str = None, file_link: str = None, additional_text: str = None, date_time: dt_datetime = None): @@ -383,7 +382,7 @@ added: {timestamp} obsidian_link = f"![[{OBSIDIAN_RESOURCES_DIR}/{audio_filename}{audio_ext}]]" body += f"{obsidian_link}\n\n" except Exception as e: - L.ERR(f"Failed in the TTS portion of clipping: {e}") + logger.error(f"Failed in the TTS portion of clipping: {e}") body += f"> [!summary]+\n" body += f"> {summary}\n\n" @@ -396,17 +395,17 @@ added: {timestamp} with open(markdown_filename, 'w', encoding=encoding) as md_file: md_file.write(markdown_content) - L.INFO(f"Successfully saved to {markdown_filename}") + logger.info(f"Successfully saved to {markdown_filename}") return markdown_filename except Exception as e: - L.ERR(f"Failed to clip: {str(e)}") + logger.error(f"Failed to clip: {str(e)}") raise HTTPException(status_code=500, detail=str(e)) -async def process_article( +async def process_article2( bg_tasks: BackgroundTasks, parsed_content: Article, tts_mode: str = "summary", @@ -437,7 +436,7 @@ async def process_article( banner_markdown = f"![[{OBSIDIAN_RESOURCES_DIR}/{banner_image}]]" except Exception as e: - L.ERR(f"No image found in article") + logger.error(f"No image found in article") authors = ', '.join('[[{}]]'.format(author) for author in parsed_content.authors) published_date = parsed_content.publish_date @@ -463,7 +462,7 @@ tags: obsidian_link = f"![[{OBSIDIAN_RESOURCES_DIR}/{audio_filename}{audio_ext}]]" body += f"{obsidian_link}\n\n" except Exception as e: - L.ERR(f"Failed to generate TTS for np3k. {e}") + logger.error(f"Failed to generate TTS for np3k. {e}") try: body += f"by {authors} in {parsed_content.canonical_link}" # update with method for getting the newspaper name @@ -473,26 +472,26 @@ tags: markdown_content = frontmatter + body except Exception as e: - L.ERR(f"Failed to combine elements of article markdown.") + logger.error(f"Failed to combine elements of article markdown.") try: with open(markdown_filename, 'w') as md_file: md_file.write(markdown_content) - L.INFO(f"Successfully saved to {markdown_filename}") + logger.info(f"Successfully saved to {markdown_filename}") add_to_daily_note return markdown_filename except Exception as e: - L.ERR(f"Failed to write markdown file") + logger.error(f"Failed to write markdown file") raise HTTPException(status_code=500, detail=str(e)) except Exception as e: - L.ERR(f"Failed to clip: {str(e)}") + logger.error(f"Failed to clip: {str(e)}") raise HTTPException(status_code=500, detail=str(e)) -async def process_article2( +async def process_article( bg_tasks: BackgroundTasks, url: str, title: Optional[str] = None, @@ -531,7 +530,7 @@ async def process_article2( banner_markdown = f"![[{OBSIDIAN_RESOURCES_DIR}/{banner_image}]]" except Exception as e: - L.ERR(f"No image found in article") + logger.error(f"No image found in article") authors = ', '.join('[[{}]]'.format(author) for author in parsed_content.get('authors', ['Unknown'])) @@ -560,7 +559,7 @@ tags: obsidian_link = f"![[{OBSIDIAN_RESOURCES_DIR}/{audio_filename}{audio_ext}]]" body += f"{obsidian_link}\n\n" except Exception as e: - L.ERR(f"Failed to generate TTS for np3k. {e}") + logger.error(f"Failed to generate TTS for np3k. {e}") try: body += f"by {authors} in [{parsed_content.get('domain', urlparse(url).netloc.replace('www.', ''))}]({url}).\n\n" @@ -570,22 +569,22 @@ tags: markdown_content = frontmatter + body except Exception as e: - L.ERR(f"Failed to combine elements of article markdown.") + logger.error(f"Failed to combine elements of article markdown.") try: with open(markdown_filename, 'w', encoding=encoding) as md_file: md_file.write(markdown_content) - L.INFO(f"Successfully saved to {markdown_filename}") + logger.info(f"Successfully saved to {markdown_filename}") add_to_daily_note return markdown_filename except Exception as e: - L.ERR(f"Failed to write markdown file") + logger.error(f"Failed to write markdown file") raise HTTPException(status_code=500, detail=str(e)) except Exception as e: - L.ERR(f"Failed to clip {url}: {str(e)}") + logger.error(f"Failed to clip {url}: {str(e)}") raise HTTPException(status_code=500, detail=str(e)) @@ -599,7 +598,7 @@ async def parse_article(url: str, source: Optional[str] = None) -> Article: article.set_html(source) article.parse() - L.INFO(f"Parsed {article.title}") + logger.info(f"Parsed {article.title}") # Update or set properties based on trafilatura and additional processing article.title = article.title or traf.title or url @@ -609,7 +608,7 @@ async def parse_article(url: str, source: Optional[str] = None) -> Article: try: article.publish_date = await loc.dt(article.publish_date, "UTC") except: - L.DEBUG(f"Failed to localize {article.publish_date}") + logger.debug(f"Failed to localize {article.publish_date}") article.publish_date = await loc.dt(dt_datetime.now(), "UTC") article.meta_description = article.meta_description or traf.description @@ -639,7 +638,7 @@ async def html_to_markdown(url: str = None, source: str = None) -> Optional[str] async with session.get(url) as response: html_content = await response.text() else: - L.ERR(f"Unable to convert nothing to markdown.") + logger.error(f"Unable to convert nothing to markdown.") return None # Use readability to extract the main content @@ -688,10 +687,10 @@ async def process_archive( markdown_path.parent.mkdir(parents=True, exist_ok=True) with open(markdown_path, 'w', encoding=encoding) as md_file: md_file.write(markdown_content) - L.DEBUG(f"Successfully saved to {markdown_path}") + logger.debug(f"Successfully saved to {markdown_path}") return markdown_path except Exception as e: - L.WARN(f"Failed to write markdown file: {str(e)}") + logger.warn(f"Failed to write markdown file: {str(e)}") return None def download_file(url, folder): @@ -715,13 +714,13 @@ def download_file(url, folder): with open(filepath, 'wb') as f: f.write(response.content) else: - L.ERR(f"Failed to download image: {url}, invalid content type: {response.headers.get('Content-Type')}") + logger.error(f"Failed to download image: {url}, invalid content type: {response.headers.get('Content-Type')}") return None else: - L.ERR(f"Failed to download image: {url}, status code: {response.status_code}") + logger.error(f"Failed to download image: {url}, status code: {response.status_code}") return None except Exception as e: - L.ERR(f"Failed to download image: {url}, error: {str(e)}") + logger.error(f"Failed to download image: {url}, error: {str(e)}") return None return filename diff --git a/sijapi/routers/note.py b/sijapi/routers/note.py index 1fd2f00..d07f761 100644 --- a/sijapi/routers/note.py +++ b/sijapi/routers/note.py @@ -21,6 +21,7 @@ from sijapi.classes import Location note = APIRouter() +logger = L.get_module_logger("note") def list_and_correct_impermissible_files(root_dir, rename: bool = False): """List and correct all files with impermissible names.""" @@ -30,7 +31,7 @@ def list_and_correct_impermissible_files(root_dir, rename: bool = False): if check_file_name(filename): file_path = Path(dirpath) / filename impermissible_files.append(file_path) - L.DEBUG(f"Impermissible file found: {file_path}") + logger.debug(f"Impermissible file found: {file_path}") # Sanitize the file name new_filename = sanitize_filename(filename) @@ -48,7 +49,7 @@ def list_and_correct_impermissible_files(root_dir, rename: bool = False): # Rename the file if rename: os.rename(file_path, new_file_path) - L.DEBUG(f"Renamed: {file_path} -> {new_file_path}") + logger.debug(f"Renamed: {file_path} -> {new_file_path}") return impermissible_files @@ -90,7 +91,7 @@ async def build_daily_note_endpoint( else: raise ValueError("Location is not provided or invalid.") except (ValueError, AttributeError, TypeError) as e: - L.WARN(f"Falling back to localized datetime due to error: {e}") + logger.warn(f"Falling back to localized datetime due to error: {e}") try: date_time = await loc.dt(date_str) places = await loc.fetch_locations(date_time) @@ -110,7 +111,7 @@ async def build_daily_note(date_time: dt_datetime, lat: float = None, lon: float Obsidian helper. Takes a datetime and creates a new daily note. Note: it uses the sijapi configuration file to place the daily note and does NOT presently interface with Obsidian's daily note or periodic notes extensions. It is your responsibility to ensure they match. ''' absolute_path, _ = assemble_journal_path(date_time) - L.WARN(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our datetime in build_daily_note.") + logger.warn(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our datetime in build_daily_note.") formatted_day = date_time.strftime("%A %B %d, %Y") # Monday May 27, 2024 formatting day_before = (date_time - timedelta(days=1)).strftime("%Y-%m-%d %A") # 2024-05-26 Sunday formatting day_after = (date_time + timedelta(days=1)).strftime("%Y-%m-%d %A") # 2024-05-28 Tuesday formatting @@ -201,7 +202,7 @@ async def update_frontmatter(date_time: dt_datetime, key: str, value: str): # Check if the file exists if not file_path.exists(): - L.CRIT(f"Markdown file not found at {file_path}") + logger.critical(f"Markdown file not found at {file_path}") raise HTTPException(status_code=404, detail="Markdown file not found.") # Read the file @@ -248,20 +249,20 @@ async def banner_endpoint(dt: str, location: str = None, forecast: str = None, m ''' Endpoint (POST) that generates a new banner image for the Obsidian daily note for a specified date, taking into account optional additional information, then updates the frontmatter if necessary. ''' - L.DEBUG(f"banner_endpoint requested with date: {dt} ({type(dt)})") + logger.debug(f"banner_endpoint requested with date: {dt} ({type(dt)})") date_time = await loc.dt(dt) - L.DEBUG(f"date_time after localization: {date_time} ({type(date_time)})") + logger.debug(f"date_time after localization: {date_time} ({type(date_time)})") context = await generate_context(dt, location, forecast, mood, other_context) jpg_path = await generate_banner(date_time, location, mood=mood, other_context=other_context) return jpg_path async def generate_banner(dt, location: Location = None, forecast: str = None, mood: str = None, other_context: str = None): - # L.DEBUG(f"Location: {location}, forecast: {forecast}, mood: {mood}, other_context: {other_context}") + # logger.debug(f"Location: {location}, forecast: {forecast}, mood: {mood}, other_context: {other_context}") date_time = await loc.dt(dt) - L.DEBUG(f"generate_banner called with date_time: {date_time}") + logger.debug(f"generate_banner called with date_time: {date_time}") destination_path, local_path = assemble_journal_path(date_time, filename="Banner", extension=".jpg", no_timestamp = True) - L.DEBUG(f"destination path generated: {destination_path}") + logger.debug(f"destination path generated: {destination_path}") if not location or not isinstance(location, Location): locations = await loc.fetch_locations(date_time) if locations: @@ -270,10 +271,10 @@ async def generate_banner(dt, location: Location = None, forecast: str = None, m forecast = await update_dn_weather(date_time, False, location.latitude, location.longitude) prompt = await generate_context(date_time, location, forecast, mood, other_context) - L.DEBUG(f"Prompt: {prompt}") + logger.debug(f"Prompt: {prompt}") final_path = await img.workflow(prompt, scene=OBSIDIAN_BANNER_SCENE, destination_path=destination_path) if not str(local_path) in str(final_path): - L.INFO(f"Apparent mismatch between local path, {local_path}, and final_path, {final_path}") + logger.info(f"Apparent mismatch between local path, {local_path}, and final_path, {final_path}") jpg_embed = f"\"![[{local_path}]]\"" await update_frontmatter(date_time, "banner", jpg_embed) return local_path @@ -301,7 +302,7 @@ async def generate_context(date_time, location: Location, forecast: str, mood: s if geocoded_location.display_name or geocoded_location.city or geocoded_location.country: return await generate_context(date_time, geocoded_location, forecast, mood, other_context) else: - L.WARN(f"Failed to get a useable location for purposes of generating a banner, but we'll generate one anyway.") + logger.warn(f"Failed to get a useable location for purposes of generating a banner, but we'll generate one anyway.") elif location and isinstance(location, str): display_name = f"Location: {location}\n" else: @@ -369,8 +370,8 @@ async def note_weather_get( force_refresh_weather = refresh == "True" try: date_time = dt_datetime.now() if date == "0" else await loc.dt(date) - L.WARN(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our dt_datetime in note_weather_get.") - L.DEBUG(f"date: {date} .. date_time: {date_time}") + logger.warn(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our dt_datetime in note_weather_get.") + logger.debug(f"date: {date} .. date_time: {date_time}") content = await update_dn_weather(date_time, force_refresh_weather) #, lat, lon) return JSONResponse(content={"forecast": content}, status_code=200) @@ -378,14 +379,14 @@ async def note_weather_get( return JSONResponse(content={"detail": str(e.detail)}, status_code=e.status_code) except Exception as e: - L.ERR(f"Error in note_weather_get: {str(e)}") + logger.error(f"Error in note_weather_get: {str(e)}") raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}") @note.post("/update/note/{date}") async def post_update_daily_weather_and_calendar_and_timeslips(date: str, refresh: str="False") -> PlainTextResponse: date_time = await loc.dt(date) - L.WARN(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our dt_datetime in post_update_daily_weather_and_calendar_and_timeslips.") + logger.warn(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our dt_datetime in post_update_daily_weather_and_calendar_and_timeslips.") force_refresh_weather = refresh == "True" await update_dn_weather(date_time, force_refresh_weather) await update_daily_note_events(date_time) @@ -393,52 +394,52 @@ async def post_update_daily_weather_and_calendar_and_timeslips(date: str, refres return f"[Refresh]({API.URL}/update/note/{date_time.strftime('%Y-%m-%d')}" async def update_dn_weather(date_time: dt_datetime, force_refresh: bool = False, lat: float = None, lon: float = None): - L.WARN(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our datetime in update_dn_weather.") + logger.warn(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our datetime in update_dn_weather.") try: if lat and lon: place = await GEO.code((lat, lon)) else: - L.DEBUG(f"Updating weather for {date_time}") + logger.debug(f"Updating weather for {date_time}") places = await loc.fetch_locations(date_time) place = places[0] lat = place.latitude lon = place.longitude - L.DEBUG(f"lat: {lat}, lon: {lon}, place: {place}") + logger.debug(f"lat: {lat}, lon: {lon}, place: {place}") city = GEO.find_override_location(lat, lon) if city: - L.INFO(f"Using override location: {city}") + logger.info(f"Using override location: {city}") else: if place.city and place.city != "": city = place.city - L.INFO(f"City in data: {city}") + logger.info(f"City in data: {city}") else: location = await GEO.code((lat, lon)) - L.DEBUG(f"location: {location}") + logger.debug(f"location: {location}") city = location.name city = city if city else location.city city = city if city else location.house_number + ' ' + location.road - L.DEBUG(f"City geocoded: {city}") + logger.debug(f"City geocoded: {city}") # Assemble journal path absolute_path, relative_path = assemble_journal_path(date_time, filename="Weather", extension=".md", no_timestamp = True) - L.DEBUG(f"Journal path: absolute_path={absolute_path}, relative_path={relative_path}") + logger.debug(f"Journal path: absolute_path={absolute_path}, relative_path={relative_path}") try: - L.DEBUG(f"passing date_time {date_time.strftime('%Y-%m-%d %H:%M:%S')}, {lat}/{lon} into get_weather") + logger.debug(f"passing date_time {date_time.strftime('%Y-%m-%d %H:%M:%S')}, {lat}/{lon} into get_weather") day = await weather.get_weather(date_time, lat, lon, force_refresh) - L.DEBUG(f"day information obtained from get_weather: {day}") + logger.debug(f"day information obtained from get_weather: {day}") if day: DailyWeather = day.get('DailyWeather') HourlyWeather = day.get('HourlyWeather') if DailyWeather: - # L.DEBUG(f"Day: {DailyWeather}") + # logger.debug(f"Day: {DailyWeather}") icon = DailyWeather.get('icon') - L.DEBUG(f"Icon: {icon}") + logger.debug(f"Icon: {icon}") weather_icon, admonition = get_icon_and_admonition(icon) if icon else (":LiSunMoon:", "ad-weather") @@ -507,38 +508,38 @@ async def update_dn_weather(date_time: dt_datetime, force_refresh: bool = False, detailed_forecast += assemble_hourly_data_table(times, condition_symbols, temps, winds) detailed_forecast += f"```\n\n" - L.DEBUG(f"Detailed forecast: {detailed_forecast}.") + logger.debug(f"Detailed forecast: {detailed_forecast}.") with open(absolute_path, 'w', encoding='utf-8') as note_file: note_file.write(detailed_forecast) - L.DEBUG(f"Operation complete.") + logger.debug(f"Operation complete.") return narrative else: - L.ERR(f"Failed to get DailyWeather from day: {day}") + logger.error(f"Failed to get DailyWeather from day: {day}") else: - L.ERR(f"Failed to get day") + logger.error(f"Failed to get day") raise HTTPException(status_code=500, detail="Failed to retrieve weather data") except HTTPException as e: - L.ERR(f"HTTP error: {e}") - L.ERR(traceback.format_exc()) + logger.error(f"HTTP error: {e}") + logger.error(traceback.format_exc()) raise e except Exception as e: - L.ERR(f"Error: {e}") - L.ERR(traceback.format_exc()) + logger.error(f"Error: {e}") + logger.error(traceback.format_exc()) raise HTTPException(status_code=999, detail=f"Error: {e}") except ValueError as ve: - L.ERR(f"Value error in update_dn_weather: {str(ve)}") - L.ERR(traceback.format_exc()) + logger.error(f"Value error in update_dn_weather: {str(ve)}") + logger.error(traceback.format_exc()) raise HTTPException(status_code=400, detail=f"Value error: {str(ve)}") except Exception as e: - L.ERR(f"Error in update_dn_weather: {str(e)}") - L.ERR(traceback.format_exc()) + logger.error(f"Error in update_dn_weather: {str(e)}") + logger.error(traceback.format_exc()) raise HTTPException(status_code=500, detail=f"Error in update_dn_weather: {str(e)}") def format_hourly_time(hour): @@ -546,8 +547,8 @@ def format_hourly_time(hour): hour_12 = convert_to_12_hour_format(hour.get("datetime")) return hour_12 except Exception as e: - L.ERR(f"Error in format_hourly_time: {str(e)}") - L.ERR(traceback.format_exc()) + logger.error(f"Error in format_hourly_time: {str(e)}") + logger.error(traceback.format_exc()) return "" def format_hourly_icon(hour, sunrise, sunset): @@ -557,7 +558,7 @@ def format_hourly_icon(hour, sunrise, sunset): precip = hour.get('precip', float(0.0)) precip_prob = hour.get('precipprob', float(0.0)) - L.DEBUG(f"precip: {precip}, prob: {precip_prob}") + logger.debug(f"precip: {precip}, prob: {precip_prob}") sp_str = None @@ -581,8 +582,8 @@ def format_hourly_icon(hour, sunrise, sunset): return formatted except Exception as e: - L.ERR(f"Error in format_hourly_special: {str(e)}") - L.ERR(traceback.format_exc()) + logger.error(f"Error in format_hourly_special: {str(e)}") + logger.error(traceback.format_exc()) return "" def format_hourly_temperature(hour): @@ -590,8 +591,8 @@ def format_hourly_temperature(hour): temp_str = f"{hour.get('temp', '')}˚ F" return temp_str except Exception as e: - L.ERR(f"Error in format_hourly_temperature: {str(e)}") - L.ERR(traceback.format_exc()) + logger.error(f"Error in format_hourly_temperature: {str(e)}") + logger.error(traceback.format_exc()) return "" def format_hourly_wind(hour): @@ -601,8 +602,8 @@ def format_hourly_wind(hour): wind_str = f"{str(windspeed)}:LiWind: {winddir}" return wind_str except Exception as e: - L.ERR(f"Error in format_hourly_wind: {str(e)}") - L.ERR(traceback.format_exc()) + logger.error(f"Error in format_hourly_wind: {str(e)}") + logger.error(traceback.format_exc()) return "" def assemble_hourly_data_table(times, condition_symbols, temps, winds): @@ -615,7 +616,7 @@ def assemble_hourly_data_table(times, condition_symbols, temps, winds): def get_icon_and_admonition(icon_str) -> Tuple: - L.DEBUG(f"Received request for emoji {icon_str}") + logger.debug(f"Received request for emoji {icon_str}") if icon_str.startswith(":") and icon_str.endswith(":"): return icon_str @@ -716,7 +717,7 @@ async def format_events_as_markdown(event_data: Dict[str, Union[str, List[Dict[s total_events = len(event_data["events"]) event_markdown = f"```ad-events" for event in event_data["events"]: - L.DEBUG(f"event busy status: {event['busy']}; all day status: {event['all_day']}") + logger.debug(f"event busy status: {event['busy']}; all day status: {event['all_day']}") if not event['name'].startswith('TC '): url = f"hook://ical/eventID={event['uid']}calendarID=17" if event['url']: @@ -794,18 +795,18 @@ async def note_events_endpoint(date: str = Query(None)): return PlainTextResponse(content=response, status_code=200) async def update_daily_note_events(date_time: dt_datetime): - L.DEBUG(f"Looking up events on date: {date_time.strftime('%Y-%m-%d')}") + logger.debug(f"Looking up events on date: {date_time.strftime('%Y-%m-%d')}") try: events = await cal.get_events(date_time, date_time) - L.DEBUG(f"Raw events: {events}") + logger.debug(f"Raw events: {events}") event_data = { "date": date_time.strftime('%Y-%m-%d'), "events": events } events_markdown = await format_events_as_markdown(event_data) - L.DEBUG(f"Markdown events: {events_markdown}") + logger.debug(f"Markdown events: {events_markdown}") absolute_path, _ = assemble_journal_path(date_time, filename="Events", extension=".md", no_timestamp = True) - L.DEBUG(f"Writing events to file: {absolute_path}") + logger.debug(f"Writing events to file: {absolute_path}") with open(absolute_path, 'w', encoding='utf-8') as note_file: note_file.write(events_markdown) @@ -813,7 +814,7 @@ async def update_daily_note_events(date_time: dt_datetime): return events_markdown except Exception as e: - L.ERR(f"Error processing events: {e}") + logger.error(f"Error processing events: {e}") raise HTTPException(status_code=500, detail=str(e)) diff --git a/sijapi/routers/rag.py b/sijapi/routers/rag.py index 41669d1..d4a616f 100644 --- a/sijapi/routers/rag.py +++ b/sijapi/routers/rag.py @@ -4,8 +4,10 @@ NOTES: Haven't yet decided if this should depend on the Obsidian and Chat module ''' from fastapi import APIRouter +from sijapi import L rag = APIRouter() +logger = L.get_module_logger("rag") rag.get("/rag/search") async def rag_search_endpoint(query: str, scope: str): diff --git a/sijapi/routers/serve.py b/sijapi/routers/serve.py index 17a4416..070d673 100644 --- a/sijapi/routers/serve.py +++ b/sijapi/routers/serve.py @@ -37,6 +37,7 @@ from sijapi.routers import loc, note serve = APIRouter(tags=["public"]) +logger = L.get_module_logger("serve") @serve.get("/pgp") async def get_pgp(): @@ -72,13 +73,13 @@ async def get_file_endpoint(file_path: str): date_time = await loc.dt(file_path); absolute_path, local_path = assemble_journal_path(date_time, no_timestamp = True) except ValueError as e: - L.DEBUG(f"Unable to parse {file_path} as a date, now trying to use it as a local path") + logger.debug(f"Unable to parse {file_path} as a date, now trying to use it as a local path") absolute_path = OBSIDIAN_VAULT_DIR / file_path if not absolute_path.suffix: absolute_path = Path(absolute_path.with_suffix(".md")) if not absolute_path.is_file(): - L.WARN(f"{absolute_path} is not a valid file it seems.") + logger.warn(f"{absolute_path} is not a valid file it seems.") elif absolute_path.suffix == '.md': try: with open(absolute_path, 'r', encoding='utf-8') as file: @@ -138,7 +139,7 @@ async def hook_changedetection(webhook_data: dict): @serve.post("/cl/search") async def hook_cl_search(request: Request, bg_tasks: BackgroundTasks): client_ip = request.client.host - L.DEBUG(f"Received request from IP: {client_ip}") + logger.debug(f"Received request from IP: {client_ip}") data = await request.json() payload = data['payload'] results = data['payload']['results'] @@ -156,7 +157,7 @@ async def hook_cl_search(request: Request, bg_tasks: BackgroundTasks): @serve.post("/cl/docket") async def hook_cl_docket(request: Request): client_ip = request.client.host - L.DEBUG(f"Received request from IP: {client_ip}") + logger.debug(f"Received request from IP: {client_ip}") data = await request.json() await cl_docket(data, client_ip) @@ -313,14 +314,14 @@ async def cl_docket_process_result(result, session): async with aiohttp.ClientSession() as session: async with session.get(url, headers=headers) as response: if response.status == 200: - L.DEBUG(f"Fetching CourtListener docket information for {docket}...") + logger.debug(f"Fetching CourtListener docket information for {docket}...") data = await response.json() court_docket = data['results'][0]['docket_number_core'] court_docket = f"{court_docket[:2]}-cv-{court_docket[2:]}" # Formatting the docket number case_name = data['results'][0]['case_name'] - L.DEBUG(f"Obtained from CourtListener: docket {court_docket}, case name {case_name}.") + logger.debug(f"Obtained from CourtListener: docket {court_docket}, case name {case_name}.") else: - L.DEBUG("Failed to fetch data from CourtListener API.") + logger.debug("Failed to fetch data from CourtListener API.") court_docket = 'NoCourtDocket' case_name = 'NoCaseName' @@ -330,12 +331,12 @@ async def cl_docket_process_result(result, session): if filepath_ia: file_url = filepath_ia - L.DEBUG(f"Found IA file at {file_url}.") + logger.debug(f"Found IA file at {file_url}.") elif filepath_local: file_url = f"{COURTLISTENER_BASE_URL}/{filepath_local}" - L.DEBUG(f"Found local file at {file_url}.") + logger.debug(f"Found local file at {file_url}.") else: - L.DEBUG(f"No file URL found in filepath_ia or filepath_local for one of the documents.") + logger.debug(f"No file URL found in filepath_ia or filepath_local for one of the documents.") continue document_number = document.get('document_number', 'NoDocumentNumber') @@ -346,7 +347,7 @@ async def cl_docket_process_result(result, session): target_path = Path(COURTLISTENER_DOCKETS_DIR) / case_shortname / "Docket" / file_name target_path.parent.mkdir(parents=True, exist_ok=True) await cl_download_file(file_url, target_path, session) - L.DEBUG(f"Downloaded {file_name} to {target_path}") + logger.debug(f"Downloaded {file_name} to {target_path}") def cl_case_details(docket): case_info = CASETABLE.get(str(docket), {"code": "000", "shortname": "UNKNOWN"}) @@ -359,18 +360,18 @@ async def cl_download_file(url: str, path: Path, session: aiohttp.ClientSession 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/93.0.4577.82 Safari/537.36' } async with aiohttp.ClientSession() as session: - L.DEBUG(f"Attempting to download {url} to {path}.") + logger.debug(f"Attempting to download {url} to {path}.") try: async with session.get(url, headers=headers, allow_redirects=True) as response: if response.status == 403: - L.ERR(f"Access denied (403 Forbidden) for URL: {url}. Skipping download.") + logger.error(f"Access denied (403 Forbidden) for URL: {url}. Skipping download.") return response.raise_for_status() # Check if the response content type is a PDF content_type = response.headers.get('Content-Type') if content_type != 'application/pdf': - L.ERR(f"Invalid content type: {content_type}. Skipping download.") + logger.error(f"Invalid content type: {content_type}. Skipping download.") return # Create an in-memory buffer to store the downloaded content @@ -385,7 +386,7 @@ async def cl_download_file(url: str, path: Path, session: aiohttp.ClientSession try: PdfReader(buffer) except Exception as e: - L.ERR(f"Invalid PDF content: {str(e)}. Skipping download.") + logger.error(f"Invalid PDF content: {str(e)}. Skipping download.") return # If the PDF is valid, write the content to the file on disk @@ -394,7 +395,7 @@ async def cl_download_file(url: str, path: Path, session: aiohttp.ClientSession file.write(buffer.getvalue()) except Exception as e: - L.ERR(f"Error downloading file: {str(e)}") + logger.error(f"Error downloading file: {str(e)}") async def cl_search_process_result(result): @@ -403,7 +404,7 @@ async def cl_search_process_result(result): court_id = result.get('court_id') case_name_short = result.get('caseNameShort') case_name = result.get('caseName') - L.DEBUG(f"Received payload for case {case_name} ({court_id}) and download url {download_url}") + logger.debug(f"Received payload for case {case_name} ({court_id}) and download url {download_url}") court_folder = court_id @@ -417,4 +418,4 @@ async def cl_search_process_result(result): target_path.parent.mkdir(parents=True, exist_ok=True) await cl_download_file(download_url, target_path, session) - L.DEBUG(f"Downloaded {file_name} to {target_path}") + logger.debug(f"Downloaded {file_name} to {target_path}") diff --git a/sijapi/routers/signal.py b/sijapi/routers/signal.py index 516e053..35b0556 100644 --- a/sijapi/routers/signal.py +++ b/sijapi/routers/signal.py @@ -2,9 +2,12 @@ Signal Bot example, repeats received messages. """ import os - +from fastapi import APIRouter from semaphore import Bot, ChatContext +from sijapi import L +signal = APIRouter() +logger = L.get_module_logger("signal") async def echo(ctx: ChatContext) -> None: if not ctx.message.empty(): diff --git a/sijapi/routers/time.py b/sijapi/routers/time.py index b9fa11b..26e1497 100644 --- a/sijapi/routers/time.py +++ b/sijapi/routers/time.py @@ -28,14 +28,9 @@ from traceback import format_exc from sijapi import L, TIMING_API_KEY, TIMING_API_URL from sijapi.routers import loc -### INITIALIZATIONS ### + time = APIRouter(tags=["private"]) - - -######################## -#### INITIALIZATION #### -######################## - +logger = L.get_module_logger("time") script_directory = os.path.dirname(os.path.abspath(__file__)) # Configuration constants @@ -63,17 +58,17 @@ async def post_time_entry_to_timing(entry: Dict): 'Accept': 'application/json', 'X-Time-Zone': 'America/Los_Angeles' } - L.DEBUG(f"Received entry: {entry}") + logger.debug(f"Received entry: {entry}") response = None # Initialize response try: async with httpx.AsyncClient() as client: response = await client.post(url, headers=headers, json=entry) response.raise_for_status() # This will only raise for 4xx and 5xx responses except httpx.HTTPStatusError as exc: - L.DEBUG(f"HTTPStatusError caught: Status code: {exc.response.status_code}, Detail: {exc.response.text}") + logger.debug(f"HTTPStatusError caught: Status code: {exc.response.status_code}, Detail: {exc.response.text}") raise HTTPException(status_code=exc.response.status_code, detail=str(exc.response.text)) except Exception as exc: - L.DEBUG(f"General exception caught: {exc}") + logger.debug(f"General exception caught: {exc}") raise HTTPException(status_code=500, detail="An unexpected error occurred") if response: diff --git a/sijapi/routers/tts.py b/sijapi/routers/tts.py index 2972607..2af272a 100644 --- a/sijapi/routers/tts.py +++ b/sijapi/routers/tts.py @@ -28,15 +28,12 @@ import os from sijapi import L, DEFAULT_VOICE, TTS_SEGMENTS_DIR, VOICE_DIR, PODCAST_DIR, TTS_OUTPUT_DIR, ELEVENLABS_API_KEY from sijapi.utilities import sanitize_filename - ### INITIALIZATIONS ### tts = APIRouter(tags=["trusted", "private"]) - +logger = L.get_module_logger("tts") DEVICE = torch.device('cpu') - MODEL_NAME = "tts_models/multilingual/multi-dataset/xtts_v2" - @tts.get("/tts/local_voices", response_model=List[str]) async def list_wav_files(): wav_files = [file.split('.')[0] for file in os.listdir(VOICE_DIR) if file.endswith(".wav")] @@ -50,7 +47,7 @@ async def list_11l_voices(): async with httpx.AsyncClient() as client: try: response = await client.get(url, headers=headers) - L.DEBUG(f"Response: {response}") + logger.debug(f"Response: {response}") if response.status_code == 200: voices_data = response.json().get("voices", []) formatted_list = "" @@ -60,7 +57,7 @@ async def list_11l_voices(): formatted_list += f"{name}: `{id}`\n" except Exception as e: - L.ERR(f"Error determining voice ID: {str(e)}") + logger.error(f"Error determining voice ID: {str(e)}") return PlainTextResponse(formatted_list, status_code=200) @@ -70,18 +67,18 @@ async def select_voice(voice_name: str) -> str: try: # Case Insensitive comparison voice_name_lower = voice_name.lower() - L.DEBUG(f"Looking for {voice_name_lower}") + logger.debug(f"Looking for {voice_name_lower}") for item in VOICE_DIR.iterdir(): - L.DEBUG(f"Checking {item.name.lower()}") + logger.debug(f"Checking {item.name.lower()}") if item.name.lower() == f"{voice_name_lower}.wav": - L.DEBUG(f"select_voice received query to use voice: {voice_name}. Found {item} inside {VOICE_DIR}.") + logger.debug(f"select_voice received query to use voice: {voice_name}. Found {item} inside {VOICE_DIR}.") return str(item) - L.ERR(f"Voice file not found") + logger.error(f"Voice file not found") raise HTTPException(status_code=404, detail="Voice file not found") except Exception as e: - L.ERR(f"Voice file not found: {str(e)}") + logger.error(f"Voice file not found: {str(e)}") return None @@ -116,8 +113,8 @@ async def generate_speech_endpoint( else: return await generate_speech(bg_tasks, text_content, voice, voice_file, model, speed, podcast) except Exception as e: - L.ERR(f"Error in TTS: {str(e)}") - L.ERR(traceback.format_exc()) + logger.error(f"Error in TTS: {str(e)}") + logger.error(traceback.format_exc()) raise HTTPException(status_code=666, detail="error in TTS") async def generate_speech( @@ -140,17 +137,17 @@ async def generate_speech( title = title if title else "TTS audio" output_path = output_dir / f"{dt_datetime.now().strftime('%Y%m%d%H%M%S')} {title}.wav" if model == "eleven_turbo_v2": - L.INFO("Using ElevenLabs.") + logger.info("Using ElevenLabs.") audio_file_path = await elevenlabs_tts(model, text, voice, title, output_dir) else: # if model == "xtts": - L.INFO("Using XTTS2") + logger.info("Using XTTS2") audio_file_path = await local_tts(text, speed, voice, voice_file, podcast, bg_tasks, title, output_path) #else: # raise HTTPException(status_code=400, detail="Invalid model specified") if podcast == True: podcast_path = Path(PODCAST_DIR) / audio_file_path.name - L.DEBUG(f"Podcast path: {podcast_path}") + logger.debug(f"Podcast path: {podcast_path}") shutil.copy(str(audio_file_path), str(podcast_path)) bg_tasks.add_task(os.remove, str(audio_file_path)) return str(podcast_path) @@ -158,7 +155,7 @@ async def generate_speech( return str(audio_file_path) except Exception as e: - L.ERR(f"Failed to generate speech: {str(e)}") + logger.error(f"Failed to generate speech: {str(e)}") raise HTTPException(status_code=500, detail=f"Failed to generate speech: {str(e)}") @@ -174,7 +171,7 @@ async def get_model(voice: str = None, voice_file: UploadFile = None): raise HTTPException(status_code=400, detail="No model or voice specified") async def determine_voice_id(voice_name: str) -> str: - L.DEBUG(f"Searching for voice id for {voice_name}") + logger.debug(f"Searching for voice id for {voice_name}") hardcoded_voices = { "alloy": "E3A1KVbKoWSIKSZwSUsW", @@ -191,23 +188,23 @@ async def determine_voice_id(voice_name: str) -> str: if voice_name in hardcoded_voices: voice_id = hardcoded_voices[voice_name] - L.DEBUG(f"Found voice ID - {voice_id}") + logger.debug(f"Found voice ID - {voice_id}") return voice_id - L.DEBUG(f"Requested voice not among the hardcoded options.. checking with 11L next.") + logger.debug(f"Requested voice not among the hardcoded options.. checking with 11L next.") url = "https://api.elevenlabs.io/v1/voices" headers = {"xi-api-key": ELEVENLABS_API_KEY} async with httpx.AsyncClient() as client: try: response = await client.get(url, headers=headers) - L.DEBUG(f"Response: {response}") + logger.debug(f"Response: {response}") if response.status_code == 200: voices_data = response.json().get("voices", []) for voice in voices_data: if voice_name == voice["voice_id"] or voice_name == voice["name"]: return voice["voice_id"] except Exception as e: - L.ERR(f"Error determining voice ID: {str(e)}") + logger.error(f"Error determining voice ID: {str(e)}") # as a last fallback, rely on David Attenborough return "b42GBisbu9r5m5n6pHF7" @@ -251,7 +248,7 @@ async def get_text_content(text: Optional[str], file: Optional[UploadFile]) -> s async def get_voice_file_path(voice: str = None, voice_file: UploadFile = None) -> str: if voice: - L.DEBUG(f"Looking for voice: {voice}") + logger.debug(f"Looking for voice: {voice}") selected_voice = await select_voice(voice) return selected_voice elif voice_file and isinstance(voice_file, UploadFile): @@ -280,7 +277,7 @@ async def get_voice_file_path(voice: str = None, voice_file: UploadFile = None) return str(new_file) else: - L.DEBUG(f"{dt_datetime.now().strftime('%Y%m%d%H%M%S')}: No voice specified or file provided, using default voice: {DEFAULT_VOICE}") + logger.debug(f"{dt_datetime.now().strftime('%Y%m%d%H%M%S')}: No voice specified or file provided, using default voice: {DEFAULT_VOICE}") selected_voice = await select_voice(DEFAULT_VOICE) return selected_voice @@ -318,7 +315,7 @@ async def local_tts( for i, segment in enumerate(segments): segment_file_path = TTS_SEGMENTS_DIR / f"segment_{i}.wav" - L.DEBUG(f"Segment file path: {segment_file_path}") + logger.debug(f"Segment file path: {segment_file_path}") # Run TTS in a separate thread await asyncio.to_thread( @@ -329,7 +326,7 @@ async def local_tts( speaker_wav=[voice_file_path], language="en" ) - L.DEBUG(f"Segment file generated: {segment_file_path}") + logger.debug(f"Segment file generated: {segment_file_path}") # Load and combine audio in a separate thread segment_audio = await asyncio.to_thread(AudioSegment.from_wav, str(segment_file_path)) @@ -404,7 +401,7 @@ def split_text(text, target_length=35, max_length=50): if segment_length + len(sentence_words) > max_length: segments.append(' '.join(current_segment)) - L.DEBUG(f"split_text - segment: {' '.join(current_segment)}, word count: {segment_length}") + logger.debug(f"split_text - segment: {' '.join(current_segment)}, word count: {segment_length}") current_segment = [sentence] else: @@ -412,7 +409,7 @@ def split_text(text, target_length=35, max_length=50): if current_segment: segments.append(' '.join(current_segment)) - L.DEBUG(f"split_text - segment: {' '.join(current_segment)}, word count: {len(current_segment)}") + logger.debug(f"split_text - segment: {' '.join(current_segment)}, word count: {len(current_segment)}") return segments @@ -424,7 +421,7 @@ def clean_text_for_tts(text: str) -> str: text = re.sub(r'\s+', ' ', text).strip() return text else: - L.DEBUG(f"No text received.") + logger.debug(f"No text received.") diff --git a/sijapi/routers/weather.py b/sijapi/routers/weather.py index c8093fd..eb6b2ff 100644 --- a/sijapi/routers/weather.py +++ b/sijapi/routers/weather.py @@ -16,6 +16,7 @@ from sijapi.utilities import haversine from sijapi.routers import loc weather = APIRouter() +logger = L.get_module_logger("weather") @weather.get("/weather/refresh", response_class=JSONResponse) async def get_refreshed_weather( @@ -34,7 +35,7 @@ async def get_refreshed_weather( tz = await GEO.tz_at(lat, lon) date_time = await loc.dt(date, tz) - L.DEBUG(f"passing date_time {date_time.strftime('%Y-%m-%d %H:%M:%S')}, {lat}/{lon} into get_weather") + logger.debug(f"passing date_time {date_time.strftime('%Y-%m-%d %H:%M:%S')}, {lat}/{lon} into get_weather") day = await get_weather(date_time, lat, lon, force_refresh=True) day_str = str(day) return JSONResponse(content={"weather": day_str}, status_code=200) @@ -43,18 +44,18 @@ async def get_refreshed_weather( return JSONResponse(content={"detail": str(e.detail)}, status_code=e.status_code) except Exception as e: - L.ERR(f"Error in note_weather_get: {str(e)}") + logger.error(f"Error in note_weather_get: {str(e)}") raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}") async def get_weather(date_time: dt_datetime, latitude: float, longitude: float, force_refresh: bool = False): - L.DEBUG(f"Called get_weather with lat: {latitude}, lon: {longitude}, date_time: {date_time}") - L.WARN(f"Using {date_time} as our datetime in get_weather.") + logger.debug(f"Called get_weather with lat: {latitude}, lon: {longitude}, date_time: {date_time}") + logger.warn(f"Using {date_time} as our datetime in get_weather.") fetch_new_data = True if force_refresh == False: daily_weather_data = await get_weather_from_db(date_time, latitude, longitude) if daily_weather_data: try: - L.DEBUG(f"Daily weather data from db: {daily_weather_data}") + logger.debug(f"Daily weather data from db: {daily_weather_data}") last_updated = str(daily_weather_data['DailyWeather'].get('last_updated')) last_updated = await loc.dt(last_updated) stored_loc_data = unhexlify(daily_weather_data['DailyWeather'].get('location')) @@ -65,67 +66,67 @@ async def get_weather(date_time: dt_datetime, latitude: float, longitude: float, hourly_weather = daily_weather_data.get('HourlyWeather') - L.DEBUG(f"Hourly: {hourly_weather}") + logger.debug(f"Hourly: {hourly_weather}") - L.DEBUG(f"\nINFO:\nlast updated {last_updated}\nstored lat: {stored_lat} - requested lat: {latitude}\nstored lon: {stored_lon} - requested lon: {longitude}\n") + logger.debug(f"\nINFO:\nlast updated {last_updated}\nstored lat: {stored_lat} - requested lat: {latitude}\nstored lon: {stored_lon} - requested lon: {longitude}\n") request_haversine = haversine(latitude, longitude, stored_lat, stored_lon) - L.DEBUG(f"\nINFO:\nlast updated {last_updated}\nstored lat: {stored_lat} - requested lat: {latitude}\nstored lon: {stored_lon} - requested lon: {longitude}\nHaversine: {request_haversine}") + logger.debug(f"\nINFO:\nlast updated {last_updated}\nstored lat: {stored_lat} - requested lat: {latitude}\nstored lon: {stored_lon} - requested lon: {longitude}\nHaversine: {request_haversine}") if last_updated and (date_time <= dt_datetime.now(TZ) and last_updated > date_time and request_haversine < 8) and hourly_weather and len(hourly_weather) > 0: - L.DEBUG(f"We can use existing data... :')") + logger.debug(f"We can use existing data... :')") fetch_new_data = False except Exception as e: - L.ERR(f"Error in get_weather: {e}") + logger.error(f"Error in get_weather: {e}") if fetch_new_data: - L.DEBUG(f"We require new data!") + logger.debug(f"We require new data!") request_date_str = date_time.strftime("%Y-%m-%d") - L.WARN(f"Using {date_time.strftime('%Y-%m-%d')} as our datetime for fetching new data.") + logger.warn(f"Using {date_time.strftime('%Y-%m-%d')} as our datetime for fetching new data.") url = f"https://weather.visualcrossing.com/VisualCrossingWebServices/rest/services/timeline/{latitude},{longitude}/{request_date_str}/{request_date_str}?unitGroup=us&key={VISUALCROSSING_API_KEY}" try: async with AsyncClient() as client: response = await client.get(url) if response.status_code == 200: - L.DEBUG(f"Successfully obtained data from VC...") + logger.debug(f"Successfully obtained data from VC...") try: weather_data = response.json() store_result = await store_weather_to_db(date_time, weather_data) if store_result == "SUCCESS": - L.DEBUG(f"New weather data for {request_date_str} stored in database...") + logger.debug(f"New weather data for {request_date_str} stored in database...") else: - L.ERR(f"Failed to store weather data for {request_date_str} in database! {store_result}") + logger.error(f"Failed to store weather data for {request_date_str} in database! {store_result}") - L.DEBUG(f"Attempting to retrieve data for {date_time}, {latitude}, {longitude}") + logger.debug(f"Attempting to retrieve data for {date_time}, {latitude}, {longitude}") daily_weather_data = await get_weather_from_db(date_time, latitude, longitude) if daily_weather_data is not None: return daily_weather_data else: raise HTTPException(status_code=500, detail="Weather data was not properly stored.") except Exception as e: - L.ERR(f"Problem parsing VC response or storing data: {e}") + logger.error(f"Problem parsing VC response or storing data: {e}") raise HTTPException(status_code=500, detail="Weather data was not properly stored.") else: - L.ERR(f"Failed to fetch weather data: {response.status_code}, {response.text}") + logger.error(f"Failed to fetch weather data: {response.status_code}, {response.text}") except Exception as e: - L.ERR(f"Exception during API call: {e}") + logger.error(f"Exception during API call: {e}") return daily_weather_data async def store_weather_to_db(date_time: dt_datetime, weather_data: dict): - L.WARN(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our datetime in store_weather_to_db") + logger.warn(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our datetime in store_weather_to_db") async with DB.get_connection() as conn: try: day_data = weather_data.get('days')[0] - L.DEBUG(f"RAW DAY_DATA: {day_data}") + logger.debug(f"RAW DAY_DATA: {day_data}") # Handle preciptype and stations as PostgreSQL arrays preciptype_array = day_data.get('preciptype', []) or [] stations_array = day_data.get('stations', []) or [] date_str = date_time.strftime("%Y-%m-%d") - L.WARN(f"Using {date_str} in our query in store_weather_to_db.") + logger.warn(f"Using {date_str} in our query in store_weather_to_db.") # Get location details from weather data if available longitude = weather_data.get('longitude') @@ -134,11 +135,11 @@ async def store_weather_to_db(date_time: dt_datetime, weather_data: dict): elevation = await GEO.elevation(latitude, longitude) location_point = f"POINTZ({longitude} {latitude} {elevation})" if longitude and latitude and elevation else None - L.WARN(f"Uncorrected datetimes in store_weather_to_db: {day_data['datetime']}, sunrise: {day_data['sunrise']}, sunset: {day_data['sunset']}") + logger.warn(f"Uncorrected datetimes in store_weather_to_db: {day_data['datetime']}, sunrise: {day_data['sunrise']}, sunset: {day_data['sunset']}") day_data['datetime'] = await loc.dt(day_data.get('datetimeEpoch')) day_data['sunrise'] = await loc.dt(day_data.get('sunriseEpoch')) day_data['sunset'] = await loc.dt(day_data.get('sunsetEpoch')) - L.WARN(f"Corrected datetimes in store_weather_to_db: {day_data['datetime']}, sunrise: {day_data['sunrise']}, sunset: {day_data['sunset']}") + logger.warn(f"Corrected datetimes in store_weather_to_db: {day_data['datetime']}, sunrise: {day_data['sunrise']}, sunset: {day_data['sunset']}") daily_weather_params = ( day_data.get('sunrise'), day_data.get('sunriseEpoch'), @@ -162,7 +163,7 @@ async def store_weather_to_db(date_time: dt_datetime, weather_data: dict): location_point ) except Exception as e: - L.ERR(f"Failed to prepare database query in store_weather_to_db! {e}") + logger.error(f"Failed to prepare database query in store_weather_to_db! {e}") try: daily_weather_query = ''' @@ -182,7 +183,7 @@ async def store_weather_to_db(date_time: dt_datetime, weather_data: dict): daily_weather_id = await conn.fetchval(daily_weather_query, *daily_weather_params) if 'hours' in day_data: - L.DEBUG(f"Processing hours now...") + logger.debug(f"Processing hours now...") for hour_data in day_data['hours']: try: await asyncio.sleep(0.01) @@ -228,22 +229,22 @@ async def store_weather_to_db(date_time: dt_datetime, weather_data: dict): ''' async with conn.transaction(): hourly_weather_id = await conn.fetchval(hourly_weather_query, *hourly_weather_params) - L.DEBUG(f"Done processing hourly_weather_id {hourly_weather_id}") + logger.debug(f"Done processing hourly_weather_id {hourly_weather_id}") except Exception as e: - L.ERR(f"EXCEPTION: {e}") + logger.error(f"EXCEPTION: {e}") except Exception as e: - L.ERR(f"EXCEPTION: {e}") + logger.error(f"EXCEPTION: {e}") return "SUCCESS" except Exception as e: - L.ERR(f"Error in dailyweather storage: {e}") + logger.error(f"Error in dailyweather storage: {e}") async def get_weather_from_db(date_time: dt_datetime, latitude: float, longitude: float): - L.WARN(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our datetime in get_weather_from_db.") + logger.warn(f"Using {date_time.strftime('%Y-%m-%d %H:%M:%S')} as our datetime in get_weather_from_db.") async with DB.get_connection() as conn: query_date = date_time.date() try: @@ -259,19 +260,19 @@ async def get_weather_from_db(date_time: dt_datetime, latitude: float, longitude daily_weather_record = await conn.fetchrow(query, query_date, longitude, latitude, longitude, latitude) if daily_weather_record is None: - L.DEBUG(f"No daily weather data retrieved from database.") + logger.debug(f"No daily weather data retrieved from database.") return None # Convert asyncpg.Record to a mutable dictionary daily_weather_data = dict(daily_weather_record) - # L.DEBUG(f"Daily weather data prior to tz corrections: {daily_weather_data}") + # logger.debug(f"Daily weather data prior to tz corrections: {daily_weather_data}") # Now we can modify the dictionary # tz = await GEO.tz_at(latitude, longitude) # daily_weather_data['datetime'] = await loc.dt(daily_weather_data.get('datetime'), tz) # daily_weather_data['sunrise'] = await loc.dt(daily_weather_data.get('sunrise'), tz) # daily_weather_data['sunset'] = await loc.dt(daily_weather_data.get('sunset'), tz) - # L.DEBUG(f"Daily weather data after tz corrections: {daily_weather_data}") + # logger.debug(f"Daily weather data after tz corrections: {daily_weather_data}") # Query to get hourly weather data query = ''' @@ -287,14 +288,14 @@ async def get_weather_from_db(date_time: dt_datetime, latitude: float, longitude # hour_data['datetime'] = await loc.dt(hour_data.get('datetime'), tz) hourly_weather_data.append(hour_data) - # L.DEBUG(f"Hourly weather data after tz corrections: {hourly_weather_data}") + # logger.debug(f"Hourly weather data after tz corrections: {hourly_weather_data}") day = { 'DailyWeather': daily_weather_data, 'HourlyWeather': hourly_weather_data, } - # L.DEBUG(f"day: {day}") + # logger.debug(f"day: {day}") return day except Exception as e: - L.ERR(f"Unexpected error occurred: {e}") + logger.error(f"Unexpected error occurred: {e}") diff --git a/sijapi/utilities.py b/sijapi/utilities.py index 6ee7fa4..c47de22 100644 --- a/sijapi/utilities.py +++ b/sijapi/utilities.py @@ -29,6 +29,7 @@ from fastapi import Depends, HTTPException, Request, UploadFile from fastapi.security.api_key import APIKeyHeader from sijapi import L, API, YEAR_FMT, MONTH_FMT, DAY_FMT, DAY_SHORT_FMT, OBSIDIAN_VAULT_DIR, ALLOWED_FILENAME_CHARS, MAX_PATH_LENGTH, ARCHIVE_DIR +logger = L.get_module_logger('utilities') api_key_header = APIKeyHeader(name="Authorization", auto_error=False) @@ -117,7 +118,7 @@ def assemble_journal_path(date_time: datetime, subdir: str = None, filename: str relative_path = relative_path / filename else: - L.DEBUG(f"This only happens, theoretically, when no filename nor subdirectory are provided, but an extension is. Which is kinda silly.") + logger.debug(f"This only happens, theoretically, when no filename nor subdirectory are provided, but an extension is. Which is kinda silly.") return None, None absolute_path = OBSIDIAN_VAULT_DIR / relative_path @@ -166,14 +167,14 @@ def get_extension(file): return file_extension except Exception as e: - L.ERR(f"Unable to get extension of {file}") + logger.error(f"Unable to get extension of {file}") raise e def sanitize_filename(text, extension: str = None, max_length: int = MAX_PATH_LENGTH): """Sanitize a string to be used as a safe filename while protecting the file extension.""" - L.DEBUG(f"Filename before sanitization: {text}") + logger.debug(f"Filename before sanitization: {text}") text = re.sub(r'\s+', ' ', text) sanitized = re.sub(ALLOWED_FILENAME_CHARS, '', text) @@ -185,7 +186,7 @@ def sanitize_filename(text, extension: str = None, max_length: int = MAX_PATH_LE base_name = base_name[:max_base_length - 5].rstrip() final_filename = base_name + extension - L.DEBUG(f"Filename after sanitization: {final_filename}") + logger.debug(f"Filename after sanitization: {final_filename}") return final_filename @@ -195,16 +196,16 @@ def check_file_name(file_name, max_length=255): needs_sanitization = False if len(file_name) > max_length: - L.DEBUG(f"Filename exceeds maximum length of {max_length}: {file_name}") + logger.debug(f"Filename exceeds maximum length of {max_length}: {file_name}") needs_sanitization = True if re.search(ALLOWED_FILENAME_CHARS, file_name): - L.DEBUG(f"Filename contains non-word characters (except space, dot, and hyphen): {file_name}") + logger.debug(f"Filename contains non-word characters (except space, dot, and hyphen): {file_name}") needs_sanitization = True if re.search(r'\s{2,}', file_name): - L.DEBUG(f"Filename contains multiple consecutive spaces: {file_name}") + logger.debug(f"Filename contains multiple consecutive spaces: {file_name}") needs_sanitization = True if file_name != file_name.strip(): - L.DEBUG(f"Filename has leading or trailing spaces: {file_name}") + logger.debug(f"Filename has leading or trailing spaces: {file_name}") needs_sanitization = True return needs_sanitization @@ -247,13 +248,13 @@ async def ocr_pdf(file_path: str) -> str: texts = await asyncio.gather(*(asyncio.to_thread(pytesseract.image_to_string, image) for image in images)) return ' '.join(texts) except Exception as e: - L.ERR(f"Error during OCR: {str(e)}") + logger.error(f"Error during OCR: {str(e)}") return "" async def extract_text_from_pdf(file_path: str) -> str: if not await is_valid_pdf(file_path): - L.ERR(f"Invalid PDF file: {file_path}") + logger.error(f"Invalid PDF file: {file_path}") return "" text = '' @@ -271,7 +272,7 @@ async def extract_text_from_pdf(file_path: str) -> str: if text and not should_use_ocr(text, num_pages): return clean_text(text) except Exception as e: - L.ERR(f"Error extracting text with PyPDF2: {str(e)}") + logger.error(f"Error extracting text with PyPDF2: {str(e)}") # If PyPDF2 extraction fails or is insufficient, fall back to pdfminer.six try: @@ -279,10 +280,10 @@ async def extract_text_from_pdf(file_path: str) -> str: if text_pdfminer and not should_use_ocr(text_pdfminer, num_pages): return clean_text(text_pdfminer) except Exception as e: - L.ERR(f"Error extracting text with pdfminer.six: {e}") + logger.error(f"Error extracting text with pdfminer.six: {e}") # If both methods fail or are deemed insufficient, use OCR as the last resort - L.DEBUG("Falling back to OCR for text extraction...") + logger.debug("Falling back to OCR for text extraction...") return await ocr_pdf(file_path) async def is_valid_pdf(file_path: str) -> bool: @@ -291,12 +292,12 @@ async def is_valid_pdf(file_path: str) -> bool: kind = filetype.guess(file_path) return kind.mime == 'application/pdf' except Exception as e: - L.ERR(f"Error checking file type: {e}") + logger.error(f"Error checking file type: {e}") return False async def extract_text_from_pdf(file_path: str) -> str: if not await is_valid_pdf(file_path): - L.ERR(f"Invalid PDF file: {file_path}") + logger.error(f"Invalid PDF file: {file_path}") return "" text = '' @@ -308,23 +309,23 @@ async def extract_text_from_pdf(file_path: str) -> str: if text.strip(): # Successfully extracted text return clean_text(text) except Exception as e: - L.ERR(f"Error extracting text with PyPDF2: {str(e)}") + logger.error(f"Error extracting text with PyPDF2: {str(e)}") try: text_pdfminer = await asyncio.to_thread(pdfminer_extract_text, file_path) if text_pdfminer.strip(): # Successfully extracted text return clean_text(text_pdfminer) except Exception as e: - L.ERR(f"Error extracting text with pdfminer.six: {str(e)}") + logger.error(f"Error extracting text with pdfminer.six: {str(e)}") # Fall back to OCR - L.DEBUG("Falling back to OCR for text extraction...") + logger.debug("Falling back to OCR for text extraction...") try: images = convert_from_path(file_path) ocr_texts = await asyncio.gather(*(asyncio.to_thread(pytesseract.image_to_string, img) for img in images)) return ' '.join(ocr_texts).strip() except Exception as e: - L.ERR(f"OCR failed: {str(e)}") + logger.error(f"OCR failed: {str(e)}") return "" async def extract_text_from_docx(file_path: str) -> str: @@ -427,7 +428,7 @@ def encode_image_to_base64(image_path): base64_str = base64.b64encode(byte_data).decode('utf-8') return base64_str else: - L.DEBUG(f"Error: File does not exist at {image_path}") + logger.debug(f"Error: File does not exist at {image_path}") def resize_and_convert_image(image_path, max_size=2160, quality=80): with Image.open(image_path) as img: @@ -471,5 +472,5 @@ async def run_ssh_command(server, command): ssh.close() return output, error except Exception as e: - L.ERR(f"SSH command failed for server {server.id}: {str(e)}") + logger.error(f"SSH command failed for server {server.id}: {str(e)}") raise \ No newline at end of file