chore: switch to ruff as formatter

Author: qtchaos

.editorconfig

@@ -4,4 +4,4 @@ root = true
 end_of_line = lf
 insert_final_newline = true
 indent_size = 4
-indent_style = space
+indent_style = tab

.github/workflows/ruff.yml

@@ -0,0 +1,8 @@
+name: Ruff
+on: [push, pull_request]
+jobs:
+    ruff:
+        runs-on: ubuntu-latest
+        steps:
+            - uses: actions/checkout@v4
+            - uses: chartboost/ruff-action@v1

.gitignore

@@ -5,4 +5,6 @@ venv
 .venv
 __pycache__
 config.json
-credentials.txt
+credentials.txt
+.ruff_cache
+.mypy_cache

.vscode/extensions.json

@@ -1,3 +1,4 @@
 {
-  "recommendations": ["editorconfig.editorconfig"]
+  "recommendations": ["editorconfig.editorconfig", "charliermarsh.ruff", "ms-python.mypy-type-checker"]
 }
+    

.vscode/settings.json

@@ -1,7 +1,7 @@
 {
-	"python.analysis.typeCheckingMode": "basic",
-	"python.analysis.autoImportCompletions": true,
-	"editor.defaultFormatter": "ms-python.autopep8",
 	"editor.formatOnSave": true,
-	"editor.formatOnPaste": true
+	"editor.formatOnPaste": true,
+	"[python]": {
+		"editor.defaultFormatter": "charliermarsh.ruff"
+	}
 }

main.py

@@ -1,4 +1,5 @@
 """Main file for the project, handles the arguments and calls the other files."""
+
 import asyncio
 import argparse
 import os
@@ -9,151 +10,209 @@
 from services.alive import keepalive
 from services.upload import upload_file
 from services.extract import extract_credentials
-from utilities.fs import Config, concrete_read_config, read_config, write_config, write_default_config, save_credentials
-from utilities.web import generate_mail, type_name, type_password, initial_setup, mail_login, get_mail
-from utilities.etc import Credentials, p_print, clear_console, Colours, clear_tmp, reinstall_tenacity, check_for_updates, delete_default
+from utilities.fs import (
+	Config,
+	concrete_read_config,
+	read_config,
+	write_config,
+	write_default_config,
+	save_credentials,
+)
+from utilities.web import (
+	generate_mail,
+	type_name,
+	type_password,
+	initial_setup,
+	mail_login,
+	get_mail,
+)
+from utilities.etc import (
+	Credentials,
+	p_print,
+	clear_console,
+	Colours,
+	clear_tmp,
+	reinstall_tenacity,
+	check_for_updates,
+	delete_default,
+)
 
 # Spooky import to check if the correct version of tenacity is installed.
 if sys.version_info.major == 3 and sys.version_info.minor <= 11:
-    try:
-        from mega import Mega
-    except AttributeError:
-        reinstall_tenacity()
+	try:
+		pass
+	except AttributeError:
+		reinstall_tenacity()
 
 default_installs = [
-    "C:/Program Files/Google/Chrome/Application/chrome.exe",
-    "C:/Program Files (x86)/Google/Chrome/Application/chrome.exe",
-    "C:/Program Files/BraveSoftware/Brave-Browser/Application/brave.exe",
-    "C:/Program Files (x86)/BraveSoftware/Brave-Browser/Application/brave.exe",
-    "C:/Program Files/Microsoft/Edge/Application/msedge.exe",
+	"C:/Program Files/Google/Chrome/Application/chrome.exe",
+	"C:/Program Files (x86)/Google/Chrome/Application/chrome.exe",
+	"C:/Program Files/BraveSoftware/Brave-Browser/Application/brave.exe",
+	"C:/Program Files (x86)/BraveSoftware/Brave-Browser/Application/brave.exe",
+	"C:/Program Files/Microsoft/Edge/Application/msedge.exe",
 ]
 args = [
-    "--no-sandbox",
-    "--disable-setuid-sandbox",
-    "--disable-infobars",
-    "--window-position=0,0",
-    "--ignore-certificate-errors",
-    "--ignore-certificate-errors-spki-list",
-    '--user-agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:126.0) Gecko/20100101 Firefox/126.0"',
+	"--no-sandbox",
+	"--disable-setuid-sandbox",
+	"--disable-infobars",
+	"--window-position=0,0",
+	"--ignore-certificate-errors",
+	"--ignore-certificate-errors-spki-list",
+	'--user-agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:126.0) Gecko/20100101 Firefox/126.0"',
 ]
 
 parser = argparse.ArgumentParser()
-parser.add_argument('-ka', '--keepalive', required=False, action='store_true',
-                    help='Logs into the accounts to keep them alive.')
-parser.add_argument('-e', '--extract', required=False, action='store_true', help='Extracts the credentials to a file.')
-parser.add_argument('-v', '--verbose', required=False, action='store_true',
-                    help='Shows storage left while using keepalive function.')
-parser.add_argument('-f', '--file', required=False,
-                    help='Uploads a file to the account.')
-parser.add_argument('-p', '--public', required=False, action='store_true',
-                    help='Generates a public link to the uploaded file, use with -f')
-parser.add_argument('-l', '--loop', required=False,
-                    help="Loops the program for a specified amount of times.", type=int)
+parser.add_argument(
+	"-ka",
+	"--keepalive",
+	required=False,
+	action="store_true",
+	help="Logs into the accounts to keep them alive.",
+)
+parser.add_argument(
+	"-e",
+	"--extract",
+	required=False,
+	action="store_true",
+	help="Extracts the credentials to a file.",
+)
+parser.add_argument(
+	"-v",
+	"--verbose",
+	required=False,
+	action="store_true",
+	help="Shows storage left while using keepalive function.",
+)
+parser.add_argument(
+	"-f", "--file", required=False, help="Uploads a file to the account."
+)
+parser.add_argument(
+	"-p",
+	"--public",
+	required=False,
+	action="store_true",
+	help="Generates a public link to the uploaded file, use with -f",
+)
+parser.add_argument(
+	"-l",
+	"--loop",
+	required=False,
+	help="Loops the program for a specified amount of times.",
+	type=int,
+)
 
 console_args = parser.parse_args()
 
 
 def setup() -> Tuple[str, Config]:
-    """Sets up the configs so everything runs smoothly."""
-
-    executable_path = ""
-    config = read_config()
-
-    if config is None:
-        write_default_config()
-        config = concrete_read_config()
-    else:
-        executable_path = config.executablePath
-
-    # If no Chromium based browser is found, ask the user for the path to one.
-    if not executable_path:
-        p_print(
-            "Failed to find a Chromium based browser. Please make sure you have one installed.", Colours.FAIL)
-        executable_path = input(
-            "Please enter the path to a Chromium based browser's executable: ")
-        if os.path.exists(executable_path):
-            p_print("Found executable!", Colours.OKGREEN)
-            write_config("executablePath", executable_path, config)
-        else:
-            p_print("Failed to find executable!", Colours.FAIL)
-            sys.exit(1)
-
-    return executable_path, config
+	"""Sets up the configs so everything runs smoothly."""
+
+	executable_path = ""
+	config = read_config()
+
+	if config is None:
+		write_default_config()
+		config = concrete_read_config()
+	else:
+		executable_path = config.executablePath
+
+	# If no Chromium based browser is found, ask the user for the path to one.
+	if not executable_path:
+		p_print(
+			"Failed to find a Chromium based browser. Please make sure you have one installed.",
+			Colours.FAIL,
+		)
+		executable_path = input(
+			"Please enter the path to a Chromium based browser's executable: "
+		)
+		if os.path.exists(executable_path):
+			p_print("Found executable!", Colours.OKGREEN)
+			write_config("executablePath", executable_path, config)
+		else:
+			p_print("Failed to find executable!", Colours.FAIL)
+			sys.exit(1)
+
+	return executable_path, config
 
 
 def loop_registrations(loop_count: int, executable_path: str, config: Config):
-    """Registers accounts in a loop."""
-    for _ in range(loop_count):
-        p_print(f"Loop {_ + 1}/{loop_count}", Colours.OKGREEN)
-        clear_tmp()
+	"""Registers accounts in a loop."""
+	for _ in range(loop_count):
+		p_print(f"Loop {_ + 1}/{loop_count}", Colours.OKGREEN)
+		clear_tmp()
 
-        credentials = asyncio.run(generate_mail())
-        asyncio.run(register(credentials, executable_path, config))
+		credentials = asyncio.run(generate_mail())
+		asyncio.run(register(credentials, executable_path, config))
 
 
 async def register(credentials: Credentials, executable_path: str, config: Config):
-    """Registers and verifies mega.nz account."""
-    browser = await pyppeteer.launch({
-        "headless": True,
-        "ignoreHTTPSErrors": True,
-        "userDataDir": f"{os.getcwd()}/tmp",
-        "args": args,
-        "executablePath": executable_path,
-        "autoClose": False,
-        "ignoreDefaultArgs": ["--enable-automation", "--disable-extensions"],
-    })
-
-    context = await browser.createIncognitoBrowserContext()
-    page = await context.newPage()
-
-    await type_name(page, credentials)
-    await type_password(page, credentials)
-    mail = await mail_login(credentials)
-
-    await asyncio.sleep(1.5)
-    message = await get_mail(mail)
-
-    await initial_setup(context, message, credentials)
-    await asyncio.sleep(0.5)
-    await browser.close()
-
-    p_print("Verified account.", Colours.OKGREEN)
-    p_print(
-        f"Email: {credentials.email}\nPassword: {credentials.password}",
-        Colours.OKCYAN,
-    )
-
-    delete_default(credentials)
-    save_credentials(credentials, config.accountFormat)
-
-    if console_args.file is not None:
-        file_size = os.path.getsize(console_args.file)
-        if os.path.exists(console_args.file) and 0 < file_size < 2e+10:
-            if file_size >= 5e+9:
-                p_print("File is larger than 5GB, mega.nz limits traffic to 5GB per IP.", Colours.WARNING)
-            upload_file(console_args.public, console_args.file, credentials)
-        else:
-            p_print("File not found.", Colours.FAIL)
-    if console_args.loop is None or console_args.loop <= 1:
-        sys.exit(0)
+	"""Registers and verifies mega.nz account."""
+	browser = await pyppeteer.launch(
+		{
+			"headless": True,
+			"ignoreHTTPSErrors": True,
+			"userDataDir": f"{os.getcwd()}/tmp",
+			"args": args,
+			"executablePath": executable_path,
+			"autoClose": False,
+			"ignoreDefaultArgs": ["--enable-automation", "--disable-extensions"],
+		}
+	)
+
+	context = await browser.createIncognitoBrowserContext()
+	page = await context.newPage()
+
+	await type_name(page, credentials)
+	await type_password(page, credentials)
+	mail = await mail_login(credentials)
+
+	await asyncio.sleep(1.5)
+	message = await get_mail(mail)
+
+	await initial_setup(context, message, credentials)
+	await asyncio.sleep(0.5)
+	await browser.close()
+
+	p_print("Verified account.", Colours.OKGREEN)
+	p_print(
+		f"Email: {credentials.email}\nPassword: {credentials.password}",
+		Colours.OKCYAN,
+	)
+
+	delete_default(credentials)
+	save_credentials(credentials, config.accountFormat)
+
+	if console_args.file is not None:
+		file_size = os.path.getsize(console_args.file)
+		if os.path.exists(console_args.file) and 0 < file_size < 2e10:
+			if file_size >= 5e9:
+				p_print(
+					"File is larger than 5GB, mega.nz limits traffic to 5GB per IP.",
+					Colours.WARNING,
+				)
+			upload_file(console_args.public, console_args.file, credentials)
+		else:
+			p_print("File not found.", Colours.FAIL)
+	if console_args.loop is None or console_args.loop <= 1:
+		sys.exit(0)
 
 
 if __name__ == "__main__":
-    clear_console()
-    check_for_updates()
-
-    executable_path, config = setup()
-    if not executable_path:
-        p_print("Failed while setting up!", Colours.FAIL)
-        sys.exit(1)
-
-    if console_args.extract:
-        extract_credentials(config.accountFormat)
-    elif console_args.keepalive:
-        keepalive(console_args.verbose)
-    elif console_args.loop is not None and console_args.loop > 1:
-        loop_registrations(console_args.loop, executable_path, config)
-    else:
-        clear_tmp()
-        credentials = asyncio.run(generate_mail())
-        asyncio.run(register(credentials, executable_path, config))
+	clear_console()
+	check_for_updates()
+
+	executable_path, config = setup()
+	if not executable_path:
+		p_print("Failed while setting up!", Colours.FAIL)
+		sys.exit(1)
+
+	if console_args.extract:
+		extract_credentials(config.accountFormat)
+	elif console_args.keepalive:
+		keepalive(console_args.verbose)
+	elif console_args.loop is not None and console_args.loop > 1:
+		loop_registrations(console_args.loop, executable_path, config)
+	else:
+		clear_tmp()
+		credentials = asyncio.run(generate_mail())
+		asyncio.run(register(credentials, executable_path, config))