dan92 commited on
Commit
0ce219a
·
verified ·
1 Parent(s): d2dab19

Upload register_bot.py

Browse files
Files changed (1) hide show
  1. register_bot.py +14 -23
register_bot.py CHANGED
@@ -16,26 +16,16 @@ from urllib.parse import quote, urlparse, parse_qs
16
  import concurrent.futures
17
  import logging
18
 
19
- # 获取当前脚本所在目录
20
- SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
21
-
22
- # 配置日志
23
- try:
24
- LOG_FILE = os.path.join(SCRIPT_DIR, 'registration_log.txt')
25
- logging.basicConfig(level=logging.INFO,
26
- format='%(asctime)s - %(levelname)s - %(message)s',
27
- filename=LOG_FILE,
28
- filemode='a')
29
- except PermissionError:
30
- # 如果无法写入文件,使用控制台日志
31
- logging.basicConfig(level=logging.INFO,
32
- format='%(asctime)s - %(levelname)s - %(message)s',
33
- stream=sys.stdout)
34
- logging.warning("Could not write to log file. Logging to console instead.")
35
 
36
  # 获取环境变量
37
- PASTE_API_URL = os.getenv('PASTE_API_URL')
38
- PASTE_API_PASSWORD = os.getenv('PASTE_API_PASSWORD')
39
 
40
  # 构建上传 URL
41
  UPLOAD_URL = PASTE_API_URL.replace('/api/paste/', '/api/admin/paste/') + '/content'
@@ -241,13 +231,13 @@ def check_email(email):
241
 
242
  def register_and_verify(num_accounts=1):
243
  successful_accounts = []
244
- with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor: # 降低并发数
245
  futures = []
246
  for _ in range(num_accounts):
247
  email = get_temp_email()
248
  if email:
249
  futures.append(executor.submit(process_account, email))
250
- time.sleep(RETRY_DELAY) # 添加延迟
251
 
252
  for future in concurrent.futures.as_completed(futures):
253
  result = future.result()
@@ -272,10 +262,11 @@ def process_account(email):
272
  def main():
273
  num_accounts = 5 # 可以根据需要修改注册数量
274
  successful_accounts = register_and_verify(num_accounts)
275
- logging.info(f"Successfully registered {len(successful_accounts)} accounts:")
 
 
276
  for account in successful_accounts:
277
- logging.info(f"Email: {account['email']}, Password: {account['password']}")
278
- print(f"Successfully registered {len(successful_accounts)} accounts. Check registration_log.txt for details.")
279
 
280
  if __name__ == "__main__":
281
  main()
 
16
  import concurrent.futures
17
  import logging
18
 
19
+ # 配置日志(仅控制台输出)
20
+ logging.basicConfig(
21
+ level=logging.INFO,
22
+ format='%(asctime)s - %(levelname)s - %(message)s',
23
+ stream=sys.stdout
24
+ )
 
 
 
 
 
 
 
 
 
 
25
 
26
  # 获取环境变量
27
+ PASTE_API_URL = os.getenv('PASTE_API_URL', "https://page.zhoudan.icu/api/paste/b40v96oX")
28
+ PASTE_API_PASSWORD = os.getenv('PASTE_API_PASSWORD', '321')
29
 
30
  # 构建上传 URL
31
  UPLOAD_URL = PASTE_API_URL.replace('/api/paste/', '/api/admin/paste/') + '/content'
 
231
 
232
  def register_and_verify(num_accounts=1):
233
  successful_accounts = []
234
+ with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
235
  futures = []
236
  for _ in range(num_accounts):
237
  email = get_temp_email()
238
  if email:
239
  futures.append(executor.submit(process_account, email))
240
+ time.sleep(RETRY_DELAY)
241
 
242
  for future in concurrent.futures.as_completed(futures):
243
  result = future.result()
 
262
  def main():
263
  num_accounts = 5 # 可以根据需要修改注册数量
264
  successful_accounts = register_and_verify(num_accounts)
265
+
266
+ # 仅输出到控制台,不写入文件
267
+ print(f"Successfully registered {len(successful_accounts)} accounts")
268
  for account in successful_accounts:
269
+ print(f"Email: {account['email']}, Password: {account['password']}")
 
270
 
271
  if __name__ == "__main__":
272
  main()