1. import argparse
  2. import re
  3. from datetime import datetime
  5. def tokenize_date(date_string):
  6. """
  7. Tokenizes a date string into year, month, and day components.
  8. Handles various date formats.
  9. """
  10. # Define regular expressions for common date formats
  11. patterns = [
  12. r"(\d{4})-(\d{2})-(\d{2})", # YYYY-MM-DD
  13. r"(\d{2})/(\d{2})/(\d{4})", # MM/DD/YYYY
  14. r"(\d{2})-(\d{2})-(\d{4})", # MM-DD-YYYY
  15. r"(\d{4})/(\d{2})/(\d{2})", # YYYY/MM/DD
  16. r"(\d{2} months? ago)", #e.g., 2 months ago
  17. r"(\d+) months? ago", #e.g., 2 months ago
  18. r"today",
  19. r"yesterday",
  20. r"tomorrow"
  21. ]
  23. for pattern in patterns:
  24. match = re.search(pattern, date_string, re.IGNORECASE)
  25. if match:
  26. try:
  27. if pattern == r"(\d{4})-(\d{2})-(\d{2})":
  28. year, month, day = map(int, match.groups())
  29. return year, month, day
  30. elif pattern == r"(\d{2})/(\d{2})/(\d{4})":
  31. month, day, year = map(int, match.groups())
  32. return year, month, day
  33. elif pattern == r"(\d{2})-(\d{2})-(\d{4})":
  34. month, day, year = map(int, match.groups())
  35. return year, month, day
  36. elif pattern == r"(\d{4})/(\d{2})/(\d{2})":
  37. year, month, day = map(int, match.groups())
  38. return year, month, day
  39. elif pattern == r"(\d{2} months? ago)":
  40. num_months = int(match.group(1))
  41. today = datetime.now()
  42. past_date = today - datetime.timedelta(days=num_months * 30) #Approximate
  43. return past_date.year, past_date.month, past_date.day
  44. elif pattern == r"(\d+) months? ago":
  45. num_months = int(match.group(1))
  46. today = datetime.now()
  47. past_date = today - datetime.timedelta(days=num_months * 30) #Approximate
  48. return past_date.year, past_date.month, past_date.day
  49. elif pattern == r"today":
  50. return datetime.now().year, datetime.now().month, datetime.now().day
  51. elif pattern == r"yesterday":
  52. yesterday = datetime.now() - datetime.timedelta(days=1)
  53. return yesterday.year, yesterday.month, yesterday.day
  54. elif pattern == r"tomorrow":
  55. tomorrow = datetime.now() + datetime.timedelta(days=1)
  56. return tomorrow.year, tomorrow.month, tomorrow.day
  57. except ValueError:
  58. return None # Handle cases where conversion to int fails
  60. return None # No match found
  64. def main():
  65. """
  66. CLI interface for date tokenization.
  67. """
  68. parser = argparse.ArgumentParser(description="Tokenize date strings.")
  69. parser.add_argument("date_string", help="The date string to tokenize.")
  70. args = parser.parse_args()
  72. token = tokenize_date(args.date_string)
  74. if token:
  75. print(f"Year: {token[0]}, Month: {token[1]}, Day: {token[2]}")
  76. else:
  77. print("Invalid date string.")
  80. if __name__ == "__main__":
  81. main()

Add your comment