1 files changed, 109 insertions, 0 deletions
diff --git a/backend/fileutil.py b/backend/fileutil.py
new file mode 100644
index 0000000..2aa2d6f
--- /dev/null
+++ b/backend/fileutil.py
@@ -0,0 +1,109 @@
+import os.path
+import urllib.request
+import json
+import time
+import configparser
+
+
+def _read_file(path: str, lines=True) -> list:
+    # reads a file and returns a list of lines
+    with open(path, "r", encoding="utf-8") as file:
+        if not lines:
+            return file.read()
+        return file.read().splitlines()
+
+
+def get_excluded_channels():
+    # gets excluded channels from exclude_channel.txt
+    if not os.path.exists(os.path.join("data", "exclude_channel.txt")):
+        open(os.path.join("data", "exclude_channel.txt"), "w").close()
+    excluded_channels = _read_file(os.path.join("data", "exclude_channel.txt"))
+    return excluded_channels
+
+def update_excluded_channels(channel_ids: list):
+    # add to exclude_channel.txt if not already there
+    excluded_channels = get_excluded_channels()
+    for channel_id in channel_ids:
+        if channel_id not in excluded_channels:
+            excluded_channels.append(channel_id)
+    with open(os.path.join("data", "exclude_channel.txt"), "w", encoding="utf-8") as file:
+        for channel_id in excluded_channels:
+            file.write(channel_id + "\n")
+
+def save_local_channels(data: list, path: str = "data"):
+    """
+    Save the channel names and ids locally for when the API is down
+    """
+    path = os.path.join(path, "channels.txt")
+    excluded_channels = get_excluded_channels()
+    if not os.path.exists(path):
+        os.makedirs(os.path.dirname(path), exist_ok=True)
+        open(path, "w").close()
+    with open(path, "w", encoding="utf-8") as file:
+        for channel in data:
+            if channel["id"] in excluded_channels:
+                continue
+            file.write(f"{channel['id']},{channel['english_name']}\n")
+
+
+def get_local_channels(path: str = "data"):
+    """
+    Get the channel names and ids locally for when the API is down
+    """
+    path = os.path.join(path, "channels.txt")
+    if not os.path.exists(path):
+        raise Exception("Local channel data not found")
+    with open(path, "r", encoding="utf-8") as file:
+        rows = file.read().splitlines()
+        return [tuple(row.split(",")) for row in rows]
+
+
+def check_diff_refresh():
+    if not os.path.exists(os.path.join("data", "last_refresh.txt")):
+        with open(
+            os.path.join("data", "last_refresh.txt"), "w", encoding="utf-8") as file:
+            file.write(time.strftime("%Y-%m-%d"))
+            return True
+    with open(os.path.join("data", "last_refresh.txt"), "r", encoding="utf-8") as file:
+        last_refresh = file.read()
+    if last_refresh != time.strftime("%Y-%m-%d"):
+        with open(
+            os.path.join("data", "last_refresh.txt"), "w", encoding="utf-8"
+        ) as file:
+            file.write(time.strftime("%Y-%m-%d"))
+        return True
+
+
+def update_data_files(url: str) -> None:
+    # Updates the local txt channel data stored in data folder
+    if not os.path.exists(os.path.join("data", "channels.txt")):
+        open(os.path.join("data", "channels.txt"), "w").close()
+    urllib.request.urlretrieve(
+        url + "channels.txt", os.path.join("data", "channels.txt")
+    )
+    # downloaded txt file from url and write to channels.txt
+
+    if not os.path.exists(os.path.join("data", "exclude_channel.txt")):
+        open(os.path.join("data", "exclude_channel.txt"), "w").close()
+    urllib.request.urlretrieve(
+        url + "exclude_channel.txt", os.path.join("data", "exclude_channel.txt")
+    )
+
+
+def load_config(ini_filepath: str) -> dict:
+    config_object = configparser.ConfigParser()
+    file = open(ini_filepath, "r")
+    config_object.read_file(file)
+    output_dict = {}
+    sections = config_object.sections()
+    for section in sections:
+        output_dict[section] = {}
+        for key in config_object[section]:
+            output_dict[section][key] = config_object[section][key]
+    return output_dict
+
+def load_json_file(json_file_path: str) -> dict:
+    with open(json_file_path, "r", encoding="utf-8") as file:
+        return json.load(file)
+
+