From 163e07adb573da8497c9ccf0ae97a6c261e7b14a Mon Sep 17 00:00:00 2001
From: Pinapelz <yukais@pinapelz.com>
Date: Sat, 13 Apr 2024 01:08:10 -0700
Subject: initial commit. Migration from GitLab

---
 fileutil.py | 93 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 93 insertions(+)
 create mode 100644 fileutil.py

(limited to 'fileutil.py')

diff --git a/fileutil.py b/fileutil.py
new file mode 100644
index 0000000..325ee0f
--- /dev/null
+++ b/fileutil.py
@@ -0,0 +1,93 @@
+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 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)
+
+
-- 
cgit v1.2.3