diff --git a/dnet-stats.py b/dnet-stats.py index 342d6ed..e5a4f15 100644 --- a/dnet-stats.py +++ b/dnet-stats.py @@ -1,12 +1,14 @@ #!/usr/bin/python3 # Description: A command-line interface to https://stats.distributed.net -# Usage: python3 dnet-stats.py -Project -User +# Usage: python3 dnet-stats.py -p -u # Author: Justin Oros # Source: https://github.com/JustinOros -# Dependencies: pip install requests beautifulsoup4 lxml +# Dependencies: pip install requests argparse beautifulsoup4 lxml from bs4 import BeautifulSoup -import requests, sys +import requests +import argparse +import sys # Distributed.net Participant Search Form searchUrl = 'https://stats.distributed.net/participant/psearch.php' @@ -23,118 +25,73 @@ validProjects = { 'OGR-28': 28, } -# Text Colors & Formatting -class Text: - Pink = '\033[95m' - Black = '\u001b[30m' - Red = '\u001b[31m' - Green = '\u001b[32m' - Yellow = '\u001b[33m' - Blue = '\u001b[34m' - Magenta = '\u001b[35m' - Cyan = '\u001b[36m' - White = '\u001b[37m' - Bold = '\033[1m' - Italics = '\x1B[3m' - Underline = '\033[4m' - Reset = '\u001b[0m' - -# Text Color Theme -class Color: - Theme1 = '\u001b[38;5;165m' - Theme2 = '\u001b[38;5;164m' - Theme3 = '\u001b[38;5;163m' - Theme4 = '\u001b[38;5;162m' - Theme5 = '\u001b[38;5;161m' - -# Help Menu (Hint) -def helpHint(): - if len(sys.argv) > 1: - if sys.argv[1].lower() == 'help': - helpMenu() - print(Text.Green + '\nType ' + Text.Bold + Text.Italics + 'dnet-stats.py help' + Text.Reset + Text.Green + ' for the help menu.\n' + Text.Reset) - exit() - -# Help Menu (Main) -def helpMenu(): - print( - Text.Bold + Text.Green + '\nHelp Menu:\n' + - Text.Bold + Text.Green + '\nUsage: ' + - Text.Reset + Text.Green + 'dnet-stats.py -Project -User \n' + - Text.Bold + Text.Green + '\nProjects: ' + - Text.Reset + Text.Green + 'RC5-56, RC5-64, RC5-72, OGR-24, OGR-25, OGR-26, OGR-27, OGR-28\n' + - Text.Bold + Text.Green + '\nExample: ' + - Text.Reset + Text.Green + 'python3 dnet-stats.py -Project RC5-72 -User bluecat9@penguinized.net\n' + Text.Reset - ) - exit() - # Parse command-line arguments -user = None -project = None +def parse_arguments(): + parser = argparse.ArgumentParser(description='Fetch Distributed.net stats for user and project.') + parser.add_argument('-u', '--user', type=str, required=True, help='Username of the participant.') + parser.add_argument('-p', '--project', type=str, required=True, choices=validProjects.keys()) + args = parser.parse_args() -# Check if the arguments are provided and valid -for i in range(1, len(sys.argv)): - if sys.argv[i].lower() == "-project" and i + 1 < len(sys.argv): - project = sys.argv[i + 1].upper() - if sys.argv[i].lower() == "-user" and i + 1 < len(sys.argv): - user = sys.argv[i + 1] + return parser.parse_args() -# Ensure both project and user are present -if not user or not project: - print(Text.Bold + Text.Red + "Error: Missing required arguments. Use -Project and -User." + Text.Reset) - helpMenu() +# Main execution function +def main(): + # Parse the command-line arguments + args = parse_arguments() -# Ensure the project is valid -if project not in validProjects: - print(Text.Bold + Text.Red + f"Error: {project} is not a valid project.\n" + Text.Reset) - helpMenu() + # Retrieve user and project from parsed arguments + user = args.user + project = args.project -# Get project ID -projectId = validProjects[project] + # Get project ID + projectId = validProjects[project] -# Prepare data for the request -data = {'project_id': projectId, 'st': user} + # Prepare data for the request + data = {'project_id': projectId, 'st': user} -# Perform a POST request to fetch the response -response = requests.post(searchUrl, data=data) + # Perform a POST request to fetch the response + response = requests.post(searchUrl, data=data) -if response: # Proceed if we received a response - soup = BeautifulSoup(response.text, 'lxml') # Load the response into BeautifulSoup + if response: # Proceed if we received a response + soup = BeautifulSoup(response.text, 'lxml') # Load the response into BeautifulSoup - summary = soup.find('td', class_='htitle').text.lstrip() # Find summary + summary = soup.find('td', class_='htitle').text.lstrip() # Find summary + + if "Summary" in summary: + print(f'\nUser: {user}') # Print user + summary = " ".join(summary.split()) # Remove extra spaces from summary + summary = summary.split('/') # Split summary and project into 2 strings + project_name = summary[0] # Store the project name from the summary into a variable + print(f'Project: {project_name}') # Print project + else: + print(f"\nError: {user} not found for project {project}.\n") # Notify if user not found + sys.exit() + + line = 0 + for match in soup.find_all('td', align='right'): # Search soup for table data + line += 1 + if line == 1: # Get overall rank + overallRank = match.text.lstrip() + if overallRank[0] != "T" and overallRank[0] != "0": + overallRank = overallRank.split('(') + overallRank = overallRank[0] + if line == 2: # Get current rank + currentRank = match.text.lstrip() + if currentRank[0] != "0": + currentRank = currentRank.split('(') + currentRank = currentRank[0] + if line == 3: # Print current and overall rank + print(f'Rank: {currentRank}') + print(f'Overall: {overallRank}') + break + + # Get last update date + lastUpdate = soup.find('td', class_="lastupdate").text.split() + lastUpdate = lastUpdate[8].lstrip() + print(f'Updated: {lastUpdate}\n') - if "Summary" in summary: - print(Text.Bold + Color.Theme1 + '\nUser: ' + user + Text.Reset) # Print user - summary = " ".join(summary.split()) # Remove extra spaces from summary - summary = summary.split('/') # Split summary and project into 2 strings - project_name = summary[0] # Store the project name from the summary into a variable - print(Text.Bold + Color.Theme2 + 'Project: ' + project_name) # Print project else: - print('\n' + Text.Bold + Text.Red + 'Error: ' + user + ' not found for project ' + project + '.\n' + Text.Reset) # Notify if user not found - exit() + print('An error has occurred while fetching data.') - line = 0 - for match in soup.find_all('td', align='right'): # Search soup for table data - line += 1 - if line == 1: # Get overall rank - overallRank = match.text.lstrip() - if overallRank[0] != "T" and overallRank[0] != "0": - overallRank = overallRank.split('(') - overallRank = overallRank[0] - if line == 2: # Get current rank - currentRank = match.text.lstrip() - if currentRank[0] != "0": - currentRank = currentRank.split('(') - currentRank = currentRank[0] - if line == 3: # Print current and overall rank - print(Text.Bold + Color.Theme3 + 'Rank: ' + Text.Underline + currentRank + Text.Reset) - print(Text.Bold + Color.Theme4 + 'Overall: ' + overallRank + Text.Reset) - break - - # Get last update date - lastUpdate = soup.find('td', class_="lastupdate").text.split() - lastUpdate = lastUpdate[8].lstrip() - print(Text.Bold + Color.Theme5 + 'Updated: ' + lastUpdate + Text.Reset + '\n') - -else: - print(Text.Bold + Text.Red + 'An error has occurred while fetching data.' + Text.Reset) +if __name__ == '__main__': + main()