Format date

utils/__init__.py

@@ -1,2 +1,3 @@
 from .embed_to_text import embed_to_text
 from .is_mobile import is_mobile
+from .format_date import format_date

utils/format_date.py

@@ -0,0 +1,26 @@
+import datetime
+import re
+
+KST = datetime.timezone(datetime.timedelta(hours=9))
+
+def format_date(argument):
+    current_date = datetime.datetime.now(tz=KST)
+    if not argument:
+        date = ( current_date.year, current_date.month, current_date.day )
+    elif argument == '내일':
+        tomorrow = current_date + datetime.timedelta(days=1)
+        date = ( tomorrow.year, tomorrow.month, tomorrow.day )
+    elif argument == '어제':
+        yesterday = current_date - datetime.timedelta(days=1)
+        date = ( yesterday.year, yesterday.month, yesterday.day )
+    elif argument == '모레':
+        day_after_tomorrow = current_date + datetime.timedelta(days=2)
+        date = ( day_after_tomorrow.year, day_after_tomorrow.month, day_after_tomorrow.day )
+    else: 
+        date = re.match(r'^(?:(?P<YY>\d{4}|\d{2})[| ])?(?P<MM>\d{2})[| ](?P<DD>\d{2})|$', argument) or re.match(r'^(?:(?P<YY>\d{4}|\d{2})년[| ])?(?P<MM>\d{1,2})월[| ](?P<DD>\d{1,2})일$', argument)
+
+    if not date:
+        return None
+
+    date = date.groups()
+    return f"{current_date.year if not date[0] else date[0]}{'0' + date[1] if len(date[1]) == 1 else date[1]}{'0' + date[2] if len(date[2]) == 1 else date[2]}"