Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Run a formatter on the codebase #150

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Run a formatter on the codebase #150

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2,123 changes: 1,476 additions & 647 deletions IMDBTraktSyncer/IMDBTraktSyncer.py
View file
Open in desktop

Large diffs are not rendered by default.

145 changes: 92 additions & 53 deletions IMDBTraktSyncer/arguments.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
import platform
import stat


def try_remove(file_path, retries=3, delay=1):
"""
Tries to remove a file or directory, retrying if it's in use or read-only.
Expand All @@ -22,21 +23,21 @@ def try_remove(file_path, retries=3, delay=1):
for root, dirs, files in os.walk(file_path, topdown=False):
for name in files:
file = os.path.join(root, name)
if sys.platform != 'win32': # chmod on Linux/macOS
if sys.platform != "win32": # chmod on Linux/macOS
os.chmod(file, 0o777) # Make file writable
os.remove(file)
for name in dirs:
folder = os.path.join(root, name)
if sys.platform != 'win32': # chmod on Linux/macOS
if sys.platform != "win32": # chmod on Linux/macOS
os.chmod(folder, 0o777) # Make folder writable
os.rmdir(folder)

if sys.platform != 'win32': # chmod on Linux/macOS
if sys.platform != "win32": # chmod on Linux/macOS
os.chmod(file_path, 0o777) # Make the top-level folder writable
os.rmdir(file_path) # Finally, remove the directory
else:
# It's a file, ensure it's writable and remove it
if sys.platform != 'win32': # chmod on Linux/macOS
if sys.platform != "win32": # chmod on Linux/macOS
os.chmod(file_path, 0o777) # Make it writable
os.remove(file_path)

Expand All @@ -50,7 +51,7 @@ def try_remove(file_path, retries=3, delay=1):
time.sleep(delay)

# If running on Windows, handle read-only files
if sys.platform == 'win32':
if sys.platform == "win32":
try:
# Remove read-only attribute on Windows
if os.path.exists(file_path):
Expand All @@ -66,39 +67,56 @@ def try_remove(file_path, retries=3, delay=1):

return False


def get_selenium_install_location():
try:
# Use pip show to get Selenium installation details
result = subprocess.run([sys.executable, '-m', 'pip', 'show', 'selenium'],
stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True)
result = subprocess.run(
[sys.executable, "-m", "pip", "show", "selenium"],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
text=True,
check=True,
)
for line in result.stdout.splitlines():
if line.startswith("Location:"):
site_packages_directory = line.split("Location:")[1].strip()
selenium_directory = os.path.join(site_packages_directory, 'selenium')
selenium_directory = os.path.join(site_packages_directory, "selenium")
return selenium_directory
except Exception as e:
print("Error finding Selenium install location using pip show:", e)
return None


def clear_selenium_manager_cache():
try:
# Get the Selenium install location
selenium_install_location = get_selenium_install_location()
if not selenium_install_location:
print("Could not determine Selenium install location. Skipping cache clear.")
print(
"Could not determine Selenium install location. Skipping cache clear."
)
return

webdriver_common_path = os.path.join(selenium_install_location, "webdriver", "common")

webdriver_common_path = os.path.join(
selenium_install_location, "webdriver", "common"
)

# Determine the OS and set the appropriate folder and file name
os_name = platform.system().lower()

if os_name == "windows":
selenium_manager_path = os.path.join(webdriver_common_path, "windows", "selenium-manager.exe")
selenium_manager_path = os.path.join(
webdriver_common_path, "windows", "selenium-manager.exe"
)
elif os_name == "linux":
selenium_manager_path = os.path.join(webdriver_common_path, "linux", "selenium-manager")
selenium_manager_path = os.path.join(
webdriver_common_path, "linux", "selenium-manager"
)
elif os_name == "darwin": # macOS
selenium_manager_path = os.path.join(webdriver_common_path, "macos", "selenium-manager")
selenium_manager_path = os.path.join(
webdriver_common_path, "macos", "selenium-manager"
)
else:
print("Unsupported operating system.")
return
Expand All @@ -109,22 +127,35 @@ def clear_selenium_manager_cache():
return

# Build the command
command = f"{selenium_manager_path} --clear-cache --browser chrome --driver chromedriver"
command = (
f"{selenium_manager_path} --clear-cache --browser chrome "
f"--driver chromedriver"
)

try:
# Run the command
result = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True)
result = subprocess.run(
command,
shell=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
text=True,
check=True,
)
print("Selenium Chromedriver cache cleared")
except subprocess.CalledProcessError as e:
print("Error running Selenium Manager command:", e.stderr)
except Exception as e:
print("An unexpected error occurred:", e)


def clear_user_data(main_directory):
"""
Deletes the credentials.txt file and Chrome user data directories under the given main directory.
Deletes the credentials.txt file and Chrome user data directories under the given
main directory.

:param main_directory: Directory path where credentials.txt and user data should be deleted.
:param main_directory: Directory path where credentials.txt and user data should be
deleted.
"""
credentials_path = os.path.join(main_directory, "credentials.txt")

Expand All @@ -135,54 +166,55 @@ def clear_user_data(main_directory):
# Construct the Chrome user data directory path dynamically
chrome_directory = os.path.join(main_directory, "Chrome")
user_data_found = False

if os.path.exists(chrome_directory):
for root, dirs, files in os.walk(chrome_directory):
for dir_name in dirs:
if dir_name == "userData":
user_data_directory = os.path.join(root, dir_name)
user_data_found = True
break

if user_data_found:
try_remove(user_data_directory)

print("Clear user data complete.")


def clear_cache(main_directory):
"""
Deletes all folders, .zip files, .txt files (except credentials.txt) in the given directory and clears selenium chromedriver cache.

Deletes all folders, .zip files, .txt files (except credentials.txt) in the given
directory and clears selenium chromedriver cache.
:param main_directory: Directory path where data should be cleared.
"""
# Check if the given directory exists
if not os.path.exists(main_directory):
print(f"Error: The directory {main_directory} does not exist.")
return

# Walk through all files and folders in the directory
for root, dirs, files in os.walk(main_directory, topdown=False):
# Delete files first
for file in files:
file_path = os.path.join(root, file)

# Skip deleting credentials.txt
if file == "credentials.txt":
print(f"Skipping {file_path} (credentials.txt)")
continue # Skip this file

# Remove .zip files
if file.endswith(".zip"):
try_remove(file_path)

# Remove .txt files (except credentials.txt)
elif file.endswith(".txt"):
try_remove(file_path)

# Remove .csv files
elif file.endswith(".csv"):
try_remove(file_path)

# Delete directories after files
for dir_name in dirs:
dir_path = os.path.join(root, dir_name)
Expand All @@ -195,46 +227,47 @@ def clear_cache(main_directory):
print(f"Permission error removing directory {dir_path}: {e}")
except Exception as e:
print(f"Error removing directory {dir_path}: {e}")

# Clear selenium chromedriver cache
clear_selenium_manager_cache()

print("Clear cache complete.")



def uninstall(main_directory):
"""
Deletes all folders, .zip files, .txt files (except credentials.txt and log.txt) in the given directory and clears selenium chromedriver cache before uninstalling.
Deletes all folders, .zip files, .txt files (except credentials.txt and log.txt) in
the given directory and clears selenium chromedriver cache before uninstalling.
:param main_directory: Directory path where data should be cleared.
"""
# Check if the given directory exists
if not os.path.exists(main_directory):
print(f"Error: The directory {main_directory} does not exist.")
return

# Walk through all files and folders in the directory
for root, dirs, files in os.walk(main_directory, topdown=False):
# Delete files first
for file in files:
file_path = os.path.join(root, file)

# Skip deleting credentials.txt
if file in ["credentials.txt", "log.txt"]:
print(f"Skipping {file_path} (credentials.txt)")
continue # Skip this file

# Remove .zip files
if file.endswith(".zip"):
try_remove(file_path)

# Remove .txt files (except credentials.txt)
elif file.endswith(".txt"):
try_remove(file_path)

# Remove .csv files
elif file.endswith(".csv"):
try_remove(file_path)

# Delete directories after files
for dir_name in dirs:
dir_path = os.path.join(root, dir_name)
Expand All @@ -247,45 +280,48 @@ def uninstall(main_directory):
print(f"Permission error removing directory {dir_path}: {e}")
except Exception as e:
print(f"Error removing directory {dir_path}: {e}")

# Clear selenium chromedriver cache
clear_selenium_manager_cache()

# Uninstall the package
print("Uninstalling the package...")
subprocess.run([sys.executable, "-m", "pip", "uninstall", "-y", "IMDBTraktSyncer"], check=True)
subprocess.run(
[sys.executable, "-m", "pip", "uninstall", "-y", "IMDBTraktSyncer"], check=True
)

print("Uninstall complete.")


def clean_uninstall(main_directory):
"""
Deletes all folders, .zip files, .txt files in the given directory and clears selenium chromedriver cache before uninstalling.
Deletes all folders, .zip files, .txt files in the given directory and clears
selenium chromedriver cache before uninstalling.
:param main_directory: Directory path where data should be cleared.
"""
# Check if the given directory exists
if not os.path.exists(main_directory):
print(f"Error: The directory {main_directory} does not exist.")
return

# Walk through all files and folders in the directory
for root, dirs, files in os.walk(main_directory, topdown=False):
# Delete files first
for file in files:
file_path = os.path.join(root, file)

# Remove .zip files
if file.endswith(".zip"):
try_remove(file_path)

# Remove .txt files (except credentials.txt)
elif file.endswith(".txt"):
try_remove(file_path)

# Remove .csv files
elif file.endswith(".csv"):
try_remove(file_path)

# Delete directories after files
for dir_name in dirs:
dir_path = os.path.join(root, dir_name)
Expand All @@ -298,15 +334,18 @@ def clean_uninstall(main_directory):
print(f"Permission error removing directory {dir_path}: {e}")
except Exception as e:
print(f"Error removing directory {dir_path}: {e}")

# Clear selenium chromedriver cache
clear_selenium_manager_cache()

# Uninstall the package
print("Uninstalling the package...")
subprocess.run([sys.executable, "-m", "pip", "uninstall", "-y", "IMDBTraktSyncer"], check=True)
subprocess.run(
[sys.executable, "-m", "pip", "uninstall", "-y", "IMDBTraktSyncer"], check=True
)

print("Clean uninstall complete.")


def print_directory(main_directory):
print(f"Install Directory: {main_directory}")
print(f"Install Directory: {main_directory}")
Loading