Add dry-run mode to 2925 pool picker

This commit is contained in:
2026-05-11 16:28:34 +08:00
parent 228d650ca8
commit 7df33e2611
2 changed files with 6 additions and 4 deletions
+4 -3
View File
@@ -96,8 +96,9 @@ def cmd_list(args):
def cmd_pick(args):
a,state=pick_account(args.preferred, args.exclude or [])
alias=generate_alias_for(a)
mark(a['email'], lastUsedAt=int(time.time()), lastError='')
print(json.dumps({'email':a['email'],'alias':alias,'baseLocal':a['email'].split('@')[0],'aliasMatchesAccount':alias.startswith(a['email'].split('@')[0])}, ensure_ascii=False))
if not args.dry_run:
mark(a['email'], lastUsedAt=int(time.time()), lastError='')
print(json.dumps({'email':a['email'],'alias':alias,'baseLocal':a['email'].split('@')[0],'aliasMatchesAccount':alias.startswith(a['email'].split('@')[0]),'dryRun':bool(args.dry_run)}, ensure_ascii=False))
def cmd_limit(args):
@@ -129,7 +130,7 @@ def main():
p=argparse.ArgumentParser()
sub=p.add_subparsers(dest='cmd', required=True)
sub.add_parser('list').set_defaults(func=cmd_list)
q=sub.add_parser('pick'); q.add_argument('--preferred'); q.add_argument('--exclude', action='append'); q.set_defaults(func=cmd_pick)
q=sub.add_parser('pick'); q.add_argument('--preferred'); q.add_argument('--exclude', action='append'); q.add_argument('--dry-run', action='store_true'); q.set_defaults(func=cmd_pick)
l=sub.add_parser('mark-limit'); l.add_argument('email'); l.add_argument('--reason', default=''); l.set_defaults(func=cmd_limit)
m=sub.add_parser('mark-login'); m.add_argument('email'); m.set_defaults(func=cmd_mark_login)
sub.add_parser('selftest').set_defaults(func=cmd_selftest)