Skip to content

Commit 621c62e

Browse files
committed
small fixes
1 parent f2bccc8 commit 621c62e

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

docs/features/custom_pollers.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ To define your own Poller, two steps must be performed:
1717
Here is the minimal viable example of a custom poller class. It uses `scrapli` library to connect to devices via SSH.
1818

1919
```python
20+
from scrapli import Scrapli
2021
from validity.pollers import CustomPoller
2122
from validity.models import Command
2223

validity/dependencies.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,7 @@ def validity_settings(django_settings: Annotated[LazySettings, django_settings])
2626
@di.dependency(scope=Singleton)
2727
def pollers_info(custom_pollers: Annotated[list[PollerInfo], "validity_settings.custom_pollers"]) -> list[PollerInfo]:
2828
return [
29-
PollerInfo(
30-
klass=NetmikoPoller, name="netmiko", verbose_name="netmiko", color="blue", command_types=["CLI", "custom"]
31-
),
29+
PollerInfo(klass=NetmikoPoller, name="netmiko", verbose_name="netmiko", color="blue", command_types=["CLI"]),
3230
PollerInfo(
3331
klass=RequestsPoller, name="requests", verbose_name="requests", color="info", command_types=["json_api"]
3432
),
@@ -37,7 +35,7 @@ def pollers_info(custom_pollers: Annotated[list[PollerInfo], "validity_settings.
3735
name="scrapli_netconf",
3836
verbose_name="scrapli_netconf",
3937
color="orange",
40-
command_types=["netconf", "custom"],
38+
command_types=["netconf"],
4139
),
4240
] + custom_pollers
4341

0 commit comments

Comments
 (0)