Skip to content

Commit b0b96b8

Browse files
author
Dahlia Li
committed
Add --version flag to CLI
1 parent 2b87064 commit b0b96b8

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

src/cautiousrobot/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
__version__ = "1.1.0"
12
from cautiousrobot.download import download_images
23
from cautiousrobot.buddy_check import BuddyCheck
34
from cautiousrobot.utils import downsample_and_save_image

src/cautiousrobot/__main__.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,21 @@
1414
from cautiousrobot.utils import process_csv, check_existing_images
1515
from cautiousrobot.buddy_check import BuddyCheck
1616
from cautiousrobot.download import download_images
17+
from cautiousrobot import __version__
1718

1819
def parse_args():
1920
available_algorithms = ', '.join(hashlib.algorithms_available)
2021

2122
parser = argparse.ArgumentParser()
2223
# Use argument groups for required vs optional (both get short flags too) https://bugs.python.org/issue9694#msg132327
23-
# Required arguments
24+
# Optional arguments
25+
parser.add_argument(
26+
"--version",
27+
action="version",
28+
version=f"%(prog)s {__version__}",
29+
help="Show version number and exit",
30+
)
31+
# Required arguments
2432
req_args = parser.add_argument_group("required arguments")
2533
req_args.add_argument("-i", "--input-file", required = True, help = "path to CSV file with urls.", nargs = "?")
2634
req_args.add_argument("-o", "--output-dir", required = True, help = "main directory to download images into.", nargs = "?")

0 commit comments

Comments
 (0)