Skip to content

openfoodfacts_proxy.cli

[docs] module openfoodfacts_proxy.cli

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
import logging

import click
import uvicorn

from openfoodfacts_proxy.__about__ import __version__
from openfoodfacts_proxy.core.sync_service_bundle import SyncServiceBundle
from openfoodfacts_proxy.infrastructure.settings import settings

logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(name)s: %(message)s")


@click.group()
@click.version_option(version=__version__)
def main() -> None:
    """OpenFoodFacts Proxy - A FastAPI reverse proxy with local caching."""


@main.command()
@click.option("--host", default="0.0.0.0", help="Host to bind to.")  # noqa: S104
@click.option("--port", default=8000, type=int, help="Port to bind to.")
@click.option("--reload", is_flag=True, default=False, help="Enable auto-reload for development.")
def serve(host: str, port: int, reload: bool) -> None:
    """Start the FastAPI proxy server."""
    uvicorn.run(
        "openfoodfacts_proxy.app:create_app",
        host=host,
        port=port,
        reload=reload,
        factory=True,
    )


@main.command("import-full")
def import_full() -> None:
    """Download and import the full MongoDB dump from OpenFoodFacts."""
    bundle = SyncServiceBundle(settings)
    click.echo("Starting full data import...")
    try:
        bundle.full_import_service.run()
        click.echo("Full import completed.")
    finally:
        bundle.close()


@main.command("import-delta")
def import_delta() -> None:
    """Download and import delta updates from OpenFoodFacts."""
    bundle = SyncServiceBundle(settings)
    click.echo("Starting delta import...")
    try:
        bundle.delta_import_service.run()
        click.echo("Delta import completed.")
    finally:
        bundle.close()