Dev 349 v2

src/sc_keeper/api.py

@@ -7,7 +7,7 @@
 from fastapi import Depends, FastAPI, HTTPException, Response
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.middleware.gzip import GZipMiddleware
-from sc_crawler.table_fields import Status
+from sc_crawler.table_fields import Status, TrafficDirection
 from sc_crawler.tables import (
     Benchmark,
     ComplianceFramework,
@@ -683,7 +683,7 @@ def search_traffic_prices(
     green_energy: options.green_energy = None,
     regions: options.regions = None,
     countries: options.countries = None,
-    direction: options.direction = None,
+    direction: options.direction = [TrafficDirection.OUT],
     monthly_traffic: options.monthly_traffic = 1,
     limit: options.limit = 50,
     page: options.page = None,
@@ -753,11 +753,11 @@ def search_traffic_prices(
     # update prices per tiers and to currency requested
     for price in prices:
 
+        def rounder(p):
+            return round(p, 6)
+
         def local_price(p):
-            return round(
-                currency_converter.convert(p, price.currency, currency),
-                6,
-            )
+            return rounder(currency_converter.convert(p, price.currency, currency))
 
         if currency:
             if hasattr(price, "price") and hasattr(price, "currency"):
@@ -774,9 +774,9 @@ def local_price(p):
                     max(monthly_traffic - traffic_paid, 0),
                     (float(tier.upper) - float(tier.lower)),
                 )
-                price.price_monthly_traffic += tier.price * traffic_tier
+                price.price_monthly_traffic += rounder(tier.price * traffic_tier)
                 traffic_paid += traffic_tier
         else:
-            price.price_monthly_traffic = price.price * monthly_traffic
+            price.price_monthly_traffic = rounder(price.price * monthly_traffic)
 
     return prices

src/sc_keeper/routers/ai.py

@@ -38,3 +38,9 @@ def assist_server_price_filters(text: str, request: Request) -> dict:
 def assist_storage_price_filters(text: str, request: Request) -> dict:
     """Extract StoragePrice JSON filters from freetext."""
     return assister(text, "/storage_prices")
+
+
+@router.get("/assist_traffic_price_filters")
+def assist_traffic_price_filters(text: str, request: Request) -> dict:
+    """Extract TrafficPrice JSON filters from freetext."""
+    return assister(text, "/traffic_prices")

tests/test_api.py

@@ -79,6 +79,7 @@ def test_healthcheck():
 ]
 
 test_traffic_prices_params = [
+    {"direction": ["inbound", "outbound"]},
     *test_general_params,
     *test_region_params,
     {"direction": ["inbound"]},
@@ -243,7 +244,7 @@ def test_traffic_prices_with_params(params):
     assert response.status_code == 200
     assert response.elapsed.total_seconds() < 5
     # if params is empty, this is the full count
-    if params == {}:
+    if params == {"direction": ["inbound", "outbound"]}:
         global count
         count = len(response.json())
     else: