Skip to content

allow for better performance when not using interface auto descriptions #103

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
loulecrivain merged 8 commits into from
Mar 17, 2026
Merged

allow for better performance when not using interface auto descriptions #103

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
63fdead
separate graphql queries from source code
loulecrivain Feb 23, 2026
59159f0
add @without_feature decorator for tests
loulecrivain Feb 24, 2026
f8e956c
differenciated queries for autodesc
loulecrivain Feb 25, 2026
efd21c5
add tests for disabled autodesc
loulecrivain Feb 25, 2026
213946d
add sharedmock as a dependency
loulecrivain Mar 2, 2026
7c54de3
fix broken multiprocess tests for autodesc feature checks
loulecrivain Mar 2, 2026
dc403fd
Nix: package sharedmock dependency
yuyuyureka Mar 4, 2026
1f05716
clip max number of inflight requests
loulecrivain Mar 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
369 changes: 28 additions & 341 deletions cosmo/clients/netbox_v4.py
View file
Open in desktop

Large diffs are not rendered by default.

33 changes: 33 additions & 0 deletions cosmo/clients/queries/connected_devices.graphql
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
query {
interface_list(filters: { $tag_filter }) {
__typename
id,
parent {
__typename
id,
connected_endpoints {
... on InterfaceType {
__typename
name
device {
name
__typename
primary_ip4 {
__typename
address
}
interfaces {
id
name
__typename
ip_addresses {
__typename
address
}
}
}
}
}
}
}
}
99 changes: 99 additions & 0 deletions cosmo/clients/queries/device.graphql
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,99 @@
query {
device_list(filters: {
name: { i_exact: $device },
}) {
__typename
id
name
custom_fields

device_type {
__typename
slug
}
platform {
__typename
manufacturer {
__typename
slug
}
slug
}
primary_ip4 {
__typename
address
}

interfaces {
__typename
id
name
enabled
type
mode
mtu
description
connected_endpoints {
... on InterfaceType {
__typename
name
device {
__typename
name
}
}
$autodesc_query_extension
}
vrf {
__typename
id
name
description
rd
export_targets {
__typename
name
}
import_targets {
__typename
name
}
}
lag {
__typename
id
name
}
ip_addresses {
__typename
address
role
}
untagged_vlan {
__typename
id
name
vid
}
tagged_vlans {
__typename
id
name
vid
}
tags {
__typename
id
name
slug
}
parent {
__typename
id
mtu
name
}
custom_fields
}
}
}
90 changes: 90 additions & 0 deletions cosmo/clients/queries/device_autodesc_query.graphql
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,90 @@
... on ProviderNetworkType {
__typename
display
}
... on CircuitTerminationType {
__typename
display
}
... on VirtualCircuitTerminationType {
__typename
display
}
... on FrontPortType {
__typename
name
device {
__typename
name
}
}
... on RearPortType {
__typename
name
device {
__typename
name
}
}
... on ConsolePortType {
__typename
name
device {
__typename
name
}
}
... on ConsoleServerPortType {
__typename
name
device {
__typename
name
}
}
}
link_peers {
... on CircuitTerminationType {
__typename
display
}
... on FrontPortType {
__typename
name
device {
__typename
name
}
}
... on RearPortType {
__typename
name
device {
__typename
name
}
}
... on ConsolePortType {
__typename
name
device {
__typename
name
}
}
... on ConsoleServerPortType {
__typename
name
device {
__typename
name
}
}
... on InterfaceType {
__typename
name
device {
__typename
name
}
}
Empty file added cosmo/clients/queries/device_no_autodesc_query.graphql
View file
Open in desktop
Empty file.
64 changes: 64 additions & 0 deletions cosmo/clients/queries/l2vpn.graphql
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,64 @@
query {
l2vpn_list (filters: {name: {starts_with: "WAN: "}}) {
__typename
id
name
type
identifier
terminations {
__typename
id
assigned_object {
__typename
... on VLANType {
__typename
id
name
interfaces_as_tagged {
id
name
__typename
device {
__typename
id
name
}
}
interfaces_as_untagged {
id
name
__typename
device {
__typename
id
name
}
}
}
... on InterfaceType {
__typename
id
name
custom_fields
untagged_vlan {
__typename
id
name
vid
}
tagged_vlans {
__typename
id
name
vid
}
device {
__typename
id
name
}
}
}
}
}
}
39 changes: 39 additions & 0 deletions cosmo/clients/queries/loopback.graphql
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
query{
interface_list(filters: {
name: {starts_with: "lo"}
}) {
__typename
name,
child_interfaces {
__typename
name,
vrf {
__typename
id
name
description
rd
export_targets {
__typename
name
}
import_targets {
__typename
name
}
},
ip_addresses {
__typename
address,
family {
__typename
value,
}
}
}
device{
__typename
name,
}
}
}
14 changes: 14 additions & 0 deletions cosmo/common.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,6 @@
from os import PathLike
from pathlib import Path
from string import Template
from abc import ABC, abstractmethod
from typing import Optional, Union, Protocol, TypeVar, Sequence

Expand Down Expand Up @@ -58,6 +61,10 @@ def head(l):
return None if not l else l[0]


def clip(x: int, lower: int, upper: int):
return min(max(x, lower), upper)


def deepsort(e):
if isinstance(e, list):
return sorted(deepsort(v) for v in e)
Expand All @@ -70,3 +77,10 @@ def without_keys(d, keys) -> dict:
if type(keys) != list:
keys = [keys]
return {k: v for k, v in d.items() if k not in keys}


class FileTemplate(Template):
def __init__(self, template_file_path: str | bytes | PathLike):
with open(template_file_path, "r") as template_file:
template = template_file.read()
super().__init__(template)
Loading
Loading