From a874a6134e48b8319b1e10f3210f091b01ca4321 Mon Sep 17 00:00:00 2001 From: Jesse Bannon Date: Sat, 4 Nov 2023 23:13:09 -0700 Subject: [PATCH] handle int/float in dl --- src/ytdl_sub/cli/parsers/dl.py | 5 ++++- tests/e2e/conftest.py | 1 - 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/ytdl_sub/cli/parsers/dl.py b/src/ytdl_sub/cli/parsers/dl.py index 157a156f..39c28626 100644 --- a/src/ytdl_sub/cli/parsers/dl.py +++ b/src/ytdl_sub/cli/parsers/dl.py @@ -134,11 +134,14 @@ class DownloadArgsParser: next_dict[arg_name_split[-1]] = arg_value - # TODO: handle ints/floats if arg_value == "True": next_dict[arg_name_split[-1]] = True elif arg_value == "False": next_dict[arg_name_split[-1]] = False + elif arg_value.isdigit(): + next_dict[arg_name_split[-1]] = int(arg_value) + elif arg_value.replace(".", "", 1).isdigit(): + next_dict[arg_name_split[-1]] = float(arg_value) return argument_dict diff --git a/tests/e2e/conftest.py b/tests/e2e/conftest.py index 250d439b..d2957a27 100644 --- a/tests/e2e/conftest.py +++ b/tests/e2e/conftest.py @@ -9,7 +9,6 @@ import pytest from ytdl_sub.cli.entrypoint import main from ytdl_sub.subscriptions.subscription import Subscription from ytdl_sub.utils.file_handler import FileHandler -from ytdl_sub.utils.system import IS_WINDOWS @pytest.fixture