66827f8947
Remove unused imports (F401) and duplicate/shadowed import redefinitions (F811) across the codebase using ruff's safe autofixes. No behavioral changes -- imports only. - ~1400 safe autofixes applied across 644 files (net -1072 lines) - __init__.py re-exports preserved (excluded from F401 removal so public re-export surfaces stay intact) - Re-exports that are imported or monkeypatched by tests but look unused in their defining module are kept with explicit # noqa: F401 (gateway/run.py load_dotenv; run_agent re-exports from agent.message_sanitization, agent.context_compressor, agent.retry_utils, agent.prompt_builder, agent.process_bootstrap, agent.codex_responses_adapter) - Unsafe F841 (unused-variable) fixes deliberately skipped -- those can change behavior when the RHS has side effects - ruff lints remain disabled in pyproject.toml (only PLW1514 is selected); this is a one-time cleanup, not a config change Verification: - python -m compileall: clean - pytest --collect-only: all 27161 tests collect (zero import errors) - core entry points import clean (run_agent, model_tools, cli, toolsets, hermes_state, batch_runner, gateway) - static scan: every name any test imports directly from an edited module still resolves
85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
nutrition_search.py — Search USDA FoodData Central for nutrition info.
|
|
|
|
Usage:
|
|
python3 nutrition_search.py "chicken breast"
|
|
python3 nutrition_search.py "rice" "eggs" "broccoli"
|
|
echo -e "oats\\nbanana\\nwhey protein" | python3 nutrition_search.py -
|
|
|
|
Reads USDA_API_KEY from environment, falls back to DEMO_KEY.
|
|
No external dependencies.
|
|
"""
|
|
import sys
|
|
import os
|
|
import json
|
|
import time
|
|
import urllib.request
|
|
import urllib.parse
|
|
|
|
API_KEY = os.environ.get("USDA_API_KEY", "DEMO_KEY")
|
|
BASE = "https://api.nal.usda.gov/fdc/v1"
|
|
|
|
|
|
def search(query, max_results=3):
|
|
encoded = urllib.parse.quote(query)
|
|
url = (
|
|
f"{BASE}/foods/search?api_key={API_KEY}"
|
|
f"&query={encoded}&pageSize={max_results}"
|
|
f"&dataType=Foundation,SR%20Legacy"
|
|
)
|
|
try:
|
|
req = urllib.request.Request(url, headers={"Accept": "application/json"})
|
|
with urllib.request.urlopen(req, timeout=15) as r:
|
|
return json.loads(r.read())
|
|
except Exception as e:
|
|
print(f" API error: {e}", file=sys.stderr)
|
|
return None
|
|
|
|
|
|
def display(food):
|
|
nutrients = {n["nutrientName"]: n.get("value", "?") for n in food.get("foodNutrients", [])}
|
|
cal = nutrients.get("Energy", "?")
|
|
prot = nutrients.get("Protein", "?")
|
|
fat = nutrients.get("Total lipid (fat)", "?")
|
|
carb = nutrients.get("Carbohydrate, by difference", "?")
|
|
fib = nutrients.get("Fiber, total dietary", "?")
|
|
sug = nutrients.get("Sugars, total including NLEA", "?")
|
|
|
|
print(f" {food.get('description', 'N/A')}")
|
|
print(f" Calories : {cal} kcal")
|
|
print(f" Protein : {prot}g")
|
|
print(f" Fat : {fat}g")
|
|
print(f" Carbs : {carb}g (fiber: {fib}g, sugar: {sug}g)")
|
|
print(f" FDC ID : {food.get('fdcId', 'N/A')}")
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) < 2:
|
|
print(__doc__)
|
|
sys.exit(1)
|
|
|
|
if sys.argv[1] == "-":
|
|
queries = [line.strip() for line in sys.stdin if line.strip()]
|
|
else:
|
|
queries = sys.argv[1:]
|
|
|
|
for query in queries:
|
|
print(f"\n--- {query.upper()} (per 100g) ---")
|
|
data = search(query, max_results=2)
|
|
if not data or not data.get("foods"):
|
|
print(" No results found.")
|
|
else:
|
|
for food in data["foods"]:
|
|
display(food)
|
|
print()
|
|
if len(queries) > 1:
|
|
time.sleep(1) # respect rate limits
|
|
|
|
if API_KEY == "DEMO_KEY":
|
|
print("\nTip: using DEMO_KEY (30 req/hr). Set USDA_API_KEY for 1000 req/hr.")
|
|
print("Free signup: https://fdc.nal.usda.gov/api-key-signup/")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |