mirror of
https://github.com/Findus23/rebound-collisions.git
synced 2024-09-19 15:53:48 +02:00
allow argparse for file choosing
This commit is contained in:
parent
c984716b15
commit
33a1690e6b
1 changed files with 5 additions and 2 deletions
7
utils.py
7
utils.py
|
@ -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]
|
||||
|
|
Loading…
Reference in a new issue