110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
#!/usr/bin/python3
|
|
# Description: A command-line interface to https://stats.distributed.net
|
|
# Usage: python3 dnet-user-stats.py -p <project> -u <username>
|
|
# Author: Justin Oros
|
|
# Source: https://github.com/JustinOros
|
|
# Dependencies: pip install requests argparse beautifulsoup4 lxml
|
|
|
|
from bs4 import BeautifulSoup
|
|
import requests, argparse, sys
|
|
|
|
# Distributed.net Participant Search Form
|
|
searchUrl = 'https://stats.distributed.net/participant/psearch.php'
|
|
|
|
# Distributed.net Projects (and associated IDs)
|
|
validProjects = {
|
|
'RC5-56': 3,
|
|
'RC5-64': 5,
|
|
'RC5-72': 8,
|
|
'OGR-24': 24,
|
|
'OGR-25': 25,
|
|
'OGR-26': 26,
|
|
'OGR-27': 27,
|
|
'OGR-28': 28,
|
|
}
|
|
|
|
# Normalize valid project names for case-insensitive lookup
|
|
normalizedProjects = {k.lower(): v for k, v in validProjects.items()}
|
|
|
|
# Parse command-line arguments
|
|
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, help='Project name (case-insensitive).')
|
|
return parser.parse_args()
|
|
|
|
# Main execution function
|
|
def main():
|
|
# Parse the command-line arguments
|
|
args = parse_arguments()
|
|
|
|
# Retrieve user and project from parsed arguments
|
|
user = args.user
|
|
project_input = args.project.strip().lower()
|
|
|
|
# Validate project name (case-insensitive)
|
|
if project_input not in normalizedProjects:
|
|
print(f"\nError: Invalid project '{args.project}'. Valid options are:")
|
|
for p in validProjects.keys():
|
|
print(f" - {p}")
|
|
sys.exit(1)
|
|
|
|
# Get project ID
|
|
projectId = normalizedProjects[project_input]
|
|
project = [k for k, v in validProjects.items() if v == projectId][0] # Get the canonical name
|
|
|
|
# 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)
|
|
|
|
if response:
|
|
soup = BeautifulSoup(response.text, 'lxml')
|
|
|
|
summary = soup.find('td', class_='htitle').text.lstrip()
|
|
|
|
if 'Summary' in summary:
|
|
print(f'\nUser: {user}')
|
|
summary = ' '.join(summary.split())
|
|
summary = summary.split('/')
|
|
project_name = summary[0]
|
|
print(f'Project: {project_name}')
|
|
else:
|
|
print(f'\nError: {user} not found for project {project}.\n')
|
|
sys.exit()
|
|
|
|
line = 0
|
|
for match in soup.find_all('td', align='right'):
|
|
line += 1
|
|
if line == 1:
|
|
overallRank = match.text.lstrip()
|
|
if overallRank[0] != 'T' and overallRank[0] != '0':
|
|
overallRank = overallRank.split('(')[0]
|
|
if line == 2:
|
|
currentRank = match.text.lstrip()
|
|
if currentRank[0] != '0':
|
|
currentRank = currentRank.split('(')[0]
|
|
if line == 3:
|
|
currentRank = f'{int(currentRank):,}'
|
|
print(f'Rank: {currentRank}')
|
|
overallRank = f'{int(overallRank):,}'
|
|
print(f'Overall: {overallRank}')
|
|
break
|
|
|
|
lastUpdate = soup.find('td', class_='lastupdate').text.split()
|
|
lastUpdate = lastUpdate[8].lstrip()
|
|
|
|
Year = lastUpdate[7:12]
|
|
Month = lastUpdate[3:6]
|
|
Day = lastUpdate[0:2]
|
|
|
|
formattedDate = f'{Month} {Day}, {Year}'
|
|
print(f'Updated: {formattedDate}\n')
|
|
|
|
else:
|
|
print('An error has occurred while fetching data.')
|
|
|
|
if __name__ == '__main__':
|
|
main()
|