utils.py 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. """统一工具函数"""
  2. import uuid
  3. import random
  4. import string
  5. from datetime import datetime
  6. from typing import Optional
  7. def generate_uuid() -> str:
  8. """生成UUID"""
  9. return str(uuid.uuid4()).replace("-", "").upper()
  10. def generate_id(prefix: str = "") -> str:
  11. """生成带前缀的ID"""
  12. uid = str(uuid.uuid4()).replace("-", "").upper()
  13. return f"{prefix}{uid}" if prefix else uid
  14. def generate_random_password(length: int = 8) -> str:
  15. """生成随机密码"""
  16. chars = string.ascii_letters + string.digits
  17. return ''.join(random.choice(chars) for _ in range(length))
  18. def get_current_time() -> datetime:
  19. """获取当前时间"""
  20. return datetime.now()
  21. def format_datetime(dt: Optional[datetime]) -> Optional[str]:
  22. """格式化时间为ISO 8601格式"""
  23. if dt is None:
  24. return None
  25. return dt.strftime("%Y-%m-%d %H:%M:%S")
  26. def parse_datetime(dt_str: str) -> Optional[datetime]:
  27. """解析ISO 8601格式时间"""
  28. if not dt_str:
  29. return None
  30. try:
  31. return datetime.fromisoformat(dt_str.replace("Z", "+00:00"))
  32. except (ValueError, AttributeError):
  33. return None
  34. def get_file_size_display(size_bytes: int) -> str:
  35. """将字节数转换为可读的文件大小"""
  36. if size_bytes is None:
  37. return "0 B"
  38. for unit in ['B', 'KB', 'MB', 'GB', 'TB']:
  39. if size_bytes < 1024.0:
  40. return f"{size_bytes:.2f} {unit}"
  41. size_bytes /= 1024.0
  42. return f"{size_bytes:.2f} PB"