From 6bc60773263e60bd3628544d28f11dc10ff72ae7 Mon Sep 17 00:00:00 2001
From: iaacornus <iaacornus.devel@gmail.com>
Date: Tue, 7 Jun 2022 20:03:36 +0800
Subject: [PATCH] repeat 3 times after fail

---
 cli.py | 22 ++++++++++++++++------
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/cli.py b/cli.py
index 3b479a0..021eba1 100644
--- a/cli.py
+++ b/cli.py
@@ -1,4 +1,5 @@
 import argparse
+from venv import create
 
 from main import main
 from setup_program import setup
@@ -57,12 +58,21 @@ def program_options():
 
     try:
         if args.create:
-            main(
-                args.subreddit,
-                args.background,
-                args.filename,
-                args.thread,
-            )
+            trial = 0
+            while trial < 3:
+                create = main(
+                    args.subreddit,
+                    args.background,
+                    args.filename,
+                    args.thread,
+                )
+                if not create:
+                    try_again = input("Something went wrong! Try again? [y/N] > ").strip()
+                    if try_again in ["y", "Y"]:
+                        trial += 1
+                        continue
+
+                break
         elif args.setup:
             setup()
         else: