Add catchup argument

Signed-off-by: Ryan Luu <[email protected]>

Author: RyanLua

src/instawebhooks/__main__.py

@@ -94,6 +94,13 @@ def closure_check_regex(arg_value):
     action="store_true",
 )
 parser.add_argument("--version", action="version", version="%(prog)s " + version)
+parser.add_argument(
+    "-p",
+    "--catchup",
+    help="send the last latest posts on startup regardless of time",
+    type=int,
+    default=0,
+)
 args = parser.parse_args()
 
 # Set the logger to debug if verbose is enabled
@@ -211,7 +218,7 @@ async def send_to_discord(post: Post):
     logger.info("New post sent to Discord successfully.")
 
 
-async def check_for_new_posts():
+async def check_for_new_posts(catchup: int = args.catchup):
     """Check for new Instagram posts and send them to Discord"""
 
     logger.info("Checking for new posts")
@@ -225,12 +232,27 @@ async def check_for_new_posts():
 
     new_posts_found = False
 
+    async def send_post(post: Post):
+        logger.info("New post found: https://www.instagram.com/p/%s", post.shortcode)
+        await send_to_discord(post)
+
+    if catchup > 0:
+        logger.info("Sending last %s posts on startup...", catchup)
+        posts_to_send = []
+        for post in takewhile(lambda _: catchup > 0, posts):
+            posts_to_send.append(post)
+            catchup -= 1
+
+        # Reverse the posts to send oldest first
+        for post in reversed(posts_to_send):
+            await send_post(post)
+            sleep(2)  # Avoid 30 requests per minute rate limit
+
     for post in takewhile(
         lambda p: p.date > until, dropwhile(lambda p: p.date > since, posts)
     ):
         new_posts_found = True
-        logger.info("New post found: https://www.instagram.com/p/%s", post.shortcode)
-        await send_to_discord(post)
+        await send_post(post)
         sleep(2)  # Avoid 30 requests per minute rate limit
 
     if not new_posts_found: