Create WikiPage class to encapsulate wiki page functions

This commit is contained in:
Chase Sterling 2016-07-04 22:22:12 -04:00
parent ac79d186ec
commit d50cf54072
5 changed files with 252 additions and 242 deletions

View file

@ -31,9 +31,9 @@ def rebuild_index():
name = filename_to_cname(page['path']) name = filename_to_cname(page['path'])
# TODO add email? # TODO add email?
body = dict(name=name, body = dict(name=name,
content=page['data'], content=page.data,
message=page['info']['message'], message=page.info['message'],
username=page['info']['author'], username=page.info['author'],
updated_on=entry['mtime'], updated_on=entry['mtime'],
created_on=entry['ctime']) created_on=entry['ctime'])
search.index_wiki(name, body) search.index_wiki(name, body)

View file

@ -47,7 +47,7 @@ class SimpleSearch(BaseSearch):
# this can be None, not sure how # this can be None, not sure how
if page: if page:
res.append(dict(name=name, content=page['data'])) res.append(dict(name=name, content=page.data))
return res return res
def users(self, query): def users(self, query):
@ -129,7 +129,7 @@ class WhooshSearch(BaseSearch):
res = [] res = []
for hit in results: for hit in results:
name = hit["path"] name = hit["path"]
page_data = g.current_wiki.get_page(name)["data"].decode("utf-8") page_data = g.current_wiki.get_page(name).data.decode("utf-8")
content = hit.highlights('body', text=page_data) content = hit.highlights('body', text=page_data)
res.append(dict(name=name, content=content)) res.append(dict(name=name, content=content))

View file

@ -39,145 +39,6 @@ class Wiki(HookMixin):
def __repr__(self): def __repr__(self):
return "Wiki: %s" % self.path return "Wiki: %s" % self.path
def _get_user(self, username, email):
if not username:
username = self.default_committer_name
if not email:
email = self.default_committer_email
return username, email
def _cache_key(self, name, sha='HEAD'):
return 'page/%s[%s]' % (name, sha)
def revert_page(self, name, commit_sha, message, username, email):
"""Revert page to passed commit sha1
:param name: Name of page to revert.
:param commit_sha: Commit Sha1 to revert to.
:param message: Commit message.
:param username: Committer name.
:param email: Committer email.
:return: Git commit sha1
"""
page = self.get_page(name, commit_sha)
if not page:
raise PageNotFound('Commit not found')
if not message:
commit_info = gittle.utils.git.commit_info(self.gittle[commit_sha.encode('latin-1')])
message = commit_info['message']
return self.write_page(name, page['data'], message=message, username=username, email=email)
def write_page(self, name, content, message=None, create=False, username=None, email=None):
"""Write page to git repo
:param name: Name of page.
:param content: Content of page.
:param message: Commit message.
:param create: Perform git add operation?
:param username: Commit Name.
:param email: Commit Email.
:return: Git commit sha1.
"""
cname = to_canonical(name)
filename = cname_to_filename(cname)
dirname = posixpath.join(self.path, posixpath.dirname(filename))
if not os.path.exists(dirname):
os.makedirs(dirname)
with open(self.path + "/" + filename, 'w') as f:
f.write(content)
if create:
self.gittle.add(filename)
if not message:
message = "Updated %s" % name
username, email = self._get_user(username, email)
ret = self.gittle.commit(name=username,
email=email,
message=message,
files=[filename])
cache.delete(self._cache_key(cname))
return ret
def rename_page(self, old_name, new_name, username=None, email=None, message=None):
"""Rename page.
:param old_name: Page that will be renamed.
:param new_name: New name of page.
:param username: Committer name
:param email: Committer email
:return: str -- Commit sha1
"""
old_filename, new_filename = map(cname_to_filename, [old_name, new_name])
if old_filename not in self.gittle.index:
# old doesn't exist
return None
if old_filename == new_filename:
return
if new_filename in self.gittle.index:
# file is being overwritten, but that is ok, it's git!
pass
username, email = self._get_user(username, email)
if not message:
message = "Moved %s to %s" % (old_name, new_name)
os.rename(os.path.join(self.path, old_filename), os.path.join(self.path, new_filename))
self.gittle.add(new_filename)
self.gittle.rm(old_filename)
commit = self.gittle.commit(name=username,
email=email,
message=message,
files=[old_filename, new_filename])
cache.delete_many(self._cache_key(old_name), self._cache_key(new_name))
return commit
def delete_page(self, name, username=None, email=None, message=None):
"""Delete page.
:param name: Page that will be deleted
:param username: Committer name
:param email: Committer email
:return: str -- Commit sha1
"""
username, email = self._get_user(username, email)
if not message:
message = "Deleted %s" % name
filename = cname_to_filename(name)
# gittle.rm won't actually remove the file, have to do it ourselves
os.remove(os.path.join(self.path, filename))
self.gittle.rm(filename)
commit = self.gittle.commit(name=username,
email=email,
message=message,
files=[filename])
cache.delete_many(self._cache_key(name))
return commit
def get_page(self, name, sha='HEAD'): def get_page(self, name, sha='HEAD'):
"""Get page data, partials, commit info. """Get page data, partials, commit info.
@ -186,67 +47,7 @@ class Wiki(HookMixin):
:return: dict :return: dict
""" """
cached = cache.get(self._cache_key(name, sha)) return WikiPage(name, self, sha=sha)
if cached:
return cached
# commit = gittle.utils.git.commit_info(self.repo[sha])
filename = cname_to_filename(name).encode('utf8')
sha = sha.encode('latin-1')
try:
data = self.gittle.get_commit_files(sha, paths=[filename]).get(filename)
if not data:
return None
partials = {}
if data.get('data'):
meta = self.get_meta(data['data'])
if meta and 'import' in meta:
for partial_name in meta['import']:
partials[partial_name] = self.get_page(partial_name)
data['partials'] = partials
data['info'] = self.get_history(name, limit=1)[0]
cache.set(self._cache_key(name, sha), data)
return data
except KeyError:
# HEAD doesn't exist yet
return None
def get_meta(self, content):
"""Get metadata from page if any.
:param content: Page content
:return: dict
"""
if not content.startswith("---"):
return None
meta_end = re.search("\n(\.{3}|\-{3})", content)
if not meta_end:
return None
try:
return yaml.safe_load(content[0:meta_end.start()])
except Exception as e:
return {'error': e.message}
def compare(self, name, old_sha, new_sha):
"""Compare two revisions of the same page.
:param name: Name of page.
:param old_sha: Older sha.
:param new_sha: Newer sha.
:return: str - Raw markup with styles
"""
# TODO: This could be effectively done in the browser
old = self.get_page(name, sha=old_sha)
new = self.get_page(name, sha=new_sha)
return ghdiff.diff(old['data'], new['data'])
def get_index(self): def get_index(self):
"""Get repo index of head. """Get repo index of head.
@ -266,28 +67,56 @@ class Wiki(HookMixin):
return rv return rv
def get_history(self, name, limit=100):
class WikiPage(object):
def __init__(self, name, wiki, sha='HEAD'):
self.name = name
self.filename = cname_to_filename(name)
self.sha = sha.encode('latin-1')
self.wiki = wiki
@property
def data(self):
cache_key = self._cache_key('data')
cached = cache.get(cache_key)
if cached:
return cached
data = self.wiki.gittle.get_commit_files(self.sha, paths=[self.filename]).get(self.filename).get('data')
cache.set(cache_key, data)
return data
@property
def info(self):
cache_key = self._cache_key('info')
cached = cache.get(cache_key)
if cached:
return cached
info = self.get_history(limit=1)[0]
cache.set(cache_key, info)
return info
def get_history(self, limit=100):
"""Get page history. """Get page history.
:param name: Name of page.
:param limit: Limit history size. :param limit: Limit history size.
:return: list -- List of dicts :return: list -- List of dicts
""" """
if not len(self.repo.open_index()): if not len(self.wiki.repo.open_index()):
# Index is empty, no commits # Index is empty, no commits
return [] return []
file_path = cname_to_filename(name)
versions = [] versions = []
walker = self.repo.get_walker(paths=[file_path], max_entries=limit) walker = self.wiki.repo.get_walker(paths=[self.filename], max_entries=limit)
for entry in walker: for entry in walker:
change_type = None change_type = None
for change in entry.changes(): for change in entry.changes():
if change.old.path == file_path: if change.old.path == self.filename:
change_type = change.type change_type = change.type
elif change.new.path == file_path: elif change.new.path == self.filename:
change_type = change.type change_type = change.type
author_name, author_email = entry.commit.author.rstrip('>').split('<') author_name, author_email = entry.commit.author.rstrip('>').split('<')
versions.append(dict( versions.append(dict(
@ -299,3 +128,188 @@ class Wiki(HookMixin):
type=change_type)) type=change_type))
return versions return versions
@property
def partials(self):
data = self.data
if not data:
return {}
partials = {}
meta = self._get_meta(data)
if meta and 'import' in meta:
for partial_name in meta['import']:
partials[partial_name] = self.wiki.get_page(partial_name, sha=self.sha)
return partials
@staticmethod
def _get_meta(content):
"""Get metadata from page if any.
:param content: Page content
:return: dict
"""
if not content.startswith("---"):
return None
meta_end = re.search("\n(\.{3}|\-{3})", content)
if not meta_end:
return None
try:
return yaml.safe_load(content[0:meta_end.start()])
except Exception as e:
return {'error': e.message}
def _cache_key(self, property):
return 'page/{0}[{1}].{2}'.format(self.name, self.sha, property)
def _get_user(self, username, email):
if not username:
username = self.wiki.default_committer_name
if not email:
email = self.wiki.default_committer_email
return username, email
def _clear_cache(self):
cache.delete_many(self._cache_key(p) for p in ['data', 'info'])
def delete(self, username=None, email=None, message=None):
"""Delete page.
:param username: Committer name
:param email: Committer email
:return: str -- Commit sha1
"""
username, email = self._get_user(username, email)
if not message:
message = "Deleted %s" % self.name
# gittle.rm won't actually remove the file, have to do it ourselves
os.remove(os.path.join(self.wiki.path, self.filename))
self.wiki.gittle.rm(self.filename)
commit = self.wiki.gittle.commit(name=username,
email=email,
message=message,
files=[self.filename])
self._clear_cache()
return commit
def rename(self, new_name, username=None, email=None, message=None):
"""Rename page.
:param new_name: New name of page.
:param username: Committer name
:param email: Committer email
:return: str -- Commit sha1
"""
assert self.sha == 'HEAD'
old_filename, new_filename = self.filename, cname_to_filename(new_name)
if old_filename not in self.wiki.gittle.index:
# old doesn't exist
return None
elif old_filename == new_filename:
return None
else:
# file is being overwritten, but that is ok, it's git!
pass
username, email = self._get_user(username, email)
if not message:
message = "Moved %s to %s" % (self.name, new_name)
os.rename(os.path.join(self.wiki.path, old_filename), os.path.join(self.wiki.path, new_filename))
self.wiki.gittle.add(new_filename)
self.wiki.gittle.rm(old_filename)
commit = self.wiki.gittle.commit(name=username,
email=email,
message=message,
files=[old_filename, new_filename])
self._clear_cache()
self.name = new_name
self.filename = new_filename
# We need to clear the cache for the new name as well as the old
self._clear_cache()
return commit
def write(self, content, message=None, create=False, username=None, email=None):
"""Write page to git repo
:param content: Content of page.
:param message: Commit message.
:param create: Perform git add operation?
:param username: Commit Name.
:param email: Commit Email.
:return: Git commit sha1.
"""
assert self.sha == 'HEAD'
dirname = posixpath.join(self.wiki.path, posixpath.dirname(self.filename))
if not os.path.exists(dirname):
os.makedirs(dirname)
with open(self.wiki.path + "/" + self.filename, 'w') as f:
f.write(content)
if create:
self.wiki.gittle.add(self.filename)
if not message:
message = "Updated %s" % self.name
username, email = self._get_user(username, email)
ret = self.wiki.gittle.commit(name=username,
email=email,
message=message,
files=[self.filename])
self._clear_cache()
return ret
def revert(self, commit_sha, message, username, email):
"""Revert page to passed commit sha1
:param commit_sha: Commit Sha1 to revert to.
:param message: Commit message.
:param username: Committer name.
:param email: Committer email.
:return: Git commit sha1
"""
assert self.sha == 'HEAD'
new_page = self.wiki.get_page(self.name, commit_sha)
if not new_page:
raise PageNotFound('Commit not found')
if not message:
commit_info = gittle.utils.git.commit_info(self.wiki.gittle[commit_sha.encode('latin-1')])
message = commit_info['message']
return self.write(new_page.data, message=message, username=username, email=email)
def compare(self, old_sha):
"""Compare two revisions of the same page.
:param old_sha: Older sha.
:return: str - Raw markup with styles
"""
# TODO: This could be effectively done in the browser
old = self.wiki.get_page(self.name, sha=old_sha)
return ghdiff.diff(old.data, self.data)
def __nonzero__(self):
# TODO: Check if page is in index of self.sha
return bool(self.data)

View file

@ -41,8 +41,8 @@ class WikiTest(WikiBaseTest):
self.assert_200(rv) self.assert_200(rv)
self.assert_context('name', 'test') self.assert_context('name', 'test')
eq_(self.get_context_variable('page')['info']['message'], 'test message') eq_(self.get_context_variable('page').data['message'], 'test message')
eq_(self.get_context_variable('page')['data'], 'testing') eq_(self.get_context_variable('page').data, 'testing')
def test_history(self): def test_history(self):
self.assert_200(self.client.get(url_for('wiki.history', name='test'))) self.assert_200(self.client.get(url_for('wiki.history', name='test')))
@ -68,7 +68,7 @@ class WikiTest(WikiBaseTest):
data = json.loads(rv1.data) data = json.loads(rv1.data)
self.client.post(url_for('wiki.revert'), data=dict(name='test', commit=data['sha'])) self.client.post(url_for('wiki.revert'), data=dict(name='test', commit=data['sha']))
self.client.get(url_for('wiki.page', name='test')) self.client.get(url_for('wiki.page', name='test'))
eq_(self.get_context_variable('page')['data'], 'testing_old') eq_(self.get_context_variable('page').data, 'testing_old')
self.assert_404(self.client.post(url_for('wiki.revert'), data=dict(name='test', commit='does not exist'))) self.assert_404(self.client.post(url_for('wiki.revert'), data=dict(name='test', commit='does not exist')))
self.app.config['WIKI_LOCKED_PAGES'] = ['test'] self.app.config['WIKI_LOCKED_PAGES'] = ['test']

View file

@ -28,7 +28,7 @@ def compare(name, fsha, dots, lsha):
if current_app.config.get('PRIVATE_WIKI') and current_user.is_anonymous(): if current_app.config.get('PRIVATE_WIKI') and current_user.is_anonymous():
return current_app.login_manager.unauthorized() return current_app.login_manager.unauthorized()
diff = g.current_wiki.compare(name, fsha, lsha) diff = g.current_wiki.get_page(name, sha=lsha).compare(fsha)
return render_template('wiki/compare.html', return render_template('wiki/compare.html',
name=name, diff=diff, old=fsha, new=lsha) name=name, diff=diff, old=fsha, new=lsha)
@ -47,11 +47,10 @@ def revert():
return dict(error=True, message="Page is locked"), 403 return dict(error=True, message="Page is locked"), 403
try: try:
sha = g.current_wiki.revert_page(cname, sha = g.current_wiki.get_page(cname).revert(commit,
commit, message=message,
message=message, username=current_user.username,
username=current_user.username, email=current_user.email)
email=current_user.email)
except PageNotFound as e: except PageNotFound as e:
return dict(error=True, message=e.message), 404 return dict(error=True, message=e.message), 404
@ -66,7 +65,7 @@ def history(name):
if current_app.config.get('PRIVATE_WIKI') and current_user.is_anonymous(): if current_app.config.get('PRIVATE_WIKI') and current_user.is_anonymous():
return current_app.login_manager.unauthorized() return current_app.login_manager.unauthorized()
hist = g.current_wiki.get_history(name) hist = g.current_wiki.get_page(name).get_history()
for item in hist: for item in hist:
item['gravatar'] = gravatar_url(item['author_email']) item['gravatar'] = gravatar_url(item['author_email'])
return render_template('wiki/history.html', name=name, history=hist) return render_template('wiki/history.html', name=name, history=hist)
@ -85,10 +84,10 @@ def edit(name):
g.assets['js'].append('editor.js') g.assets['js'].append('editor.js')
return render_template('wiki/edit.html', return render_template('wiki/edit.html',
name=cname, name=cname,
content=page.get('data'), content=page.data,
info=page.get('info'), info=page.info,
sha=page.get('sha'), sha=page.sha,
partials=page.get('partials')) partials=page.partials)
@blueprint.route("/_create/", defaults={'name': None}) @blueprint.route("/_create/", defaults={'name': None})
@ -167,12 +166,11 @@ def page_write(name):
if cname in current_app.config.get('WIKI_LOCKED_PAGES'): if cname in current_app.config.get('WIKI_LOCKED_PAGES'):
return dict(error=True, message="Page is locked"), 403 return dict(error=True, message="Page is locked"), 403
sha = g.current_wiki.write_page(cname, sha = g.current_wiki.get_page(cname).write(request.form['content'],
request.form['content'], message=request.form['message'],
message=request.form['message'], create=True,
create=True, username=current_user.username,
username=current_user.username, email=current_user.email)
email=current_user.email)
elif request.method == 'PUT': elif request.method == 'PUT':
edit_cname = to_canonical(request.form['name']) edit_cname = to_canonical(request.form['name'])
@ -181,13 +179,12 @@ def page_write(name):
return dict(error=True, message="Page is locked"), 403 return dict(error=True, message="Page is locked"), 403
if edit_cname != cname: if edit_cname != cname:
g.current_wiki.rename_page(cname, edit_cname) g.current_wiki.get_page(cname).rename(edit_cname)
sha = g.current_wiki.write_page(edit_cname, sha = g.current_wiki.get_page(edit_cname).write(request.form['content'],
request.form['content'], message=request.form['message'],
message=request.form['message'], username=current_user.username,
username=current_user.username, email=current_user.email)
email=current_user.email)
return dict(sha=sha) return dict(sha=sha)
@ -196,9 +193,8 @@ def page_write(name):
if cname in current_app.config.get('WIKI_LOCKED_PAGES'): if cname in current_app.config.get('WIKI_LOCKED_PAGES'):
return dict(error=True, message="Page is locked"), 403 return dict(error=True, message="Page is locked"), 403
sha = g.current_wiki.delete_page(cname, sha = g.current_wiki.get_page(cname).delete(username=current_user.username,
username=current_user.username, email=current_user.email)
email=current_user.email)
return dict(sha=sha) return dict(sha=sha)
@ -216,6 +212,6 @@ def page(name):
data = g.current_wiki.get_page(cname) data = g.current_wiki.get_page(cname)
if data: if data:
return render_template('wiki/page.html', name=cname, page=data, partials=data.get('partials')) return render_template('wiki/page.html', name=cname, page=data, partials=data.partials)
else: else:
return redirect(url_for('wiki.create', name=cname)) return redirect(url_for('wiki.create', name=cname))