Skip to content
Snippets Groups Projects
Verified Commit 2ea0f7b4 authored by Karel van Klink's avatar Karel van Klink :smiley_cat:
Browse files

Remove site tier from KentikClient

parent 32ec1310
No related branches found
No related tags found
No related merge requests found
......@@ -7,7 +7,6 @@ import requests
from pydantic import BaseModel
from requests import Response
from gso.products.product_blocks.site import SiteTier
from gso.settings import load_oss_params
logger = logging.getLogger(__name__)
......@@ -20,7 +19,6 @@ class NewKentikDevice(BaseModel):
device_description: str
sending_ips: list[str]
site_id: int
site_tier: SiteTier
device_snmp_ip: str
device_bgp_flowspec: bool
device_bgp_neighbor_ip: str
......@@ -117,7 +115,7 @@ class KentikClient:
plan_id = self.get_plan_by_name(self.config.placeholder_license_key)["id"]
request_body = {
"device": {
**device.model_dump(exclude=set("device_name" "site_tier")),
**device.model_dump(exclude=set("device_name")),
"device_name": device.device_description,
"device_type": self.config.device_type,
"device_subtype": self.config.device_type,
......
......@@ -130,12 +130,10 @@ def create_kentik_device(subscription: Router) -> State:
msg = "Site could not be found in Kentik."
raise ProcessFailureError(msg, details=subscription.router.router_site.site_name)
site_tier = subscription.router.router_site.site_tier
new_device = NewKentikDevice(
device_name=subscription.router.router_fqdn,
device_description=str(subscription.subscription_id),
sending_ips=[str(subscription.router.router_lo_ipv4_address)],
site_tier=site_tier,
site_id=kentik_site["id"],
device_snmp_ip=str(subscription.router.router_lo_ipv4_address),
device_bgp_flowspec=False,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment