93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
#!python
|
|
|
|
import os
|
|
import argparse
|
|
import datetime
|
|
import traceback
|
|
|
|
|
|
DEFAULT_FRONT_MATTER = {
|
|
'slug': '',
|
|
'title': '',
|
|
'description': '',
|
|
'toc': True,
|
|
'authors': [],
|
|
'tags': [],
|
|
'categories': [],
|
|
'series': [],
|
|
'date': '',
|
|
'lastmod': '',
|
|
'featuredVideo': '',
|
|
'featuredImage': '',
|
|
'draft': False,
|
|
'enableComment': True
|
|
}
|
|
|
|
|
|
def standard_time(time_str):
|
|
dt = datetime.datetime.strptime(time_str, "%Y-%m-%d")
|
|
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')
|
|
|
|
for key in DEFAULT_FRONT_MATTER.keys():
|
|
if key in ('title', 'slug'):
|
|
f.write(f'{key}: {args.posttitle}\n')
|
|
elif key in ('date', 'lastmod'):
|
|
if args.postdate is not None:
|
|
f.write(f'{key}: {standard_time(args.postdate)}\n')
|
|
else:
|
|
f.write(f'{key}: {datetime.datetime.now().isoformat()}\n')
|
|
else:
|
|
if type(DEFAULT_FRONT_MATTER[key]) is str:
|
|
if DEFAULT_FRONT_MATTER[key] == '':
|
|
f.write(f'{key}: \n')
|
|
else:
|
|
f.write(f'{key}: {DEFAULT_FRONT_MATTER[key]}\n')
|
|
elif type(DEFAULT_FRONT_MATTER[key]) is bool:
|
|
f.write(f'{key}: {"true" if DEFAULT_FRONT_MATTER[key] else "false"}\n')
|
|
else:
|
|
f.write(f'{key}: {DEFAULT_FRONT_MATTER[key]}\n')
|
|
|
|
f.write('---\n\n')
|
|
|
|
|
|
def mkdir(dirpath):
|
|
if os.path.isdir(dirpath) is False:
|
|
os.makedirs(dirpath)
|
|
|
|
|
|
def main(args):
|
|
if args.posttitle is None:
|
|
print("Need a title!")
|
|
return
|
|
|
|
os.chdir('content/posts')
|
|
dir_name = os.path.join(str(get_year(args.postdate)), args.posttitle)
|
|
try:
|
|
mkdir(dir_name)
|
|
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: {dir_name}")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-t", "--posttitle", default=None, help="The title 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)
|