From 795e22c043ffd70df6ac8b580544fe30c2662860 Mon Sep 17 00:00:00 2001 From: Fedor Lyanguzov Date: Thu, 9 Jan 2025 17:38:11 +0300 Subject: [PATCH] Add little type annotations --- vpn_manager/peers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vpn_manager/peers.py b/vpn_manager/peers.py index 929d559..cfd2d9c 100644 --- a/vpn_manager/peers.py +++ b/vpn_manager/peers.py @@ -10,7 +10,7 @@ class Peer: endpoint: str = None -def format_static_peer(static_peer, routes, keepalive=30): +def format_static_peer(static_peer: Peer, routes, keepalive=30): return ( "[Peer]\n" f"PublicKey = {static_peer.public_key}\n" @@ -20,7 +20,7 @@ def format_static_peer(static_peer, routes, keepalive=30): ).strip() -def format_interface(peer, dns, forward=False): +def format_interface(peer: Peer, dns, forward=False): if forward: forward = ( "PostUp = iptables -A FORWARD -i wg0 -j ACCEPT; iptables -t nat -A POSTROUTING -o eth0 -j MASQUERADE\n"