Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
initial draft of single mode (#36)
Signed-off-by: Chris Warrick <kwpolska@gmail.com>
  • Loading branch information
Kwpolska committed May 8, 2015
1 parent dcff05f commit 87e83a2
Show file tree
Hide file tree
Showing 4 changed files with 205 additions and 43 deletions.
44 changes: 44 additions & 0 deletions coil/data/templates/jinja/coil_rebuild_single.tmpl
@@ -0,0 +1,44 @@
{# -*- coding: utf-8 -*- #}
{% extends 'base.tmpl' %}
{% block content %}
<div class="page-header">
<h1 class="build-status-caption"><i class="build-status-icon fa fa-fw fa-cog"></i> Rebuild</h1>
</div>

<div class="panel panel-default">
<div class="panel-heading"><h3 class="panel-title">
<a data-toggle="collapse" href="#collapseOutput" aria-expanded="false" aria-controls="collapseOutput"><i class="fa fa-bars"></i> Advanced information</a>
</h3></div>
<div class="panel-body collapse" id="collapseOutput">
<h4>Build</h4>
<pre><code id="outputb">{{ outputb }}</code></pre>

<h4>Unused files removed</h4>
<pre><code id="outputo">{{ outputo }}</code></pre>
</div>
</div>

{% endblock %}

{% block extra_js %}
<script>
$(document).ready(function() {
fs = $('.build-status-icon');
fsc = $('.build-status-caption');
if ({{ bstatus }} == 1) {
fs.removeClass('fa-cog');
fs.addClass('fa-check');
fsc.addClass('text-success');
clearInterval(intID);
} else {
pb.addClass('progress-bar-danger');
fs.removeClass('fa-cog');
fs.addClass('fa-times');
fsc.addClass('text-danger');
$("#collapseOutput").collapse({
show: true
});
}
});
</script>
{% endblock %}
44 changes: 44 additions & 0 deletions coil/data/templates/mako/coil_rebuild_single.tmpl
@@ -0,0 +1,44 @@
## -*- coding: utf-8 -*-
<%inherit file="base.tmpl"/>
<%block name="content">
<div class="page-header">
<h1 class="build-status-caption"><i class="build-status-icon fa fa-fw fa-cog"></i> Rebuild</h1>
</div>

<div class="panel panel-default">
<div class="panel-heading"><h3 class="panel-title">
<a data-toggle="collapse" href="#collapseOutput" aria-expanded="false" aria-controls="collapseOutput"><i class="fa fa-bars"></i> Advanced information</a>
</h3></div>
<div class="panel-body collapse" id="collapseOutput">
<h4>Build</h4>
<pre><code id="outputb">${outputb}</code></pre>

<h4>Unused files removed</h4>
<pre><code id="outputo">${outputo}</code></pre>
</div>
</div>

</%block>

<%block name="extra_js">
<script>
$(document).ready(function() {
fs = $('.build-status-icon');
fsc = $('.build-status-caption');
if (${bstatus} == 1) {
fs.removeClass('fa-cog');
fs.addClass('fa-check');
fsc.addClass('text-success');
clearInterval(intID);
} else {
pb.addClass('progress-bar-danger');
fs.removeClass('fa-cog');
fs.addClass('fa-times');
fsc.addClass('text-danger');
$("#collapseOutput").collapse({
show: true
});
}
});
</script>
</%block>
28 changes: 23 additions & 5 deletions coil/tasks.py
Expand Up @@ -88,6 +88,28 @@ def orphans(dburl, sitedir):
job = get_current_job(db)
job.meta.update({'out': '', 'return': None, 'status': None})
job.save()
returncode, out = orphans_single()

job.meta.update({'out': out, 'return': returncode, 'status':
returncode == 0})
job.save()
os.chdir(oldcwd)
return returncode

def build_single(mode):
"""Build, in the single-user mode."""
if mode == 'force':
amode = ['-a']
else:
amode = []
p = subprocess.Popen([executable, '-m', 'nikola', 'build'] + amode,
stderr=subprocess.PIPE)
p.wait()
out = ''.join(p.stderr.readlines())
return (p.returncode == 0), out

def orphans_single():
"""Remove all orphans in the site, in the single user-mode."""
p = subprocess.Popen([executable, '-m', 'nikola', 'orphans'],
stdout=subprocess.PIPE)
p.wait()
Expand All @@ -97,8 +119,4 @@ def orphans(dburl, sitedir):
os.unlink(f)

out = '\n'.join(files)
job.meta.update({'out': ''.join(out), 'return': p.returncode, 'status':
p.returncode == 0})
job.save()
os.chdir(oldcwd)
return p.returncode
return p.returncode, out
132 changes: 94 additions & 38 deletions coil/web.py
Expand Up @@ -118,10 +118,14 @@ def configure_site():

app.secret_key = _site.config.get('COIL_SECRET_KEY')
app.config['COIL_URL'] = _site.config.get('COIL_URL')
app.config['COIL_SINGLE'] = _site.config.get('COIL_SINGLE', False)
app.config['REDIS_URL'] = _site.config.get('COIL_REDIS_URL',
'redis://localhost:6379/0')
db = redis.StrictRedis.from_url(app.config['REDIS_URL'])
q = rq.Queue(name='coil', connection=db)
if app.config['COIL_SINGLE']:
app.config['COIL_USERS'] = _site.config.get('COIL_USERS', {})
else:
db = redis.StrictRedis.from_url(app.config['REDIS_URL'])
q = rq.Queue(name='coil', connection=db)

_site.template_hooks['menu'].append(generate_menu)
_site.template_hooks['menu_alt'].append(generate_menu_alt)
Expand Down Expand Up @@ -177,7 +181,11 @@ def configure_site():
_site.template_system.inject_directory(tmpl_dir)

# Site proxy
site = SiteProxy(db, _site, app.logger)
if app.config['COIL_SINGLE']:
site = _site
else:
site = SiteProxy(db, _site, app.logger)

configure_url(app.config['COIL_URL'])


Expand Down Expand Up @@ -227,7 +235,11 @@ def generate_menu():
:return: HTML fragment
:rtype: str
"""
if db.get('site:needs_rebuild') not in ('0', '-1'):
if db is not None:
needs_rebuild = db.get('site:needs_rebuild')
else:
needs_rebuild = site.coil_needs_rebuild
if needs_rebuild not in ('0', '-1'):
return ('</li><li><a href="{0}"><i class="fa fa-fw '
'fa-warning"></i> <strong>Rebuild</strong></a></li>'.format(
url_for('rebuild')))
Expand All @@ -245,7 +257,7 @@ def generate_menu_alt():
if not current_user.is_authenticated():
return ('<li><a href="{0}"><i class="fa fa-fw fa-sign-in"></i> '
'Log in</a></li>'.format(url_for('login')))
if current_user.is_admin:
if db is not None and current_user.is_admin:
edit_entry = (
'<li><a href="{0}"><i class="fa fa-fw fa-users"></i> '
'Manage users</a></li>'
Expand Down Expand Up @@ -434,13 +446,20 @@ def get_user(uid):
:raises ValueError: uid is not an integer
:raises KeyError: if user does not exist
"""
d = db.hgetall('user:{0}'.format(uid))
if d:
for p in PERMISSIONS:
d[p] = d.get(p) == '1'
return User(uid=uid, **d)
if db is not None:
d = db.hgetall('user:{0}'.format(uid))
if d:
for p in PERMISSIONS:
d[p] = d.get(p) == '1'
return User(uid=uid, **d)
else:
return None
else:
return None
d = app.config['COIL_USERS'].get(uid)
if d:
return User(uid=uid, **d)
else:
return None


def find_user_by_name(username):
Expand All @@ -450,11 +469,14 @@ def find_user_by_name(username):
:return: the user
:rtype: User object or None
"""
uid = db.hget('users', username)
if uid:
return get_user(uid)
if db is not None:
uid = db.hget('users', username)
if uid:
return get_user(uid)
else:
return None
for u in app.config['COIL_USERS']:
if u['username'] == username:
return get_user(uid)


def write_user(user):
Expand Down Expand Up @@ -638,7 +660,10 @@ def edit(path):
with io.open(meta_path, 'w+', encoding='utf-8') as fh:
fh.write(write_metadata(meta))
scan_site()
db.set('site:needs_rebuild', '1')
if db is not None:
db.set('site:needs_rebuild', '1')
else:
site.coil_needs_rebuild = '1'
post = find_post(path)
context['action'] = 'save'
else:
Expand All @@ -651,11 +676,16 @@ def edit(path):

context['post'] = post
users = []
uids = db.hgetall('users').values()
for u in uids:
realname, active = db.hmget('user:{0}'.format(u), 'realname', 'active')
if active == '1':
users.append((u, realname))
if db is not None:
uids = db.hgetall('users').values()
for u in uids:
realname, active = db.hmget('user:{0}'.format(u), 'realname', 'active')
if active == '1':
users.append((u, realname))
else:
for u, d in app.config['COIL_USERS'].values():
if d['active']:
users.append((u, d['realname']))
context['users'] = sorted(users)
context['current_auid'] = current_auid
context['title'] = 'Editing {0}'.format(post.title())
Expand Down Expand Up @@ -689,14 +719,19 @@ def delete():
meta_path = os.path.splitext(path)[0] + '.meta'
os.unlink(meta_path)
scan_site()
db.set('site:needs_rebuild', '1')
if db is not None:
db.set('site:needs_rebuild', '1')
else:
site.coil_needs_rebuild = '1'
return redirect(url_for('index'))


@app.route('/api/rebuild/')
@login_required
def api_rebuild():
"""Rebuild the site (internally)."""
if db is None:
return '{"error": "single-user mode"}'
build_job = q.fetch_job('build')
orphans_job = q.fetch_job('orphans')

Expand Down Expand Up @@ -737,17 +772,25 @@ def rebuild(mode=''):
return error('You are not permitted to rebuild the site.</p>'
'<p class="lead">Contact an administartor for '
'more information.', 401)
db.set('site:needs_rebuild', '-1')
if not q.fetch_job('build') and not q.fetch_job('orphans'):
b = q.enqueue_call(func=coil.tasks.build,
args=(app.config['REDIS_URL'],
app.config['NIKOLA_ROOT'], mode), job_id='build')
q.enqueue_call(func=coil.tasks.orphans,
args=(app.config['REDIS_URL'],
app.config['NIKOLA_ROOT']), job_id='orphans',
depends_on=b)
if db is not None:
db.set('site:needs_rebuild', '-1')
if not q.fetch_job('build') and not q.fetch_job('orphans'):
b = q.enqueue_call(func=coil.tasks.build,
args=(app.config['REDIS_URL'],
app.config['NIKOLA_ROOT'], mode), job_id='build')
q.enqueue_call(func=coil.tasks.orphans,
args=(app.config['REDIS_URL'],
app.config['NIKOLA_ROOT']), job_id='orphans',
depends_on=b)
return render('coil_rebuild.tmpl', {'title': 'Rebuild'})
else:
status, outputb = coil.tasks.build_single(mode)
_, outputo = coil.tasks.orphans_single()
return render('coil_rebuild_single.tmpl', {'title': 'Rebuild',
'status': '1' if status else '0',
'outputb': outputb,
'outputo': outputo})

return render('coil_rebuild.tmpl', {'title': 'Rebuild'})


@app.route('/new/<obj>/', methods=['POST'])
Expand Down Expand Up @@ -791,7 +834,10 @@ def new(obj):
del _site.config['ADDITIONAL_METADATA']['author.uid']
# reload post list and go to index
scan_site()
db.set('site:needs_rebuild', '1')
if db is not None:
db.set('site:needs_rebuild', '1')
else:
site.coil_needs_rebuild = '1'
return redirect(url_for('index'))


Expand Down Expand Up @@ -892,8 +938,10 @@ def acp_users():
"""List all users."""
if current_user.must_change_password:
return redirect(url_for('acp_account') + '?status=pwdchange')
elif not current_user.is_admin:
if not current_user.is_admin:
return error("Not authorized to edit users.", 401)
if not db:
return error('The ACP is not available in single-user mode.', 500)

alert = ''
alert_status = ''
Expand Down Expand Up @@ -923,8 +971,10 @@ def acp_users_edit():
global current_user
if current_user.must_change_password:
return redirect(url_for('acp_account') + '?status=pwdchange')
elif not current_user.is_admin:
if not current_user.is_admin:
return error("Not authorized to edit users.", 401)
if not db:
return error('The ACP is not available in single-user mode.', 500)
data = request.form

form = UserEditForm()
Expand Down Expand Up @@ -997,8 +1047,10 @@ def acp_users_import():
"""Import users from a TSV file."""
if current_user.must_change_password:
return redirect(url_for('acp_account') + '?status=pwdchange')
elif not current_user.is_admin:
if not current_user.is_admin:
return error("Not authorized to edit users.", 401)
if not db:
return error('The ACP is not available in single-user mode.', 500)

form = UserImportForm()
if not form.validate():
Expand All @@ -1015,8 +1067,10 @@ def acp_users_delete():
"""Delete or undelete an user account."""
if current_user.must_change_password:
return redirect(url_for('acp_account') + '?status=pwdchange')
elif not current_user.is_admin:
if not current_user.is_admin:
return error("Not authorized to edit users.", 401)
if not db:
return error('The ACP is not available in single-user mode.', 500)

form = UserDeleteForm()
if not form.validate():
Expand All @@ -1040,8 +1094,10 @@ def acp_users_permissions():
"""Change user permissions."""
if current_user.must_change_password:
return redirect(url_for('acp_account') + '?status=pwdchange')
elif not current_user.is_admin:
if not current_user.is_admin:
return error("Not authorized to edit users.", 401)
if not db:
return error('The ACP is not available in single-user mode.', 500)

form = PermissionsForm()
users = []
Expand Down

0 comments on commit 87e83a2

Please sign in to comment.