mypy

Velleman · Dec 9, 2024 · e2c95ea · e2c95ea
1 parent b9748bb
commit e2c95ea
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/linkplay/endpoint.py b/src/linkplay/endpoint.py
@@ -37,10 +37,11 @@ def __init__(
             "http",
             "https",
         ], "Protocol must be either 'http' or 'https'"
-        if (protocol == "http" and port != 80) or (protocol == "https" and port != 443):
-            self._endpoint: str = f"{protocol}://{endpoint}:{port}"
-        else:
-            self._endpoint: str = f"{protocol}://{endpoint}"
+        include_port = (protocol == "http" and port != 80) or (
+            protocol == "https" and port != 443
+        )
+        port_suffix = f":{port}" if include_port else ""
+        self._endpoint: str = f"{protocol}://{endpoint}{port_suffix}"
 
         self._session: ClientSession = session