Move posts to respective folder

newpost.py

@@ -1,7 +1,7 @@
 #!python
 
+import os
 import argparse
-import os, sys
 import datetime
 import traceback
 
@@ -29,9 +29,14 @@ def standard_time(time_str):
     return dt.isoformat()
 
 
+def get_year(time_str):
+    dt = datetime.datetime.strptime(time_str, "%Y-%m-%d")
+    return dt.year
+
+
 def create_post(filepath, args):
     with open(filepath, 'w') as f:
-		f.write('---\n');
+        f.write('---\n')
 
         for key in DEFAULT_FRONT_MATTER.keys():
             if key in ('title', 'slug'):
@@ -52,12 +57,12 @@ def create_post(filepath, args):
                 else:
                     f.write(f'{key}: {DEFAULT_FRONT_MATTER[key]}\n')
 
-		f.write('---\n\n');
+        f.write('---\n\n')
 
 
 def mkdir(dirpath):
     if os.path.isdir(dirpath) is False:
-		os.mkdir(dirpath)
+        os.makedirs(dirpath)
 
 
 def main(args):
@@ -66,23 +71,22 @@ def main(args):
         return
 
     os.chdir('content/posts')
+    dir_name = os.path.join(str(get_year(args.postdate)), args.posttitle)
     try:
-		mkdir(args.posttitle)
+        mkdir(dir_name)
-		mkdir(os.path.join(args.posttitle, 'images'))
+        mkdir(os.path.join(dir_name, 'images'))
+
+        os.chdir(dir_name)
+        create_post('index.md', args)
     except:
         print(traceback.format_exc())
-		print(f"Cannot create folder: {args.posttitle}")
+        print(f"Cannot create folder: {dir_name}")
-
-	os.chdir(args.posttitle)
-	create_post('index.md', args)
-
-
 
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument("-t", "--posttitle", default=None, help="The title of post.")
-	parser.add_argument("-d", "--postdate", default=None, help="The date of post.")
+    parser.add_argument("-d", "--postdate", default=datetime.datetime.now().strftime('%Y-%m-%d'), help="The date of post.")
 
     args = parser.parse_args()
     main(args)