reformat using black

khodedawsh · Nov 3, 2024 · efd7b2f · efd7b2f
1 parent c1b0f1c
commit efd7b2f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/v2share/__init__.py b/v2share/__init__.py
@@ -5,4 +5,11 @@
 from .wireguard import WireGuardConfig
 from .xray import XrayConfig
 
-__all__ = ["V2Data", "SingBoxConfig", "ClashConfig", "ClashMetaConfig", "XrayConfig"]
+__all__ = [
+    "V2Data",
+    "SingBoxConfig",
+    "ClashConfig",
+    "ClashMetaConfig",
+    "XrayConfig",
+    "WireGuardConfig",
+]
diff --git a/v2share/data.py b/v2share/data.py
@@ -173,6 +173,8 @@ def to_link(self):
 
         if self.protocol == "wireguard":
             payload = {"publickey": self.path, "address": self.client_address}
+            if isinstance(self.mtu, int):
+                payload.update({"mtu": self.mtu})
             return (
                 "wireguard://"
                 + f"{self.ed25519}@{self.address}:{self.port}?"

diff --git a/v2share/wireguard.py b/v2share/wireguard.py
@@ -21,7 +21,7 @@ def render(self, sort: bool, shuffle: bool) -> str:
             config += "PrivateKey = " + proxy.ed25519 + "\n"
             if proxy.mtu is not None:
                 config += "MTU = " + str(proxy.mtu) + "\n"
-            if proxy.dns_servers is not None:
+            if proxy.dns_servers:
                 config += "DNS = " + ",".join(proxy.dns_servers) + "\n"
             config += "\n"
             config += "[Peer]\nPublicKey = " + proxy.path + "\n"