mirror of
https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl
synced 2025-08-05 06:00:39 +08:00
80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
"""
|
|
This is the Firecrawl package.
|
|
|
|
This package provides a Python SDK for interacting with the Firecrawl API.
|
|
It includes methods to scrape URLs, perform searches, initiate and monitor crawl jobs,
|
|
and check the status of these jobs.
|
|
|
|
For more information visit https://github.com/firecrawl/
|
|
"""
|
|
|
|
import logging
|
|
import os
|
|
|
|
from .firecrawl import FirecrawlApp # noqa
|
|
|
|
__version__ = "1.6.0"
|
|
|
|
# Define the logger for the Firecrawl project
|
|
logger: logging.Logger = logging.getLogger("firecrawl")
|
|
|
|
|
|
def _configure_logger() -> None:
|
|
"""
|
|
Configure the firecrawl logger for console output.
|
|
|
|
The function attaches a handler for console output with a specific format and date
|
|
format to the firecrawl logger.
|
|
"""
|
|
try:
|
|
# Create the formatter
|
|
formatter = logging.Formatter(
|
|
"[%(asctime)s - %(name)s:%(lineno)d - %(levelname)s] %(message)s",
|
|
datefmt="%Y-%m-%d %H:%M:%S",
|
|
)
|
|
|
|
# Create the console handler and set the formatter
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setFormatter(formatter)
|
|
|
|
# Add the console handler to the firecrawl logger
|
|
logger.addHandler(console_handler)
|
|
except Exception as e:
|
|
logger.error("Failed to configure logging: %s", e)
|
|
|
|
|
|
def setup_logging() -> None:
|
|
"""Set up logging based on the FIRECRAWL_LOGGING_LEVEL environment variable."""
|
|
# Check if the firecrawl logger already has a handler
|
|
if logger.hasHandlers():
|
|
return # To prevent duplicate logging
|
|
|
|
# Check if the FIRECRAWL_LOGGING_LEVEL environment variable is set
|
|
if not (env := os.getenv("FIRECRAWL_LOGGING_LEVEL", "").upper()):
|
|
# Attach a no-op handler to prevent warnings about no handlers
|
|
logger.addHandler(logging.NullHandler())
|
|
return
|
|
|
|
# Attach the console handler to the firecrawl logger
|
|
_configure_logger()
|
|
|
|
# Set the logging level based on the FIRECRAWL_LOGGING_LEVEL environment variable
|
|
if env == "DEBUG":
|
|
logger.setLevel(logging.DEBUG)
|
|
elif env == "INFO":
|
|
logger.setLevel(logging.INFO)
|
|
elif env == "WARNING":
|
|
logger.setLevel(logging.WARNING)
|
|
elif env == "ERROR":
|
|
logger.setLevel(logging.ERROR)
|
|
elif env == "CRITICAL":
|
|
logger.setLevel(logging.CRITICAL)
|
|
else:
|
|
logger.setLevel(logging.INFO)
|
|
logger.warning("Unknown logging level: %s, defaulting to INFO", env)
|
|
|
|
|
|
# Initialize logging configuration when the module is imported
|
|
setup_logging()
|
|
logger.debug("Debugging logger setup")
|