You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
3.6 KiB
110 lines
3.6 KiB
"""
|
|
|
|
Setup Script for RedditVideoMakerBot
|
|
|
|
"""
|
|
|
|
# Imports
|
|
import os
|
|
import time
|
|
from utils.console import print_markdown
|
|
from utils.console import print_step
|
|
from utils.console import print_substep
|
|
from rich.console import Console
|
|
from utils.loader import Loader
|
|
from os.path import exists
|
|
console = Console()
|
|
|
|
setup_done = exists(".setup-done-before")
|
|
|
|
if setup_done == True:
|
|
console.log("[red]Setup was already completed! Please make sure you have to run this script again. If you have to, please delete the file .setup-done-before")
|
|
exit()
|
|
|
|
# These lines ensure the user:
|
|
# - knows they are in setup mode
|
|
# - knows that they are about to erase any other setup files/data.
|
|
|
|
print_step("Setup Assistant")
|
|
|
|
print_markdown(
|
|
"### You're in the setup wizard. Ensure you're supposed to be here, then type yes to continue. If you're not sure, type no to quit."
|
|
)
|
|
|
|
# This Input is used to ensure the user is sure they want to continue.
|
|
ensureSetupIsRequired = input("Are you sure you want to continue? > ").casefold()
|
|
if ensureSetupIsRequired != "yes":
|
|
console.print("[red]Exiting...")
|
|
time.sleep(0.5)
|
|
exit()
|
|
else:
|
|
# Again, let them know they are about to erase all other setup data.
|
|
console.print("[bold red] This will overwrite your current settings. Are you sure you want to continue? [bold green]yes/no")
|
|
overwriteSettings = input("Are you sure you want to continue? > ").casefold()
|
|
if overwriteSettings != "yes":
|
|
console.print("[red]Abort mission! Exiting...")
|
|
time.sleep(0.5)
|
|
exit()
|
|
else:
|
|
# Once they confirm, move on with the script.
|
|
console.print("[bold green]Alright! Let's get started!")
|
|
time.sleep(1)
|
|
|
|
console.log("Ensure you have the following ready to enter:")
|
|
console.log("[bold green]Reddit Client ID")
|
|
console.log("[bold green]Reddit Client Secret")
|
|
console.log("[bold green]Reddit Username")
|
|
console.log("[bold green]Reddit Password")
|
|
console.log("[bold green]Reddit 2FA (yes or no)")
|
|
time.sleep(0.5)
|
|
console.print("[green]If you don't have these, please follow the instructions in the README.md file to set them up.")
|
|
console.print("[green]If you do have these, type yes to continue. If you dont, go ahead and grab those quickly and come back.")
|
|
confirmUserHasCredentials = input("Are you sure you have the credentials? > ").casefold()
|
|
if confirmUserHasCredentials != "yes":
|
|
console.print("[red]I don't understand that.")
|
|
console.print("[red]Exiting...")
|
|
exit()
|
|
else:
|
|
console.print("[bold green]Alright! Let's get started!")
|
|
time.sleep(1)
|
|
|
|
"""
|
|
|
|
Begin the setup process.
|
|
|
|
"""
|
|
|
|
console.log("Enter your credentials now.")
|
|
cliID = input("Client ID > ")
|
|
cliSec = input("Client Secret > ")
|
|
user = input("Username > ")
|
|
passw = input("Password > ")
|
|
twofactor = input("2fa Enabled? (yes/no) > ")
|
|
console.log("Attempting to save your credentials...")
|
|
loader = Loader("Saving Credentials...", "Done!").start()
|
|
# you can also put a while loop here, e.g. while VideoIsBeingMade == True: ...
|
|
time.sleep(0.5)
|
|
console.log("Removing old .env file...")
|
|
os.remove(".env")
|
|
time.sleep(0.5)
|
|
console.log("Creating new .env file...")
|
|
with open('.env', 'a') as f:
|
|
f.write(f'REDDIT_CLIENT_ID="{cliID}"\n')
|
|
time.sleep(0.5)
|
|
f.write(f'REDDIT_CLIENT_SECRET="{cliSec}"\n')
|
|
time.sleep(0.5)
|
|
f.write(f'REDDIT_USERNAME="{user}"\n')
|
|
time.sleep(0.5)
|
|
f.write(f'REDDIT_PASSWORD="{passw}"\n')
|
|
time.sleep(0.5)
|
|
f.write(f'REDDIT_2FA="{twofactor}"\n')
|
|
|
|
with open('.setup-done-before', 'a') as f:
|
|
f.write("This file blocks the setup assistant from running again. Delete this file to run setup again.")
|
|
|
|
loader.stop()
|
|
|
|
console.log("[bold green]Setup Complete! Returning...")
|
|
|
|
# Post-Setup: send message and try to run main.py again.
|
|
os.system("python3 main.py") |