feat: enhance file write

site_list.py

@@ -1,27 +1,61 @@
 #!/usr/bin/env python
-# This module generates the listing of supported sites which can be found in
-# sites.md. It also organizes all the sites in alphanumeric order
 import json
+import os
 
-# Read the data.json file
-with open("sherlock/resources/data.json", "r", encoding="utf-8") as data_file:
-    data = json.load(data_file)
+import logging
 
-# Sort the social networks in alphanumeric order
-social_networks = sorted(data.items())
+def load_data(file_path):
+    """
+Load data from a JSON file.
 
-# Write the list of supported sites to sites.md
-with open("sites.md", "w") as site_file:
-    site_file.write(f"## List Of Supported Sites ({len(social_networks)} Sites In Total!)\n")
-    for social_network, info in social_networks:
-        url_main = info["urlMain"]
-        is_nsfw = "**(NSFW)**" if info.get("isNSFW") else ""
-        site_file.write(f"1. ![](https://www.google.com/s2/favicons?domain={url_main}) [{social_network}]({url_main}) {is_nsfw}\n")
+Parameters:
+- file_path (str): The path to the JSON file.
 
-# Overwrite the data.json file with sorted data
-with open("sherlock/resources/data.json", "w") as data_file:
-    sorted_data = json.dumps(data, indent=2, sort_keys=True)
-    data_file.write(sorted_data)
-    data_file.write("\n")
+Returns:
+- dict: The loaded data as a dictionary.
 
-print("Finished updating supported site listing!")
+Raises:
+- FileNotFoundError: If the file does not exist.
+- JSONDecodeError: If the file is not a valid JSON.
+
+Example:
+    >>> load_data("data.json")
+    {'key1': 'value1', 'key2': 'value2'}
+"""
+    try:
+        with open(file_path, "r", encoding="utf-8") as file:
+            return json.load(file)
+    except (FileNotFoundError, json.JSONDecodeError) as e:
+        logging.error(f"Error loading data: {e}")
+        return None
+def write_markdown_file(social_networks, file_path):
+    lines = [
+        f"1. ![](https://www.google.com/s2/favicons?domain={info['urlMain']}) [{network}]({info['urlMain']}) {'**(NSFW)**' if info.get('isNSFW') else ''}\n"
+        for network, info in social_networks
+    ]
+    with open(file_path, "w") as file:
+        file.write(f"## List Of Supported Sites ({len(social_networks)} Sites In Total!)\n")
+        file.writelines(lines)
+
+def save_sorted_data(data, file_path):
+    with open(file_path, "w") as file:
+        sorted_data = json.dumps(data, indent=2, sort_keys=True)
+        file.write(sorted_data)
+        file.write("\n")
+
+def main():
+    data_file_path = "sherlock/resources/data.json"
+    md_file_path = "sites.md"
+
+    data = load_data(data_file_path)
+    if data is None:
+        return
+
+    sorted_social_networks = sorted(data.items())
+    write_markdown_file(sorted_social_networks, md_file_path)
+    save_sorted_data(data, data_file_path)
+
+    print("Finished updating supported site listing!")
+
+if __name__ == "__main__":
+    main()