main.py

import time

from selenium import webdriver

from interactor import Bot
from miscellaneous import formatted_time, set_environment
from settings import Settings

if set_environment():
	print("Enter your input in config\\config.json before running again.")
	exit(0)

global_settings = Settings()
global_settings.load_config()
print(global_settings.briefing())

print('Setting driver options.')
options = webdriver.ChromeOptions()
options.add_argument('-headless')
options.add_argument('--log-level=3')

# Chrome specific.
service = webdriver.ChromeService(log_output='chrome-driver.log')

print('Loading web driver.')
driver = webdriver.Chrome(options=options, service=service)
driver.implicitly_wait(10)

print('Setting bot.')
bot = Bot(driver, global_settings)

print('Loading login page.')
driver.get("https://discord.com/login")

# Auto login.

h = 1
while h <= global_settings.login_attempts:
	try:
		print(f'Login attempt ({h}/{global_settings.login_attempts}): ', end='')
		bot.perform_login()
		print('Success.')
		break
	except Exception as e:
		print('Failure.')
	print('Refreshing login page.')
	driver.refresh()
	h += 1

if h > global_settings.login_attempts:
	print('Login could not be performed.')
	exit(1)

# Target server selection.

h = 1
while h <= global_settings.selection_attempts:
	try:
		print(f'Server selection attempt ({h}/{global_settings.selection_attempts}): ', end='')
		bot.select_server()
		print('Success.')	
		break
	except Exception as e:
		print('Failure.')
	print('Refreshing channels page.')
	driver.refresh()
	h += 1

if h > global_settings.selection_attempts:
	print('Target server could not be selected.')
	exit(1)

# Initializing timer.

print('Starting timer.')
start_time = time.time()

# Search and deletion of messages.

nil_count, progress = 0, 0
while nil_count <= global_settings.nil_tolerance and progress < global_settings.target_count:

	print(f'\nNew iteration of search and delete.')
	driver.refresh()

	try:
		# Inform the bot with the right number of messages to delete.
		status = bot.search_and_delete(global_settings.target_count - progress)
	except Exception as e:
		print('\nUnexpected error:', e)
		break

	if status == 0:
		# Total iterations with zero messages deleted.
		nil_count += 1
	else:
		# Total messages deleted.
		progress += status

	elapsed_time = time.time() - start_time
	print(f'\nTotal progress: {progress}\nZero status occurences: {nil_count}\nElapsed time: {formatted_time(int(elapsed_time))}')

# Finishing session.

print('\nInteraction finished.\nClosing the driver.')
driver.quit()