generated from nhcarrigan/template
feat: cohort scripts
CI / dependency-pin-check-typescript (pull_request) Failing after 4s
CI / typescript (pull_request) Has been skipped
CI / dependency-pin-check-python (pull_request) Failing after 4s
CI / python (pull_request) Has been skipped
Security Scan and Upload / Security & DefectDojo Upload (pull_request) Successful in 1m23s
CI / dependency-pin-check-typescript (pull_request) Failing after 4s
CI / typescript (pull_request) Has been skipped
CI / dependency-pin-check-python (pull_request) Failing after 4s
CI / python (pull_request) Has been skipped
Security Scan and Upload / Security & DefectDojo Upload (pull_request) Successful in 1m23s
This commit is contained in:
@@ -0,0 +1,108 @@
|
||||
import json
|
||||
import os
|
||||
import time
|
||||
import urllib.request
|
||||
import urllib.error
|
||||
|
||||
# Configuration
|
||||
BOT_TOKEN = os.environ["DISCORD_BOT_TOKEN"]
|
||||
GUILD_ID = "692816967895220344"
|
||||
BASE_URL = "https://discord.com/api/v10"
|
||||
|
||||
# Read Discord IDs from table.md
|
||||
with open("table.md", "r") as f:
|
||||
content = f.read()
|
||||
|
||||
lines = content.strip().split("\n")
|
||||
|
||||
# Find the table header line (starts with |)
|
||||
header_line = None
|
||||
header_idx = 0
|
||||
for i, line in enumerate(lines):
|
||||
if line.startswith("| Discord"):
|
||||
header_line = line
|
||||
header_idx = i
|
||||
break
|
||||
|
||||
if not header_line:
|
||||
print("Could not find table header!")
|
||||
exit(1)
|
||||
|
||||
headers = [h.strip() for h in header_line.split("|")[1:-1]]
|
||||
|
||||
discord_idx = 0 # Discord ID is the first column
|
||||
|
||||
discord_ids = []
|
||||
for line in lines[header_idx + 2:]: # Skip header and separator
|
||||
if not line.startswith("|"):
|
||||
continue
|
||||
cols = [c.strip() for c in line.split("|")[1:-1]]
|
||||
if len(cols) > discord_idx:
|
||||
discord_id = cols[discord_idx].strip()
|
||||
if discord_id and discord_id.isdigit():
|
||||
discord_ids.append(discord_id)
|
||||
|
||||
print(f"Found {len(discord_ids)} Discord IDs to verify")
|
||||
|
||||
# Verify each ID against the guild
|
||||
verified = []
|
||||
missing = []
|
||||
errors = []
|
||||
|
||||
for i, discord_id in enumerate(discord_ids):
|
||||
url = f"{BASE_URL}/guilds/{GUILD_ID}/members/{discord_id}"
|
||||
req = urllib.request.Request(url)
|
||||
req.add_header("Authorization", f"Bot {BOT_TOKEN}")
|
||||
|
||||
try:
|
||||
response = urllib.request.urlopen(req)
|
||||
data = json.loads(response.read().decode())
|
||||
username = data.get("user", {}).get("username", "Unknown")
|
||||
verified.append((discord_id, username))
|
||||
print(f"[{i+1}/{len(discord_ids)}] ✓ {discord_id} - {username}")
|
||||
except urllib.error.HTTPError as e:
|
||||
if e.code == 404:
|
||||
missing.append(discord_id)
|
||||
print(f"[{i+1}/{len(discord_ids)}] ✗ {discord_id} - NOT IN SERVER")
|
||||
elif e.code == 429:
|
||||
# Rate limited - wait and retry
|
||||
retry_after = json.loads(e.read().decode()).get("retry_after", 1)
|
||||
print(f"[{i+1}/{len(discord_ids)}] Rate limited, waiting {retry_after}s...")
|
||||
time.sleep(retry_after + 0.5)
|
||||
# Retry
|
||||
try:
|
||||
req2 = urllib.request.Request(url)
|
||||
req2.add_header("Authorization", f"Bot {BOT_TOKEN}")
|
||||
response = urllib.request.urlopen(req2)
|
||||
data = json.loads(response.read().decode())
|
||||
username = data.get("user", {}).get("username", "Unknown")
|
||||
verified.append((discord_id, username))
|
||||
print(f"[{i+1}/{len(discord_ids)}] ✓ {discord_id} - {username} (after retry)")
|
||||
except urllib.error.HTTPError as e2:
|
||||
if e2.code == 404:
|
||||
missing.append(discord_id)
|
||||
print(f"[{i+1}/{len(discord_ids)}] ✗ {discord_id} - NOT IN SERVER (after retry)")
|
||||
else:
|
||||
errors.append((discord_id, f"HTTP {e2.code}"))
|
||||
print(f"[{i+1}/{len(discord_ids)}] ? {discord_id} - Error {e2.code}")
|
||||
else:
|
||||
errors.append((discord_id, f"HTTP {e.code}"))
|
||||
print(f"[{i+1}/{len(discord_ids)}] ? {discord_id} - Error {e.code}")
|
||||
|
||||
# Small delay to avoid rate limits
|
||||
time.sleep(0.1)
|
||||
|
||||
print(f"\n=== SUMMARY ===")
|
||||
print(f"Verified: {len(verified)}")
|
||||
print(f"Missing: {len(missing)}")
|
||||
print(f"Errors: {len(errors)}")
|
||||
|
||||
# Save results
|
||||
with open("discord_verification.json", "w") as f:
|
||||
json.dump({
|
||||
"verified": verified,
|
||||
"missing": missing,
|
||||
"errors": errors
|
||||
}, f, indent=2)
|
||||
|
||||
print("\nResults saved to discord_verification.json")
|
||||
Reference in New Issue
Block a user