diff options
| -rw-r--r-- | fietsboek/scripts/fietsctl.py | 20 |
1 files changed, 18 insertions, 2 deletions
diff --git a/fietsboek/scripts/fietsctl.py b/fietsboek/scripts/fietsctl.py index d08be17..7702f05 100644 --- a/fietsboek/scripts/fietsctl.py +++ b/fietsboek/scripts/fietsctl.py @@ -384,6 +384,18 @@ def cmd_track_list(config: str): @cmd_track.command("add") @config_option +@click.option( + "--visibility", + help="Visibility of the added tracks.", + type=click.Choice([vis.name for vis in models.track.Visibility]), + default="PUBLIC", +) +@click.option( + "--track-type", + help="Type of the added tracks.", + type=click.Choice([typ.name for typ in models.track.TrackType]), + default="ORGANIC", +) @optgroup.group("User selection", cls=RequiredMutuallyExclusiveOptionGroup) @optgroup.option("--id", "-i", "id_", help="Database ID of the user.", type=int) @optgroup.option("--email", "-e", help="Email address of the user.") @@ -392,6 +404,8 @@ def cmd_track_list(config: str): def cmd_track_add( ctx: click.Context, config: str, + visibility: str, + track_type: str, id_: int | None, email: str | None, filename: list[str], @@ -407,6 +421,8 @@ def cmd_track_add( query = select(models.User).filter_by(id=id_) else: query = models.User.query_by_email(email) + visibility = models.track.Visibility[visibility] + track_type = models.track.TrackType[track_type] with env["request"].tm: request = env["request"] dbsession = request.dbsession @@ -440,8 +456,8 @@ def cmd_track_add( user, title, track.date, - models.track.Visibility.PRIVATE, - models.track.TrackType.ORGANIC, + visibility, + track_type, "", [], [], |