def init():
if request.method == 'GET':
if g.blog:
return redirect(url_for('dashboard'))
else:
g.blog = DEFAULT_CONFIG
return render_template('init.html')
elif request.method == 'POST':
for key in request.form:
if not request.form[key]:
flash(MSG_SETTINGS_FILL)
g.blog = request.form
return render_template('init.html')
from yak import init
init(blog_dir, request.form)
g.blog = read_config(blog_dir)
hg_init(blog_dir)
for post in publish():
source = os.path.join(blog_dir, 'publish', post)
hg_add(source)
hg_commit(source, 'new blog')
bake_blog()
flash(MSG_INIT_SUCCESS)
return redirect(url_for('dashboard'))
def edit(filename):
"""
Possible actions: [post rename, edit markdown]
"""
if request.method == 'GET':
location = get_location(filename)
if location:
with open(os.path.join(blog_dir, location, filename),
'r', 'utf-8') as f:
markdown = f.read()
if location == 'publish':
action = 'Save and Publish'
else:
action = 'Save'
edit_commits = hg_edit_commits(filename)
try:
past = edit_commits[1]['node']
except IndexError:
past = None
return render_template('edit_post.html', filename=filename,
markdown=markdown, action=action, past=past)
else:
flash(MSG_POST_NOT_FOUND.format(filename[:-3]))
return redirect(url_for('posts'))
elif request.method == 'POST':
new_filename = request.form['filename']
if not new_filename.endswith('.md'):
new_filename += '.md'
markdown = request.form['markdown']
action = request.form['action']
valid_filename = is_valid_filename(filename)
if not valid_filename:
flash(MSG_POST_FILENAME_INVALID)
elif not is_valid_post(markdown, valid_filename['published']):
flash(MSG_POST_CONTENT_INVALID)
else:
location = get_location(filename)
if not (new_filename == filename) and get_location(new_filename):
flash(MSG_POST_EXISTS)
else:
with open(os.path.join(blog_dir, location, filename),
'w', 'utf-8') as f:
f.write(markdown)
if new_filename == filename:
hg_commit(os.path.join(blog_dir, location, filename),
'edited post {}'.format(filename))
else:
hg_rename(
os.path.join(blog_dir, location, filename),
os.path.join(blog_dir, location, new_filename)
)
flash(MSG_POST_SAVED.format(new_filename[:-3]))
if 'Publish' in action:
bake_blog()
return redirect(url_for('posts'))
return render_template('edit_post.html',
filename=filename, markdown=markdown, action=action)
def new():
"""
Actions are sent as forms, and redirects are made by `referer`
"""
if request.method == 'GET':
filename, markdown = default_post()
return render_template('new_post.html',
filename=filename, markdown=markdown)
elif request.method == 'POST':
filename = request.form['filename']
if not filename.endswith('.md'):
filename += '.md'
markdown = request.form['markdown']
action = request.form['action']
valid_filename = is_valid_filename(filename)
if not valid_filename:
flash(MSG_POST_FILENAME_INVALID)
elif get_location(filename):
flash(MSG_POST_EXISTS)
elif not is_valid_post(markdown, valid_filename['published']):
flash(MSG_POST_CONTENT_INVALID)
else:
if 'Draft' in action:
dest = 'drafts'
flash(MSG_POST_SAVED.format(filename[:-3]))
elif 'Publish' in action:
dest = 'publish'
flash(MSG_POST_PUBLISHED.format(filename[:-3]))
with open(os.path.join(blog_dir, dest, filename),
'w', 'utf-8') as f:
f.write(markdown)
if 'Publish' in action:
bake_blog()
source = os.path.join(blog_dir, get_location(filename))
hg_add(source)
hg_commit(source, 'new post {} in {}'.format(filename, dest))
return redirect(url_for('posts'))
return render_template(request.form['referer'],
filename=filename, markdown=markdown)
