From 5e33bd56159a24c6ad7690ca46cf87ed975cc03f Mon Sep 17 00:00:00 2001
From: Larissa Glass <larissa.glass@unibas.ch>
Date: Wed, 9 Nov 2022 13:42:46 +0100
Subject: [PATCH] refactor output filename generation
---
tsg/cli.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/tsg/cli.py b/tsg/cli.py
index 29de297..f448fcc 100644
--- a/tsg/cli.py
+++ b/tsg/cli.py
@@ -34,11 +34,13 @@ def get_args():
def output_filename(filename: str) -> str:
filepath = Path(filename)
- if filename.endswith(".csv") or filename.endswith(".tsv"):
- return "generated_" + filepath.stem + ".csv"
- if filename.endswith(".gtf"):
- return "generated_" + filepath.name
-
+ if filepath.suffix == ".csv" or filepath.suffix == ".tsv":
+ outfile = "generated_" + filepath.stem + ".csv"
+ elif filepath.suffix == ".gtf":
+ outfile = "generated_" + filepath.name
+ else:
+ raise NotImplementedError()
+ return outfile
def app():
args = get_args()
--
GitLab