Aria2TUI 0.1.5.0__py3-none-any.whl

aria2tui/aria2tui_app.py

@@ -0,0 +1,289 @@
+#!/bin/python
+
+import os
+import sys
+from sys import exit
+import tempfile
+import time
+import toml
+import json
+import curses
+
+os.chdir(os.path.dirname(os.path.realpath(__file__)))
+os.chdir("../..")
+sys.path.append(os.path.expanduser("../../"))
+
+# from list_picker.ui.list_picker_colours import get_colours, get_help_colours, get_notification_colours
+# from list_picker.utils.options_selectors import default_option_input, output_file_option_selector, default_option_selector
+# from list_picker.utils.table_to_list_of_lists import *
+# from list_picker.utils.utils import *
+# from list_picker.utils.sorting import *
+# from list_picker.utils.filtering import *
+# from list_picker.ui.input_field import *
+# from list_picker.utils.clipboard_operations import *
+# from list_picker.utils.searching import search
+# from list_picker.ui.help_screen import help_lines
+# from list_picker.ui.keys import list_picker_keys, notification_keys, options_keys, menu_keys
+# from list_picker.utils.generate_data import generate_list_picker_data
+# from list_picker.utils.dump import dump_state, load_state, dump_data
+# from list_picker.list_picker_app import Picker, start_curses, close_curses
+from list_picker.list_picker_app import *
+
+from aria2tui.lib.aria2c_wrapper import *
+from aria2tui.utils.aria2c_utils import *
+from aria2tui.ui.aria2_detailing import highlights, menu_highlights, modes, operations_highlights
+from aria2tui.ui.aria2tui_keys import download_option_keys
+from aria2tui.graphing.speed_graph import graph_speeds, graph_speeds_gid
+from aria2tui.ui.aria2tui_menu_options import menu_options, download_options, Option, menu_data, downloads_data, dl_operations_data
+
+
+class Aria2TUI:
+    def __init__(
+        self,
+        stdscr: curses.window,
+        download_options: list[Option],
+        menu_options: list[Option],
+        menu_data: dict,
+        downloads_data: dict,
+        dl_operations_data: dict,
+    ):
+        self.stdscr = stdscr
+        self.download_options = download_options
+        self.menu_options = menu_options
+        self.menu_data = menu_data
+        self.downloads_data = downloads_data
+        self.dl_operations_data = dl_operations_data
+        self.add_graph_to_options()
+        self.add_require_option_to_dl_operations()
+
+
+    def add_require_option_to_dl_operations(self) -> None:
+        self.dl_operations_data["require_option"] =  [False if option.name != "Change Position" else True for option in self.download_options]
+        self.dl_operations_data["option_functions"] = [None if option.name != "Change Position" else lambda stdscr, refresh_screen_function: default_option_selector(stdscr, field_name="Download Position") for option in self.download_options]
+
+    def add_graph_to_options(self) -> None:
+        """ Add the transmission speed graphs to the menu options. """
+        self.download_options.append(
+            Option("Transfer Speed Graph *experimental*", graph_speeds_gid, {
+                "stdscr": self.stdscr,
+                "get_data_function": lambda gid: sendReq(tellStatus(gid)),
+
+                "graph_wh" : lambda: (
+                    9*os.get_terminal_size()[0]//10,
+                    9*os.get_terminal_size()[1]//10,
+                ),
+                "timeout": 1000,
+
+                "xposf" : lambda: os.get_terminal_size()[0]//20,
+                "yposf" : lambda: os.get_terminal_size()[1]//20,
+                "title": "Download Transfer Speeds",
+            }),
+        )
+
+        self.menu_options.append(
+            Option("Transfer Speed Graph *experimental*", graph_speeds, {
+                "stdscr": self.stdscr,
+                "get_data_function": lambda: sendReq(getGlobalStat()),
+                "graph_wh" : lambda: (
+                    9*os.get_terminal_size()[0]//10,
+                    9*os.get_terminal_size()[1]//10,
+                ),
+                "xposf" : lambda: os.get_terminal_size()[0]//20,
+                "yposf" : lambda: os.get_terminal_size()[1]//20,
+                "title": "Global Transfer Speeds",
+            }),
+        )
+
+    def run(self) -> None:
+        """ Run Aria2TUI app loop. """
+        while True:
+            ## SELECT DOWNLOADS
+            DownloadsPicker = Picker(self.stdscr, **self.downloads_data)
+            selected_downloads, opts, self.downloads_data = DownloadsPicker.run()
+
+            if selected_downloads:
+                operation_names = [option.name for option in self.download_options]
+                self.dl_operations_data["items"] = operation_names
+
+                items, header = self.downloads_data["items"], self.downloads_data["header"]
+                gid_index, fname_index = header.index("GID"), header.index("Name")
+                gids = [item[gid_index] for i, item in enumerate(items) if i in selected_downloads]
+                fnames = [item[fname_index] for i, item in enumerate(items) if i in selected_downloads]
+
+                # Display the download names in an infobox when selecting which operation to perform
+                self.dl_operations_data["display_infobox"] = True
+                self.dl_operations_data["infobox_items"] = fnames
+                self.dl_operations_data["infobox_title"] = f"{len(fnames)} Selected..."
+
+                ## SELECT OPERATION TO APPLY TO SELECTED DOWNLOADS
+                DownloadOperationPicker = Picker(self.stdscr, **self.dl_operations_data)
+                selected_operation, opts, self.dl_operations_data = DownloadOperationPicker.run()
+                if selected_operation:
+                    operation = download_options[selected_operation[0]]
+
+                    user_opts = self.dl_operations_data["user_opts"]
+                    view = False
+                    if operation.meta_args and "view" in operation.meta_args and operation.meta_args["view"]: view=True
+                    picker_view = False
+                    if operation.meta_args and "picker_view" in operation.meta_args and operation.meta_args["picker_view"]: picker_view=True
+
+
+                    ## APPLY THE SELECTED OPERATION TO THE SELECTED DOWNLOADS
+                    applyToDownloads(self.stdscr, gids, operation.name, operation.function, operation.function_args, user_opts, view, fnames=fnames, picker_view=picker_view)
+                    self.downloads_data["selections"] = {}
+                    self.dl_operations_data["user_opts"] = ""
+                else: continue
+
+            else: 
+
+                self.menu_data["items"] = [menu_option.name for menu_option in self.menu_options]
+                while True:
+                    ## SELECT MENU OPTION
+                    MenuPicker = Picker(self.stdscr, **self.menu_data)
+                    selected_menu, opts, self.menu_data = MenuPicker.run()
+
+                    # If we exit from the menu then exit altogether
+                    if not selected_menu: 
+                        close_curses(self.stdscr)
+                        return 
+
+                    menu_option = self.menu_options[selected_menu[0]]
+                    if menu_option.name == "View Downloads":
+                        self.downloads_data["auto_refresh"] = False
+                        break
+                    elif menu_option.name == "Watch Downloads":
+                        self.downloads_data["auto_refresh"] = True
+                        break
+
+                    ## if it is a view operation such as "View Global Stats" then send the request and open it with nvim
+                    elif "view" in menu_option.meta_args and menu_option.meta_args["view"]:
+                        # Ensure that the screen is cleared after nvim closes, otherwise artifcats remain.
+                        self.downloads_data["clear_on_start"] = True
+                        self.menu_data["clear_on_start"] = True
+                        response = sendReq(menu_option.function(**menu_option.function_args))
+                        with tempfile.NamedTemporaryFile(delete=False, mode='w') as tmpfile:
+                            tmpfile.write(json.dumps(response, indent=4))
+                            tmpfile_path = tmpfile.name
+                        # cmd = r"""nvim -i NONE -c 'setlocal bt=nofile' -c 'silent! %s/^\s*"function"/\0' -c 'norm ggn'""" + f" {tmpfile_path}"
+                        cmd = f"nvim {tmpfile_path}"
+                        process = subprocess.run(cmd, shell=True, stderr=subprocess.PIPE)
+                    elif "picker_view" in menu_option.meta_args and menu_option.meta_args["picker_view"]:
+                        self.downloads_data["clear_on_start"] = True
+                        self.menu_data["clear_on_start"] = True
+                        response = sendReq(menu_option.function(**menu_option.function_args))
+                        response = flatten_data(response)
+                        resp_list = [[key, val] for key, val in response.items()]
+                        x = Picker(
+                                self.stdscr,
+                                items = resp_list,
+                                header = ["Key", "Val"],
+                                title=menu_option.name,
+                        )
+                        x.run()
+                    else:
+                        if "display_message" in menu_option.meta_args and menu_option.meta_args["display_message"]:
+                            display_message(self.stdscr, menu_option.meta_args["display_message"])
+                        return_val = menu_option.function(**menu_option.function_args)
+
+                        # Add notification of success or failure to listpicker
+                        if return_val not in ["", None, []]:
+                            self.downloads_data["startup_notification"] = str(return_val)
+                        self.stdscr.clear()
+                        self.stdscr.refresh()
+                        break
+
+def display_message(stdscr: curses.window, msg: str) -> None:
+    h, w = stdscr.getmaxyx()
+    if (h>8 and w >20):
+        stdscr.addstr(h//2, (w-len(msg))//2, msg)
+        stdscr.refresh()
+
+
+def handleAriaStartPromt(stdscr):
+    ## Check if aria is running
+    curses.init_pair(1, 253, 232)
+    stdscr.bkgd(' ', curses.color_pair(1))  # Apply background color
+    stdscr.refresh()
+    config = get_config()
+
+    colour_theme_number=config["appearance"]["theme"]
+    while True:
+        connection_up = testConnection()
+        can_connect = testAriaConnection()
+        if not can_connect:
+            if not connection_up:
+                header, choices = ["Aria2c Connection Down. Do you want to start it?"], ["Yes", "No"]
+                connect_data = {
+                    "items": choices,
+                    "title": "Aria2TUI",
+                    "header": header,
+                    "max_selected": 1,
+                    "colour_theme_number": colour_theme_number,
+                }
+                ConnectionPicker = Picker(stdscr, **connect_data)
+
+                choice, opts, function_data = ConnectionPicker.run()
+
+                if choice == [1] or choice == []:
+                    close_curses(stdscr)
+                    exit()
+
+                config = get_config()
+
+                h, w = stdscr.getmaxyx()
+                if (h>8 and w >20):
+                    s = "Starting Aria2c Now..."
+                    stdscr.addstr(h//2-1, (w-len(s))//2, s)
+                    s = f'startupcmds = {config["general"]["startupcmds"]}'
+                    stdscr.addstr(h//2+1, (w-min(len(s), w))//2, s[:w])
+                    stdscr.refresh()
+
+                for cmd in config["general"]["startupcmds"]:
+                    subprocess.Popen(cmd, shell=True, stderr=subprocess.PIPE, stdout=subprocess.PIPE)
+
+                time.sleep(2)
+            else:
+                h, w = stdscr.getmaxyx()
+                if (h>8 and w >20):
+                    s = "The connection is up but unresponsive..."
+                    stdscr.addstr(h//2, (w-len(s))//2, s)
+                    s="Is your token correct in config.toml?"
+                    stdscr.addstr(h//2+1, (w-len(s))//2, s)
+                    stdscr.refresh()
+                    stdscr.timeout(5000)
+                    stdscr.getch()
+                close_curses(stdscr)
+                exit()
+        else:
+            break
+
+
+def aria2tui() -> None:
+    """ Main function """
+
+    ## Run curses
+    stdscr = start_curses()
+
+    ## Check if aria is running and prompt the user to start it if not
+    handleAriaStartPromt(stdscr)
+
+    app = Aria2TUI(
+        stdscr, 
+        download_options,
+        menu_options,
+        menu_data,
+        downloads_data,
+        dl_operations_data,
+    )
+    app.run()
+    # begin(stdscr)
+
+    ## Clean up curses and clear terminal
+    stdscr.clear()
+    stdscr.refresh()
+    close_curses(stdscr)
+    os.system('cls' if os.name == 'nt' else 'clear')
+
+if __name__ == "__main__":
+    # global menu_options, download_options, menu_data, downloads_data, dl_operations_data
+    aria2tui()

aria2tui/graphing/speed_graph.py

@@ -0,0 +1,213 @@
+import sys, os
+# sys.path.append(os.path.expanduser("../../../list_picker/"))
+# sys.path.append(os.path.expanduser("../../../aria2tui/"))
+os.chdir(os.path.dirname(os.path.realpath(__file__)))
+os.chdir("../../..")
+from aria2tui.lib.aria2c_wrapper import *
+from aria2tui.utils.aria2c_utils import *
+# from list_picker.list_picker_app import Picker, start_curses, close_curses
+from list_picker import *
+from list_picker.list_picker_app import *
+import time
+import re
+from typing import Callable
+
+def escape_ansi(line: str):
+    ansi_escape =re.compile(r'(\x9B|\x1B\[)[0-?]*[ -\/]*[@-~]')
+    return ansi_escape.sub('', line)
+
+def handle_plotille_not_found(stdscr: curses.window) -> None:
+    h, w = stdscr.getmaxyx()
+    s = "ModuleNotFoundError: No module named 'plotille'"
+    stdscr.addstr(h//2, (w - len(s))//2, s)
+    stdscr.refresh()
+    stdscr.getch()
+
+
+def graph_speeds(
+        stdscr: curses.window,
+        get_data_function: Callable,
+        timeout:int=1000,
+        title:str="",
+        refresh_time: int = 2,
+        xposf: Callable = lambda: 0,
+        yposf: Callable = lambda: 0,
+        graph_wh: Callable[None,Tuple[int, int]] = lambda: os.get_terminal_size(),
+
+    ) -> None:
+    try:
+        import plotille as plt
+    except:
+        handle_plotille_not_found(stdscr)
+        return None
+
+    initial_time = time.time()
+    x, y, y2 = [], [], []
+    while time.time()-initial_time < timeout:
+
+
+        resp = get_data_function()
+        x.append(time.time()-initial_time)
+        down = int(resp['result']['downloadSpeed'])
+        up = int(resp['result']['uploadSpeed'])
+
+        y.append(down)
+        y2.append(up)
+
+        fig = plt.Figure()
+        fig.plot(x, y)
+        fig.plot(x, y2)
+
+        fig.y_ticks_fkt = lambda y, _: bytes_to_human_readable(y)
+        fig.x_ticks_fkt = lambda x, _: f"{int(x)}s"
+        fig.set_y_limits(min_=0)
+        fig.set_x_limits(min_=0)
+        fig.text([x[-1]], [y[0]], ['Dn'])
+        fig.text([x[0]], [y2[0]], ['Up'])
+
+        width, height = graph_wh()
+        fig.width = width - 7
+        fig.height = height - 4
+        globh, globw = stdscr.getmaxyx()
+        xpos, ypos = xposf(), yposf()
+        maxw, maxh = globw-xpos-1, globh-ypos
+
+        stdscr.erase()
+
+        # Draw title
+        stdscr.addstr(ypos, xpos, f"{title:^{min(width,maxw)}}")
+
+        # Draw graph
+        lines = fig.show().split('\n')
+        for i, line in enumerate(lines):
+            if i > maxh-2: break
+            line = escape_ansi(line)
+
+            stdscr.addstr(i+1+ypos, xpos , line[:min(width,maxw)])
+
+        # Show the extreme points of the width and also the last printable char
+        show_control_chars = False
+        if show_control_chars:
+            stdscr.addstr(0,0, f"{maxw}, {maxh}")
+            stdscr.addstr(ypos, xpos, "*", curses.A_REVERSE)
+            stdscr.addstr(ypos+maxh-1, xpos, "*", curses.A_REVERSE)
+            stdscr.addstr(ypos, xpos+maxw-1, "*", curses.A_REVERSE)
+            stdscr.addstr(ypos+maxh-1, xpos+maxw-2, "*", curses.A_REVERSE)
+            stdscr.addstr(ypos+ min(height, maxh-1), xpos, "+", curses.A_REVERSE)
+            stdscr.addstr(ypos, xpos+min(maxw-1, width), "+", curses.A_REVERSE)
+            stdscr.addstr(ypos+min(height, maxh-1), xpos+min(width, maxw-2), "+", curses.A_REVERSE)
+
+        stdscr.refresh()
+        key = stdscr.getch()
+        if key in [3, ord('q')]: 
+            return None
+
+def graph_speeds_gid(
+        stdscr: curses.window,
+        get_data_function: Callable,
+        timeout:int=1000,
+        title:str="",
+        refresh_time: int = 2,
+        xposf: Callable = lambda: 0,
+        yposf: Callable = lambda: 0,
+        graph_wh: Callable[None,Tuple[int, int]] = lambda: os.get_terminal_size(),
+        gid:str = ""
+
+    ) -> None:
+    try:
+        import plotille as plt
+    except:
+        handle_plotille_not_found(stdscr)
+        return None
+
+    initial_time = time.time()
+    x, y, y2 = [], [], []
+    while time.time()-initial_time < timeout:
+
+        resp = get_data_function(gid)
+        x.append(time.time()-initial_time)
+        down = int(resp['result']['downloadSpeed'])
+        up = int(resp['result']['uploadSpeed'])
+
+        y.append(down)
+        y2.append(up)
+
+        fig = plt.Figure()
+        fig.plot(x, y)
+        fig.plot(x, y2)
+
+        fig.y_ticks_fkt = lambda y, _: bytes_to_human_readable(y)
+        fig.x_ticks_fkt = lambda x, _: f"{int(x)}s"
+        fig.set_y_limits(min_=0)
+        fig.set_x_limits(min_=0)
+        fig.text([x[-1]], [y[0]], ['Dn'])
+        fig.text([x[0]], [y2[0]], ['Up'])
+
+        width, height = graph_wh()
+        fig.width = width - 7
+        fig.height = height - 4
+        globh, globw = stdscr.getmaxyx()
+        xpos, ypos = xposf(), yposf()
+        maxw, maxh = globw-xpos-1, globh-ypos
+
+        stdscr.erase()
+
+        # Draw title
+        stdscr.addstr(ypos, xpos, f"{title:^{min(width,maxw)}}")
+
+        # Draw graph
+        lines = fig.show().split('\n')
+        for i, line in enumerate(lines):
+            if i > maxh-2: break
+            line = escape_ansi(line)
+
+            stdscr.addstr(i+1+ypos, xpos , line[:min(width,maxw)])
+
+        # Show the extreme points of the width and also the last printable char
+        show_control_chars = False
+        if show_control_chars:
+            stdscr.addstr(0,0, f"{maxw}, {maxh}")
+            stdscr.addstr(ypos, xpos, "*", curses.A_REVERSE)
+            stdscr.addstr(ypos+maxh-1, xpos, "*", curses.A_REVERSE)
+            stdscr.addstr(ypos, xpos+maxw-1, "*", curses.A_REVERSE)
+            stdscr.addstr(ypos+maxh-1, xpos+maxw-2, "*", curses.A_REVERSE)
+            stdscr.addstr(ypos+ min(height, maxh-1), xpos, "+", curses.A_REVERSE)
+            stdscr.addstr(ypos, xpos+min(maxw-1, width), "+", curses.A_REVERSE)
+            stdscr.addstr(ypos+min(height, maxh-1), xpos+min(width, maxw-2), "+", curses.A_REVERSE)
+
+        stdscr.refresh()
+        key = stdscr.getch()
+        stdscr.refresh()
+        if key in [3, ord('q')]: 
+            return None
+        resp = get_data_function(gid)
+
+
+
+if __name__ == "__main__":
+    title = "Global Transfer Speeds"
+    end_time = 180
+    get_data_function = lambda: sendReq(getGlobalStat())
+    gid = "60e0c88ed77a24d6"
+    get_data_function = lambda: sendReq(tellStatus(gid))
+    wait_time = 2
+
+    stdscr = start_curses()
+    size_func = lambda: (
+        3*os.get_terminal_size()[0]//4,
+        3*os.get_terminal_size()[1]//4,
+    )
+    xposf = lambda: os.get_terminal_size()[0]//8
+    yposf = lambda: os.get_terminal_size()[1]//8
+    graph_speeds(
+        stdscr,
+        get_data_function=get_data_function,
+        timeout=end_time,
+        refresh_time=wait_time,
+        title=title,
+        graph_wh= size_func,
+        xposf=xposf,
+        yposf=yposf,
+            
+    )
+    close_curses(stdscr)

aria2tui/graphing/speed_graph_plain_text.py

@@ -0,0 +1,56 @@
+import time
+import os
+import subprocess
+import toml
+from urllib import request as rq
+import json
+import sys
+os.chdir(os.path.dirname(os.path.realpath(__file__)))
+os.chdir("../../..")
+# sys.path.append(os.path.expanduser("../list_picker/"))
+import tempfile
+import tabulate
+from typing import Callable, Tuple
+
+from aria2tui.lib.aria2c_wrapper import *
+from aria2tui.utils.aria_adduri import addDownloadFull
+# from list_picker.utils.utils import *
+from list_picker import *
+from list_picker.list_picker_app import *
+
+def graph_speeds_no_curses(get_data_function, end_time, title, wait_time):
+    ticker = 0
+    x, y, y2 = [], [], []
+    while ticker < end_time:
+        x.append(ticker)
+        resp = get_data_function()
+        down = int(resp['result']['downloadSpeed'])
+        up = int(resp['result']['uploadSpeed'])
+        # down = bytes_to_human_readable(resp['result']['downloadSpeed'])
+        y.append(down)
+        y2.append(up)
+        fig = plt.Figure()
+        fig.lc = 25
+        fig.color_mode = 'byte'
+        # fig.lc = 'red'
+        fig.bg=100
+        fig.plot(x, y, lc=200, label='Download speed')
+        fig.plot(x, y2, lc=25, label='Upload speed')
+        fig.y_ticks_fkt = lambda y, _: bytes_to_human_readable(y)
+        fig.x_ticks_fkt = lambda x, _: f"{int(x)}s"
+        fig.set_y_limits(min_=0)
+        fig.set_x_limits(min_=0)
+# min(4, max(0, len(y)-1))
+        fig.text([x[-1]], [y[0]], ['Down'], lc=200)
+        fig.text([x[0]], [y2[0]], ['Up'], lc=25)
+        width, height = os.get_terminal_size()
+        fig.width = width - 28
+        fig.height = height - 8
+        os.system('cls' if os.name == 'nt' else 'clear')
+        print()
+        print(f"{title:^{width}}")
+        print()
+        print(fig.show())
+        # print(plotille.plot(x, y, height=20, width=60))
+        time.sleep(wait_time)
+        ticker += wait_time