date.py 757 B

123456789101112131415161718192021222324252627282930313233
  1. """
  2. @author: luojunhui
  3. """
  4. from datetime import datetime, timedelta
  5. def five_days_before(ori_dt):
  6. """
  7. Generate date in 3 days
  8. :param ori_dt:
  9. :return:
  10. """
  11. now_date = datetime.strptime(ori_dt, "%Y%m%d")
  12. seven_before = now_date - timedelta(days=5)
  13. return seven_before.strftime("%Y%m%d")
  14. def generate_daily_strings(start_date, end_date):
  15. """
  16. Generate daily date_str
  17. :param start_date:
  18. :param end_date:
  19. :return:
  20. """
  21. start = datetime.strptime(start_date, "%Y%m%d")
  22. end = datetime.strptime(end_date, "%Y%m%d")
  23. current = start
  24. date_strings = []
  25. while current <= end:
  26. date_strings.append(current.strftime("%Y%m%d"))
  27. current += timedelta(days=1)
  28. return date_strings