mirror/rebound-collisions
1
0
Fork 0
mirror of https://github.com/Findus23/rebound-collisions.git synced 2024-09-19 15:53:48 +02:00
Code Releases Activity

allow argparse for file choosing

Browse source
This commit is contained in:
Lukas Winkler 2020-12-28 17:23:03 +01:00
parent c984716b15
commit 33a1690e6b
Signed by: lukas
GPG key ID: 54DE4D798D244853
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
utils.py
View file

@ -18,10 +18,13 @@ def clamp(n: float, smallest: float, largest: float) -> float:
return max(smallest, min(n, largest))
def filename_from_argv() -> Path:
def filename_from_argv(argument: str = None) -> Path:
if len(argv) < 2:
raise ValueError("specify filename")
fn = argv[1]
if argument:
fn = argument
else:
fn = argv[1]
fn = fn.replace(".bin", "").replace(".meta.json", "")
if fn.endswith("."):
fn = fn[:-1]