su-automod/functions/check_ip_tracker.py

54 lines
1.7 KiB
Python

import aiohttp
import var
import requests
from beloved_logger import logger
from functions.roll_token import roll_token
async def check_ip_tracker():
with open('data/tracked-users.txt', 'r') as file:
reader = csv.reader(file)
user_ids = [row[0] for row in reader]
async with aiohttp.ClientSession() as session:
tasks = [fetch_data(session, user_id) for user_id in user_ids]
results = await asyncio.gather(*tasks)
for user_id, log_times in results:
all_results.append((user_id, log_times))
print("Results stored in memory.")
print(all_results)
async def get_user_ip(user_id):
await roll_token()
url = f"https://sketchersunited.org/admin/ip_tracker?profile_id={user_id}"
cookies = {
var.cookie_name: var.cookie_value,
'sketchers_united_session': var.session,
'XSRF-TOKEN': var.xsrf_token
}
headers = {
'User-Agent': 'automod',
'Accept': 'application/json, */*'
}
async with aiohttp.ClientSession() as session:
try:
async with session.get(url, cookies=cookies, headers=headers) as response:
if response.status != 200:
logger.info(f"Failed to fetch data for user {user_id}: {response.status}")
return None
data = await response.json()
logger.info(f"Data for user {user_id}: {data}")
return data
except Exception as e:
logger.error(f"Error fetching data for user {user_id}: {e}")
return None