cs2tracker 2.1.9__py3-none-any.whl → 2.1.10__py3-none-any.whl

cs2tracker/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '2.1.9'
-__version_tuple__ = version_tuple = (2, 1, 9)
+__version__ = version = '2.1.10'
+__version_tuple__ = version_tuple = (2, 1, 10)

cs2tracker/application.py

@@ -3,13 +3,16 @@ import tkinter as tk
 from shutil import copy
 from subprocess import Popen
 from threading import Thread
-from tkinter import messagebox
+from tkinter import messagebox, ttk
+from tkinter.filedialog import askopenfilename, asksaveasfile
 from typing import cast
 
 import matplotlib.pyplot as plt
+import sv_ttk
 from matplotlib.axes import Axes
 from matplotlib.dates import DateFormatter
 
+from cs2tracker.background_task import BackgroundTask
 from cs2tracker.constants import (
     CONFIG_FILE,
     CONFIG_FILE_BACKUP,
@@ -19,20 +22,19 @@ from cs2tracker.constants import (
     POWERSHELL_COLORIZE_OUTPUT,
     PYTHON_EXECUTABLE,
     RUNNING_IN_EXE,
-    TEXT_EDITOR,
     OSType,
 )
+from cs2tracker.price_logs import PriceLogs
 from cs2tracker.scraper import Scraper
 
 APPLICATION_NAME = "CS2Tracker"
+WINDOW_SIZE = "630x380"
 
-WINDOW_SIZE = "550x500"
-BACKGROUND_COLOR = "#1e1e1e"
-BUTTON_COLOR = "#3c3f41"
-BUTTON_HOVER_COLOR = "#505354"
-BUTTON_ACTIVE_COLOR = "#5c5f61"
-FONT_STYLE = "Segoe UI"
-FONT_COLOR = "white"
+CONFIG_EDITOR_TITLE = "Config Editor"
+CONFIG_EDITOR_SIZE = "800x750"
+
+NEW_CUSTOM_ITEM_TITLE = "Add Custom Item"
+NEW_CUSTOM_ITEM_SIZE = "500x200"
 
 SCRAPER_WINDOW_HEIGHT = 40
 SCRAPER_WINDOW_WIDTH = 120
@@ -42,85 +44,64 @@ SCRAPER_WINDOW_BACKGROUND_COLOR = "Black"
 class Application:
     def __init__(self):
         self.scraper = Scraper()
+        self.application_window = None
+        self.config_editor_window = None
 
     def run(self):
         """Run the main application window with buttons for scraping prices, editing the
         configuration, showing history in a chart, and editing the log file.
         """
-        application_window = self._configure_window()
-        application_window.mainloop()
-
-    def _add_button(self, frame, text, command):
-        """Create and style a button for the main application window."""
-        button_style = {
-            "font": (FONT_STYLE, 12),
-            "fg": FONT_COLOR,
-            "bg": BUTTON_COLOR,
-            "activebackground": BUTTON_ACTIVE_COLOR,
-        }
-        button = tk.Button(frame, text=text, command=command, **button_style)
-        button.pack(pady=5, fill="x")
-        button.bind("<Enter>", lambda _: button.config(bg=BUTTON_HOVER_COLOR))
-        button.bind("<Leave>", lambda _: button.config(bg=BUTTON_COLOR))
-        return button
-
-    def _add_checkbox(self, frame, text, variable, command):
-        checkbox = tk.Checkbutton(
+        self.application_window = self._configure_window()
+
+        sv_ttk.use_dark_theme()
+
+        self.application_window.mainloop()
+
+    def _add_button(self, frame, text, command, row):
+        """Create and style a button for the button frame."""
+        grid_pos = {"row": row, "column": 0, "sticky": "ew", "padx": 10, "pady": 10}
+        button = ttk.Button(frame, text=text, command=command)
+        button.grid(**grid_pos)
+
+    def _configure_button_frame(self, main_frame):
+        """Configure the button frame of the application main frame."""
+        button_frame = ttk.Frame(main_frame, style="Card.TFrame", padding=15)
+        button_frame.columnconfigure(0, weight=1)
+        button_frame.grid(row=0, column=0, padx=10, pady=(7, 20), sticky="nsew")
+
+        self._add_button(button_frame, "Run!", self.scrape_prices, 0)
+        self._add_button(button_frame, "Edit Config", self._edit_config, 1)
+        self._add_button(button_frame, "Reset Config", self._reset_config, 2)
+        self._add_button(button_frame, "Show History", self._draw_plot, 3)
+        self._add_button(button_frame, "Export History", self._export_log_file, 4)
+        self._add_button(button_frame, "Import History", self._import_log_file, 5)
+
+    def _add_checkbox(
+        self, frame, text, variable, command, row
+    ):  # pylint: disable=too-many-arguments,too-many-positional-arguments
+        """Create and style a checkbox for the checkbox frame."""
+        grid_pos = {"row": row, "column": 0, "sticky": "w", "padx": (10, 0), "pady": 5}
+        checkbox = ttk.Checkbutton(
             frame,
             text=text,
             variable=variable,
             command=command,
-            bg=BACKGROUND_COLOR,
-            fg=FONT_COLOR,
-            selectcolor=BUTTON_COLOR,
-            activebackground=BACKGROUND_COLOR,
-            font=(FONT_STYLE, 10),
-            anchor="w",
-            padx=20,
-        )
-        checkbox.pack(fill="x", anchor="w", pady=2)
-
-    def _configure_window(self):
-        """Configure the main application window UI and add buttons for the main
-        functionalities.
-        """
-        window = tk.Tk()
-        window.title(APPLICATION_NAME)
-        window.geometry(WINDOW_SIZE)
-        window.configure(bg=BACKGROUND_COLOR)
-        if OS == OSType.WINDOWS:
-            app_id = "cs2tracker.unique.id"
-            ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID(app_id)
-        icon = tk.PhotoImage(file=ICON_FILE)
-        window.wm_iconphoto(False, icon)
-
-        frame = tk.Frame(window, bg=BACKGROUND_COLOR, padx=30, pady=30)
-        frame.pack(expand=True, fill="both")
-
-        label = tk.Label(
-            frame,
-            text=f"Welcome to {APPLICATION_NAME}!",
-            font=(FONT_STYLE, 16, "bold"),
-            fg=FONT_COLOR,
-            bg=BACKGROUND_COLOR,
+            style="Switch.TCheckbutton",
         )
-        label.pack(pady=(0, 30))
-
-        self._add_button(frame, "Run!", self.scrape_prices)
-        self._add_button(frame, "Edit Config", self._edit_config)
-        self._add_button(frame, "Reset Config", self._confirm_reset_config)
-        self._add_button(frame, "Show History (Chart)", self._draw_plot)
-        self._add_button(frame, "Show History (File)", self._edit_log_file)
+        checkbox.grid(**grid_pos)
 
-        checkbox_frame = tk.Frame(frame, bg=BACKGROUND_COLOR)
-        checkbox_frame.pack(pady=(20, 0), fill="x")
+    def _configure_checkbox_frame(self, main_frame):
+        """Configure the checkbox frame for background tasks and settings."""
+        checkbox_frame = ttk.LabelFrame(main_frame, text="Settings", padding=15)
+        checkbox_frame.grid(row=0, column=1, padx=10, pady=(0, 20), sticky="nsew")
 
-        background_checkbox_value = tk.BooleanVar(value=self.scraper.identify_background_task())
+        background_checkbox_value = tk.BooleanVar(value=BackgroundTask.identify())
         self._add_checkbox(
             checkbox_frame,
-            "Daily Background Calculations",
+            "Background Task",
             background_checkbox_value,
             lambda: self._toggle_background_task(background_checkbox_value.get()),
+            0,
         )
 
         discord_webhook_checkbox_value = tk.BooleanVar(
@@ -130,9 +111,12 @@ class Application:
         )
         self._add_checkbox(
             checkbox_frame,
-            "Receive Discord Notifications",
+            "Discord Notifications",
             discord_webhook_checkbox_value,
-            lambda: self._toggle_discord_webhook(discord_webhook_checkbox_value.get()),
+            lambda: discord_webhook_checkbox_value.set(
+                self._toggle_discord_webhook(discord_webhook_checkbox_value.get())
+            ),
+            1,
         )
 
         use_proxy_checkbox_value = tk.BooleanVar(
@@ -142,9 +126,48 @@ class Application:
             checkbox_frame,
             "Proxy Requests",
             use_proxy_checkbox_value,
-            lambda: self._toggle_use_proxy(use_proxy_checkbox_value.get()),
+            lambda: use_proxy_checkbox_value.set(
+                self._toggle_use_proxy(use_proxy_checkbox_value.get())
+            ),
+            2,
         )
 
+        dark_theme_checkbox_value = tk.BooleanVar(value=True)
+        self._add_checkbox(
+            checkbox_frame, "Dark Theme", dark_theme_checkbox_value, sv_ttk.toggle_theme, 3
+        )
+
+    def _configure_main_frame(self, window):
+        """Configure the main frame of the application window with buttons and
+        checkboxes.
+        """
+        main_frame = ttk.Frame(window, padding=15)
+        main_frame.columnconfigure(0, weight=1)
+        main_frame.columnconfigure(1, weight=1)
+        main_frame.rowconfigure(0, weight=1)
+
+        self._configure_button_frame(main_frame)
+        self._configure_checkbox_frame(main_frame)
+
+        main_frame.pack(expand=True, fill="both")
+
+    def _configure_window(self):
+        """Configure the main application window UI and add buttons for the main
+        functionalities.
+        """
+        window = tk.Tk()
+        window.title(APPLICATION_NAME)
+        window.geometry(WINDOW_SIZE)
+
+        if OS == OSType.WINDOWS:
+            app_id = "cs2tracker.unique.id"
+            ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID(app_id)
+
+        icon = tk.PhotoImage(file=ICON_FILE)
+        window.wm_iconphoto(True, icon)
+
+        self._configure_main_frame(window)
+
         return window
 
     def _construct_scraper_command_windows(self):
@@ -200,57 +223,280 @@ class Application:
             # TODO: implement external window for Linux
             self.scraper.scrape_prices()
 
-    def _edit_config(self):
-        """Edit the configuration file using the specified text editor."""
-        _popen_and_call(
-            popen_args={"args": [TEXT_EDITOR, CONFIG_FILE], "shell": True},
-            callback=self.scraper.parse_config,
+    def _make_tree_editable(self, editor_frame, tree):
+        """
+        Add a binding to the treeview that allows double-clicking on a cell to edit its
+        value.
+
+        Source: https://stackoverflow.com/questions/75787251/create-an-editable-tkinter-treeview-with-keyword-connection
+        """
+
+        def set_cell_value(event):
+            def save_edit(event):
+                tree.set(row, column=column, value=event.widget.get())
+                event.widget.destroy()
+
+            try:
+                row = tree.identify_row(event.y)
+                column = tree.identify_column(event.x)
+                item_text = tree.set(row, column)
+                if item_text.strip() == "":
+                    left_item_text = tree.item(row, "text")
+                    # Don't allow editing of section headers
+                    if any(left_item_text == section for section in self.scraper.config.sections()):
+                        return
+                x, y, w, h = tree.bbox(row, column)
+                entryedit = ttk.Entry(editor_frame)
+                entryedit.place(x=x, y=y, width=w, height=h + 3)  # type: ignore
+                entryedit.insert("end", item_text)
+                entryedit.bind("<Return>", save_edit)
+                entryedit.focus_set()
+                entryedit.grab_set()
+            except Exception:
+                pass
+
+        def destroy_entries(_):
+            """Destroy any entry widgets in the treeview when the mouse wheel is
+            used.
+            """
+            for widget in editor_frame.winfo_children():
+                if isinstance(widget, ttk.Entry):
+                    widget.destroy()
+
+        def destroy_entry(event):
+            """Destroy the entry widget if the user clicks outside of it."""
+            if isinstance(event.widget, ttk.Entry):
+                event.widget.destroy()
+
+        tree.bind("<Double-1>", set_cell_value)
+        self.config_editor_window.bind("<MouseWheel>", destroy_entries)  # type: ignore
+        self.config_editor_window.bind("<Button-1>", destroy_entry)  # type: ignore
+
+    def _configure_treeview(self, editor_frame):
+        """Add a treeview to the editor frame to display and edit configuration
+        options.
+        """
+        scrollbar = ttk.Scrollbar(editor_frame)
+        scrollbar.pack(side="right", fill="y", padx=(5, 0))
+
+        tree = ttk.Treeview(
+            editor_frame,
+            columns=(1,),
+            height=10,
+            selectmode="browse",
+            yscrollcommand=scrollbar.set,
         )
+        scrollbar.config(command=tree.yview)
+
+        tree.column("#0", anchor="w", width=200)
+        tree.column(1, anchor="w", width=25)
+        tree.heading("#0", text="Option")
+        tree.heading(1, text="Value")
+
+        for section in self.scraper.config.sections():
+            if section == "App Settings":
+                continue
+            section_level = tree.insert("", "end", iid=section, text=section)
+            for config_option, value in self.scraper.config.items(section):
+                title_option = config_option.replace("_", " ").title()
+                tree.insert(section_level, "end", text=title_option, values=(value,))
+
+        self._make_tree_editable(editor_frame, tree)
+
+        return tree
+
+    def _configure_save_button(self, button_frame, tree):
+        """Save updated options and values from the treeview back to the config file."""
+
+        def save_config():
+            for child in tree.get_children():
+                for item in tree.get_children(child):
+                    title_option = tree.item(item, "text")
+                    config_option = title_option.lower().replace(" ", "_")
+                    value = tree.item(item, "values")[0]
+                    section = tree.parent(item)
+                    section_name = tree.item(section, "text")
+                    if section_name == "Custom Items":
+                        # custom items are already saved upon creation (Saving them again would result in duplicates)
+                        continue
+                    self.scraper.config.set(section_name, config_option, value)
+
+            self.scraper.config.write_to_file()
+            if self.scraper.config.valid:
+                messagebox.showinfo(
+                    "Config Saved", "The configuration has been saved successfully."
+                )
+            else:
+                messagebox.showerror(
+                    "Config Error",
+                    f"The configuration is invalid. ({self.scraper.config.last_error})",
+                )
+
+        save_button = ttk.Button(button_frame, text="Save", command=save_config)
+        save_button.pack(side="left", expand=True, padx=5)
+
+    def _configure_custom_item_button(self, button_frame, tree):
+        """Add a button that opens an entry dialog to add a custom item to the
+        configuration.
+        """
 
-    def _confirm_reset_config(self):
-        confirm = messagebox.askokcancel(
-            "Reset Config", "Are you sure you want to reset the config file?"
+        def add_custom_item(item_url, item_owned):
+            """Add a custom item to the configuration."""
+            if not item_url or not item_owned:
+                messagebox.showerror("Input Error", "All fields must be filled out.")
+                return
+
+            try:
+                if int(item_owned) < 0:
+                    raise ValueError("Owned count must be a non-negative integer.")
+            except ValueError as error:
+                messagebox.showerror("Input Error", f"Invalid owned count: {error}")
+                return
+
+            self.scraper.config.set("Custom Items", item_url, item_owned)
+            self.scraper.config.write_to_file()
+            if self.scraper.config.valid:
+                tree.insert("Custom Items", "end", text=item_url, values=(item_owned,))
+                messagebox.showinfo("Custom Item Added", "Custom item has been added successfully.")
+            else:
+                self.scraper.config.remove_option("Custom Items", item_url)
+                messagebox.showerror(
+                    "Config Error",
+                    f"The configuration is invalid. ({self.scraper.config.last_error})",
+                )
+
+        def open_custom_item_dialog():
+            """Open a dialog to enter custom item details."""
+            dialog = tk.Toplevel(self.config_editor_window)
+            dialog.title(NEW_CUSTOM_ITEM_TITLE)
+            dialog.geometry(NEW_CUSTOM_ITEM_SIZE)
+
+            dialog_frame = ttk.Frame(dialog, padding=10)
+            dialog_frame.pack(expand=True, fill="both")
+
+            ttk.Label(dialog_frame, text="Item URL:").pack(pady=5)
+            item_url_entry = ttk.Entry(dialog_frame)
+            item_url_entry.pack(fill="x", padx=10)
+
+            ttk.Label(dialog_frame, text="Owned Count:").pack(pady=5)
+            item_owned_entry = ttk.Entry(dialog_frame)
+            item_owned_entry.pack(fill="x", padx=10)
+
+            add_button = ttk.Button(
+                dialog_frame,
+                text="Add",
+                command=lambda: add_custom_item(item_url_entry.get(), item_owned_entry.get()),
+            )
+            add_button.pack(pady=10)
+
+        custom_item_button = ttk.Button(
+            button_frame, text="Add Custom Item", command=open_custom_item_dialog
         )
-        if confirm:
-            self._reset_config()
+        custom_item_button.pack(side="left", expand=True, padx=5)
+
+    def _configure_editor_frame(self):
+        """Configure the main editor frame which displays the configuration options in a
+        structured way.
+        """
+        editor_frame = ttk.Frame(self.config_editor_window, padding=30)
+        editor_frame.pack(expand=True, fill="both")
+
+        tree = self._configure_treeview(editor_frame)
+        tree.pack(expand=True, fill="both")
+
+        button_frame = ttk.Frame(editor_frame, padding=10)
+
+        self._configure_save_button(button_frame, tree)
+        self._configure_custom_item_button(button_frame, tree)
+
+        button_frame.pack(side="bottom", padx=10, pady=(0, 10))
+
+    def _edit_config(self):
+        """Open a new window with a config editor GUI."""
+        self.config_editor_window = tk.Toplevel(self.application_window)
+        self.config_editor_window.geometry(CONFIG_EDITOR_SIZE)
+        self.config_editor_window.title(CONFIG_EDITOR_TITLE)
+
+        self._configure_editor_frame()
 
     def _reset_config(self):
         """Reset the configuration file to its default state."""
-        copy(CONFIG_FILE_BACKUP, CONFIG_FILE)
-        self.scraper.parse_config()
+        confirm = messagebox.askokcancel(
+            "Reset Config", "Are you sure you want to reset the configuration?"
+        )
+        if confirm:
+            copy(CONFIG_FILE_BACKUP, CONFIG_FILE)
+            self.scraper.load_config()
 
     def _draw_plot(self):
         """Draw a plot of the scraped prices over time."""
-        dates, dollars, euros = self.scraper.read_price_log()
+        dates, usd_prices, eur_prices = PriceLogs.read()
 
         fig, ax_raw = plt.subplots(figsize=(10, 8), num="CS2Tracker Price History")
         fig.suptitle("CS2Tracker Price History", fontsize=16)
         fig.autofmt_xdate()
 
         ax = cast(Axes, ax_raw)
-        ax.plot(dates, dollars, label="Dollars")
-        ax.plot(dates, euros, label="Euros")
+        ax.plot(dates, usd_prices, label="Dollars")
+        ax.plot(dates, eur_prices, label="Euros")
         ax.legend()
-        date_formatter = DateFormatter("%d-%m-%Y")
+        date_formatter = DateFormatter("%Y-%m-%d")
         ax.xaxis.set_major_formatter(date_formatter)
 
         plt.show()
 
-    def _edit_log_file(self):
-        """Opens the file containing past price calculations."""
-        Popen([TEXT_EDITOR, OUTPUT_FILE], shell=True)
+    def _export_log_file(self):
+        """Lets the user export the log file to a different location."""
+        export_path = asksaveasfile(
+            title="Export Log File",
+            defaultextension=".csv",
+            filetypes=[("CSV files", "*.csv")],
+        )
+        if export_path:
+            copy(OUTPUT_FILE, export_path.name)
+
+    def _import_log_file(self):
+        """Lets the user import a log file from a different location."""
+        import_path = askopenfilename(
+            title="Import Log File",
+            defaultextension=".csv",
+            filetypes=[("CSV files", "*.csv")],
+        )
+        if not PriceLogs.validate_file(import_path):
+            return
+        copy(import_path, OUTPUT_FILE)
 
     def _toggle_background_task(self, enabled: bool):
         """Toggle whether a daily price calculation should run in the background."""
-        self.scraper.toggle_background_task(enabled)
+        BackgroundTask.toggle(enabled)
 
     def _toggle_use_proxy(self, enabled: bool):
         """Toggle whether the scraper should use proxy servers for requests."""
+        proxy_api_key = self.scraper.config.get("User Settings", "proxy_api_key", fallback=None)
+        if not proxy_api_key and enabled:
+            messagebox.showerror(
+                "Config Error",
+                "You need to enter a valid crawlbase API key into the config file to use this feature.",
+            )
+            return False
+
         self.scraper.toggle_use_proxy(enabled)
+        return True
 
     def _toggle_discord_webhook(self, enabled: bool):
         """Toggle whether the scraper should send notifications to a Discord webhook."""
+        discord_webhook_url = self.scraper.config.get(
+            "User Settings", "discord_webhook_url", fallback=None
+        )
+        if not discord_webhook_url and enabled:
+            messagebox.showerror(
+                "Config Error",
+                "You need to enter a valid Discord webhook URL into the config file to use this feature.",
+            )
+            return False
+
         self.scraper.toggle_discord_webhook(enabled)
+        return True
 
 
 def _popen_and_call(popen_args, callback):

cs2tracker/background_task.py

@@ -0,0 +1,109 @@
+import os
+from subprocess import DEVNULL, call
+
+from cs2tracker.constants import (
+    BATCH_FILE,
+    OS,
+    PROJECT_DIR,
+    PYTHON_EXECUTABLE,
+    RUNNING_IN_EXE,
+    OSType,
+)
+from cs2tracker.padded_console import PaddedConsole
+
+WIN_BACKGROUND_TASK_NAME = "CS2Tracker Daily Calculation"
+WIN_BACKGROUND_TASK_SCHEDULE = "DAILY"
+WIN_BACKGROUND_TASK_TIME = "12:00"
+WIN_BACKGROUND_TASK_CMD = (
+    f"powershell -WindowStyle Hidden -Command \"Start-Process '{BATCH_FILE}' -WindowStyle Hidden\""
+)
+
+console = PaddedConsole()
+
+
+class BackgroundTask:
+    @classmethod
+    def identify(cls):
+        """
+        Search the OS for a daily background task that runs the scraper.
+
+        :return: True if a background task is found, False otherwise.
+        """
+        if OS == OSType.WINDOWS:
+            cmd = ["schtasks", "/query", "/tn", WIN_BACKGROUND_TASK_NAME]
+            return_code = call(cmd, stdout=DEVNULL, stderr=DEVNULL)
+            found = return_code == 0
+            return found
+        else:
+            # TODO: implement finder for cron jobs
+            return False
+
+    @classmethod
+    def _toggle_batch_file(cls, enabled: bool):
+        """
+        Create or delete a batch file that runs the scraper.
+
+        :param enabled: If True, the batch file will be created; if False, the batch
+            file will be deleted.
+        """
+        if enabled:
+            with open(BATCH_FILE, "w", encoding="utf-8") as batch_file:
+                if RUNNING_IN_EXE:
+                    # The python executable is set to the executable itself
+                    # for executables created with PyInstaller
+                    batch_file.write(f"{PYTHON_EXECUTABLE} --only-scrape\n")
+                else:
+                    batch_file.write(f"cd {PROJECT_DIR}\n")
+                    batch_file.write(f"{PYTHON_EXECUTABLE} -m cs2tracker --only-scrape\n")
+        else:
+            if os.path.exists(BATCH_FILE):
+                os.remove(BATCH_FILE)
+
+    @classmethod
+    def _toggle_windows(cls, enabled: bool):
+        """
+        Create or delete a daily background task that runs the scraper on Windows.
+
+        :param enabled: If True, the task will be created; if False, the task will be
+            deleted.
+        """
+        cls._toggle_batch_file(enabled)
+        if enabled:
+            cmd = [
+                "schtasks",
+                "/create",
+                "/tn",
+                WIN_BACKGROUND_TASK_NAME,
+                "/tr",
+                WIN_BACKGROUND_TASK_CMD,
+                "/sc",
+                WIN_BACKGROUND_TASK_SCHEDULE,
+                "/st",
+                WIN_BACKGROUND_TASK_TIME,
+            ]
+            return_code = call(cmd, stdout=DEVNULL, stderr=DEVNULL)
+            if return_code == 0:
+                console.print("[bold green][+] Background task enabled.")
+            else:
+                console.print("[bold red][!] Failed to enable background task.")
+        else:
+            cmd = ["schtasks", "/delete", "/tn", WIN_BACKGROUND_TASK_NAME, "/f"]
+            return_code = call(cmd, stdout=DEVNULL, stderr=DEVNULL)
+            if return_code == 0:
+                console.print("[bold green][-] Background task disabled.")
+            else:
+                console.print("[bold red][!] Failed to disable background task.")
+
+    @classmethod
+    def toggle(cls, enabled: bool):
+        """
+        Create or delete a daily background task that runs the scraper.
+
+        :param enabled: If True, the task will be created; if False, the task will be
+            deleted.
+        """
+        if OS == OSType.WINDOWS:
+            cls._toggle_windows(enabled)
+        else:
+            # TODO: implement toggle for cron jobs
+            pass

cs2tracker/constants.py

@@ -18,7 +18,6 @@ class OSType(enum.Enum):
 
 
 OS = OSType.WINDOWS if sys.platform.startswith("win") else OSType.LINUX
-TEXT_EDITOR = "notepad" if OS == OSType.WINDOWS else "nano"
 PYTHON_EXECUTABLE = sys.executable
 
 
@@ -81,10 +80,10 @@ AUTHOR_STRING = (
 
 POWERSHELL_COLORIZE_OUTPUT = (
     "%{ "
-    "if($_ -match 'Version|\\\\|_') { Write-Host $_ -ForegroundColor yellow } "
+    "if($_ -match '\\[!\\]') { Write-Host $_ -ForegroundColor red } "
+    "elseif($_ -match 'Version|\\\\|_') { Write-Host $_ -ForegroundColor yellow } "
     "elseif($_ -match 'USD|EUR|^[-|\\s]+$') { Write-Host $_ -ForegroundColor green } "
     "elseif($_ -match 'Case|Capsule|-[A-Za-z0-9]') { Write-Host $_ -ForegroundColor magenta } "
-    "elseif($_ -match '\\[!\\]') { Write-Host $_ -ForegroundColor red } "
     "elseif($_ -match 'Legends|Challengers|Contenders|Champions|Finalists') { Write-Host $_ -ForegroundColor blue } "
     "else { Write-Host $_ } "
     "}"