[pt] search by year

hugsy · Dec 15, 2022 · ea342f1 · ea342f1
1 parent fa6d246
commit ea342f1
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/patch_tuesday.py b/patch_tuesday.py
@@ -176,6 +176,8 @@ def get_patch_tuesday_data_soup(month: datetime.date) -> bs4.BeautifulSoup:
     parser.add_argument("-m", "--months", help="Specify the Patch Tuesday month(s) (can be repeated)",
                         default=[], action='append',
                         type=lambda x: datetime.datetime.strptime(x, "%Y-%b"), metavar="YYYY-ABBREVMONTH")
+    parser.add_argument("-y", "--years", help="Specify a year - will be expended in months (can be repeated)",
+                        default=[], action='append', type=int)
     parser.add_argument("--list-products", help="List all products", action="store_true")
     parser.add_argument("--brief", help="Display a summary", action="store_true")
     parser.add_argument("-v", "--verbose", action="count", dest="verbose",
@@ -192,6 +194,11 @@ def get_patch_tuesday_data_soup(month: datetime.date) -> bs4.BeautifulSoup:
         print(f"Using default product as '{DEFAULT_PRODUCT}'")
         args.products = (DEFAULT_PRODUCT,)
 
+    if args.years:
+        for year in args.years:
+            for i in range(12):
+                args.months.append(datetime.date(year, 1 + i, 1))
+
     if not len(args.months):
         print(f"Using default month as '{datetime.date.today().strftime('%B %Y')}'")
         args.months = (datetime.date.today(),)