resolve ruff issues

Author: GreatBahram

src/shecan/api.py

@@ -5,7 +5,7 @@
 """
 
 import sys
-from typing import List, NamedTuple
+from typing import NamedTuple
 
 from shecan.db import ShecanConfig
 from shecan.utils import get_shecan_ips
@@ -36,7 +36,7 @@ def list_dns():
     return ips
 
 
-def current_dns() -> List[Resolver]:
+def current_dns() -> list[Resolver]:
     """List current dns servers in /etc/resolv.conf."""
     resolv_list = []
     if sys.platform == "linux":

src/shecan/cli.py

@@ -5,6 +5,7 @@
 import sys
 from pathlib import Path
 from tempfile import gettempdir
+from typing import Optional
 
 from tabulate import tabulate
 
@@ -20,7 +21,7 @@ class TextStyle:
     NC = "\033[0m"  # No Color
 
 
-def colorify(text: str, style: TextStyle):
+def colorify(text: str, style: TextStyle) -> str:
     return f"{style}{text}{TextStyle.NC}"
 
 
@@ -33,7 +34,7 @@ def list_dns() -> None:
     print(tabulate(dns_servers, headers=["ID", "IP"], stralign="center"))
 
 
-def update_resolv_file(content):
+def update_resolv_file(content) -> None:
     resolv_file = Path("/etc", "resolv.conf")
     tmp_resolv_file = Path(gettempdir()).joinpath("resolv.conf")
 
@@ -59,7 +60,7 @@ def update_resolv_file(content):
         logger.error(f"Could not update resolv file: {e}")
 
 
-def restore_resolv_file():
+def restore_resolv_file() -> None:
     tmp_resolv_file = Path(gettempdir(), "resolv.conf")
     resolv_file = Path("/etc", "resolv.conf")
     if tmp_resolv_file.exists():
@@ -94,7 +95,7 @@ def show_current_dns() -> None:
     print(tabulate(resolvers, headers=["Type", "IP"], stralign="center"))
 
 
-def shecan_cli():
+def shecan_cli() -> Optional[int]:
     parser = argparse.ArgumentParser(description="CLI for shecan.ir website.")
     parser.add_argument(
         "--debug",

src/shecan/db.py

@@ -14,13 +14,13 @@ def __init__(self, config_file: str = DB_PATH) -> None:
         self._parser = ConfigParser()
         self._name = "shecan"
 
-    def _initialize(self):
+    def _initialize(self) -> None:
         """Initialize the shecan config file"""
         if not Path(self.config_file).exists():
             with open(self.config_file, mode="w") as fp:
                 fp.write(f"[{self._name}]")
 
-    def _read_config(self):
+    def _read_config(self) -> None:
         """Read the configuration from a file."""
         self._initialize()
         self._parser.read(self.config_file)
@@ -33,7 +33,7 @@ def __exit__(self, *exceptions):
         with open(self.config_file, mode="w") as fp:
             self._parser.write(fp)
 
-    def update(self, ips: list):
+    def update(self, ips: list) -> None:
         """Store all Shecan's ip addresses."""
         self._parser[self._name]["dns-ips"] = ",".join(ips)
 

src/shecan/utils.py

@@ -4,7 +4,6 @@
 This module provides utility functions that are used within shecan.
 """
 
-from typing import Tuple
 from urllib import request
 
 from bs4 import BeautifulSoup
@@ -13,7 +12,7 @@
 CSS_SELECTOR = ".shecan-dns-ips"
 
 
-def get_shecan_ips(url: str = BASE_URL, selector: str = CSS_SELECTOR) -> Tuple[str]:
+def get_shecan_ips(url: str = BASE_URL, selector: str = CSS_SELECTOR) -> tuple[str]:
     """Retrieve shecan DNS IP addresses."""
     data = request.urlopen(BASE_URL).read().decode("utf-8")
     soup = BeautifulSoup(data, "lxml")