59 lines
2.5 KiB
Python
59 lines
2.5 KiB
Python
import logging, time, requests, json, os, threading
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
from datetime import datetime
|
|
|
|
from cm_bot_hal import CM_BOT_HAL
|
|
|
|
# Suppress httpx logs
|
|
logging.getLogger("httpx").setLevel(logging.WARNING)
|
|
|
|
# Create logs directory if it doesn't exist
|
|
logs_dir = "logs"
|
|
if not os.path.exists(logs_dir):
|
|
os.makedirs(logs_dir)
|
|
|
|
# api_url = 'https://api.luckytown888.net'
|
|
api_url = 'http://api-server:3000'
|
|
max_threading = 5
|
|
|
|
def transfer(data: dict, local_logger):
|
|
bot = CM_BOT_HAL()
|
|
thread_name = threading.current_thread().name
|
|
local_logger.info(f"[Thread-{thread_name}] [Start] Transfer Credit from {data['f_username']} to {data['t_username']}")
|
|
result = bot.transfer_credit_api(data['f_username'], data['f_password'], data['t_username'], data['t_password'])
|
|
local_logger.info(f"[Thread-{thread_name}] {result}")
|
|
local_logger.info(f"[Thread-{thread_name}] [Done] {data['f_username']} transfer done!")
|
|
del bot
|
|
time.sleep(1)
|
|
|
|
while True:
|
|
weekday = int(datetime.now().strftime("%w"))
|
|
hour = int(datetime.now().strftime("%H"))
|
|
minutes = int(datetime.now().strftime("%M"))
|
|
if weekday == 1 and (hour >= 6 and hour < 13):
|
|
local_logger = logging.getLogger(f"{__name__}")
|
|
|
|
# Configure file handler for logging to logs folder
|
|
log_filename = f"{logs_dir}/transfer_credit_{datetime.now().strftime('%Y%m%d_%H%M')}.log"
|
|
file_handler = logging.FileHandler(log_filename)
|
|
file_handler.setLevel(logging.INFO)
|
|
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
file_handler.setFormatter(formatter)
|
|
|
|
# Add file handler to logger if not already added
|
|
if not any(isinstance(handler, logging.FileHandler) for handler in local_logger.handlers):
|
|
local_logger.addHandler(file_handler)
|
|
local_logger.setLevel(logging.INFO)
|
|
|
|
response = requests.get(f'{api_url}/user')
|
|
items = json.loads(response.text)
|
|
total_items = len(items) if isinstance(items, list) else 0
|
|
if total_items == 0:
|
|
local_logger.info("No items to process.")
|
|
else:
|
|
local_logger.info(f"Processing {total_items} transfer items...")
|
|
with ThreadPoolExecutor(max_workers=max_threading) as executor:
|
|
results = list(executor.map(lambda item: transfer(item, local_logger), items))
|
|
local_logger.info(f"Completed processing {total_items} transfer items.")
|
|
time.sleep(5 * 60)
|