123 lines
3.4 KiB
Python
123 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
发送带抄送的邮件（自动获取发件地址、校验参数）
|
|
|
|
用法:
|
|
python mail_send_with_cc.py \
|
|
--to colleague@company.com \
|
|
--cc boss@company.com,team@company.com \
|
|
--subject "周报" \
|
|
--body "本周完成任务A和任务B"
|
|
|
|
python mail_send_with_cc.py --dry-run \
|
|
--to a@b.com --subject "test" --body "hello"
|
|
"""
|
|
|
|
import sys
|
|
import json
|
|
import subprocess
|
|
import re
|
|
import argparse
|
|
from typing import List, Any, Optional
|
|
|
|
EMAIL_PATTERN = re.compile(
|
|
r'^[a-zA-Z0-9._%+\-]+@[a-zA-Z0-9.\-]+\.[a-zA-Z]{2,}$'
|
|
)
|
|
|
|
|
|
def run_dws(
|
|
args: List[str], dry_run: bool = False,
|
|
) -> Optional[Any]:
|
|
cmd = ['dws'] + args
|
|
if dry_run:
|
|
print(f"[dry-run] {' '.join(cmd)}")
|
|
return {'dry_run': True}
|
|
try:
|
|
result = subprocess.run(
|
|
cmd, capture_output=True, text=True, timeout=60
|
|
)
|
|
if result.returncode != 0:
|
|
print(f"错误：{result.stderr.strip()}", file=sys.stderr)
|
|
return None
|
|
return json.loads(result.stdout)
|
|
except (subprocess.TimeoutExpired, json.JSONDecodeError,
|
|
FileNotFoundError) as e:
|
|
print(f"错误：{e}", file=sys.stderr)
|
|
return None
|
|
|
|
|
|
def validate_emails(emails_str: str) -> bool:
|
|
for email in emails_str.split(','):
|
|
email = email.strip()
|
|
if not EMAIL_PATTERN.match(email):
|
|
print(f"错误：无效邮箱地址 '{email}'")
|
|
return False
|
|
return True
|
|
|
|
|
|
def get_my_email(dry_run: bool = False) -> Optional[str]:
|
|
data = run_dws([
|
|
'mail', 'mailbox', 'list', '--format', 'json',
|
|
], dry_run=dry_run)
|
|
if dry_run:
|
|
return '<MY_EMAIL>'
|
|
if not data:
|
|
return None
|
|
if isinstance(data, list) and data:
|
|
item = data[0]
|
|
return (item.get('email') or item.get('address')
|
|
if isinstance(item, dict) else str(item))
|
|
if isinstance(data, dict):
|
|
return data.get('email') or data.get('address')
|
|
return None
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
description='发送带抄送的邮件'
|
|
)
|
|
parser.add_argument('--to', required=True, help='收件人')
|
|
parser.add_argument('--cc', default='', help='抄送人')
|
|
parser.add_argument('--subject', required=True, help='标题')
|
|
parser.add_argument('--body', required=True, help='正文')
|
|
parser.add_argument('--dry-run', action='store_true')
|
|
args = parser.parse_args()
|
|
|
|
if not validate_emails(args.to):
|
|
sys.exit(1)
|
|
if args.cc and not validate_emails(args.cc):
|
|
sys.exit(1)
|
|
|
|
print('📬 获取发件邮箱...')
|
|
from_email = get_my_email(dry_run=args.dry_run)
|
|
if not from_email and not args.dry_run:
|
|
print('错误：无法获取发件邮箱')
|
|
sys.exit(1)
|
|
|
|
cmd_args = [
|
|
'mail', 'message', 'send',
|
|
'--from', from_email or '<MY_EMAIL>',
|
|
'--to', args.to,
|
|
'--subject', args.subject,
|
|
'--body', args.body,
|
|
'--format', 'json',
|
|
]
|
|
if args.cc:
|
|
cmd_args.extend(['--cc', args.cc])
|
|
|
|
print('📤 发送邮件...')
|
|
result = run_dws(cmd_args, dry_run=args.dry_run)
|
|
if result:
|
|
print(f" ✓ 邮件已发送")
|
|
print(f" 收件人: {args.to}")
|
|
if args.cc:
|
|
print(f" 抄送: {args.cc}")
|
|
print(f" 主题: {args.subject}")
|
|
else:
|
|
print(' ✗ 发送失败')
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|