Newer
Older
# -*- coding: utf-8 -*-
from __future__ import division
__copyright__ = "Copyright (C) 2014 Andreas Kloeckner"
__license__ = """
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
"""
from django.utils.translation import ugettext as _
from django.utils.timezone import now
from django.core.exceptions import ObjectDoesNotExist, ImproperlyConfigured
from django.core.urlresolvers import NoReverseMatch
from markdown.extensions import Extension
from markdown.treeprocessors import Treeprocessor
from jinja2 import BaseLoader as BaseTemplateLoader, TemplateNotFound
Andreas Klöckner
committed
from yaml import load as load_yaml
class SubdirRepoWrapper(object):
def __init__(self, repo, subdir):
self.repo = repo
# This wrapper should only get used if there is a subdir to be had.
assert subdir
self.subdir = subdir
def controldir(self):
return self.repo.controldir()
def close(self):
self.repo.close()
def get_course_repo_path(course):
from os.path import join
return join(settings.GIT_ROOT, course.identifier)
def get_course_repo(course):
Andreas Klöckner
committed
from dulwich.repo import Repo
repo = Repo(get_course_repo_path(course))
if course.course_root_path:
return SubdirRepoWrapper(repo, course.course_root_path)
else:
return repo
def get_repo_blob(repo, full_name, commit_sha):
:arg full_name: A Unicode string indicating the file name.
:arg commit_sha: A byte string containing the commit hash
"""
if isinstance(repo, SubdirRepoWrapper):
# full_name must be non-empty
full_name = repo.subdir + "/" + full_name
repo = repo.repo
names = full_name.split("/")
# Allow non-ASCII file name
full_name = full_name.encode('utf-8')
tree_sha = repo[commit_sha].tree
tree = repo[tree_sha]
if not full_name:
return tree
try:
for name in names[:-1]:
if not name:
# tolerate empty path components (begrudgingly)
continue
mode, blob_sha = tree[name.encode()]
tree = repo[blob_sha]
mode, blob_sha = tree[names[-1].encode()]
return repo[blob_sha]
except KeyError:
raise ObjectDoesNotExist(_("resource '%s' not found") % full_name)
def get_repo_blob_data_cached(repo, full_name, commit_sha):
"""
:arg commit_sha: A byte string containing the commit hash
"""
if isinstance(commit_sha, six.binary_type):
from six.moves.urllib.parse import quote_plus
cache_key = "%R%1".join((
quote_plus(repo.controldir()),
quote_plus(full_name),
commit_sha.decode(),
".".join(str(s) for s in sys.version_info[:2]),
))
try:
import django.core.cache as cache
except ImproperlyConfigured:
cache_key = None
if cache_key is None:
result = get_repo_blob(repo, full_name, commit_sha).data
assert isinstance(result, six.binary_type)
return result
# Byte string is wrapped in a tuple to force pickling because memcache's
# python wrapper appears to auto-decode/encode string values, thus trying
# to decode our byte strings. Grr.
result = None
# Memcache is apparently limited to 250 characters.
if len(cache_key) < 240:
result = def_cache.get(cache_key)
(result,) = result
assert isinstance(result, six.binary_type), cache_key
return result
result = get_repo_blob(repo, full_name, commit_sha).data
if len(result) <= getattr(settings, "RELATE_CACHE_MAX_BYTES", 0):
def_cache.add(cache_key, (result,), None)
assert isinstance(result, six.binary_type)
def is_repo_file_accessible_as(access_kind, repo, commit_sha, path):
"""
:arg commit_sha: A byte string containing the commit hash
"""
from os.path import dirname, basename, join
attributes_path = join(dirname(path), ".attributes.yml")
from course.content import get_raw_yaml_from_repo
try:
attributes = get_raw_yaml_from_repo(
repo, attributes_path, commit_sha.encode())
except ObjectDoesNotExist:
# no attributes file: not public
return False
path_basename = basename(path)
access_patterns = attributes.get(access_kind, [])
from fnmatch import fnmatch
if isinstance(access_patterns, list):
for pattern in access_patterns:
if isinstance(pattern, six.string_types):
if fnmatch(path_basename, pattern):
return True
JINJA_YAML_RE = re.compile(
r"^\[JINJA\]\s*$(.*?)^\[\/JINJA\]\s*$",
re.MULTILINE | re.DOTALL)
YAML_BLOCK_START_SCALAR_RE = re.compile(
r":\s*[|>](?:[0-9][-+]?|[-+][0-9]?)?(?:\s*\#.*)?$")
GROUP_COMMENT_START = re.compile(r"^\s*#\s*\{\{\{")
LEADING_SPACES_RE = re.compile(r"^( *)")
def process_yaml_for_expansion(yaml_str):
lines = yaml_str.split("\n")
jinja_lines = []
i = 0
line_count = len(lines)
while i < line_count:
l = lines[i]
if GROUP_COMMENT_START.match(l):
jinja_lines.append("{% raw %}")
jinja_lines.append(l)
jinja_lines.append("{% endraw %}")
i += 1
elif YAML_BLOCK_START_SCALAR_RE.search(l):
unprocessed_block_lines = []
unprocessed_block_lines.append(l)
block_start_indent = len(LEADING_SPACES_RE.match(l).group(1))
i += 1
while i < line_count:
l = lines[i]
if not l.rstrip():
i += 1
continue
line_indent = len(LEADING_SPACES_RE.match(l).group(1))
if line_indent <= block_start_indent:
break
else:
i += 1
jinja_lines.append("{% raw %}")
jinja_lines.extend(unprocessed_block_lines)
jinja_lines.append("{% endraw %}")
else:
i += 1
return "\n".join(jinja_lines)
class GitTemplateLoader(BaseTemplateLoader):
def __init__(self, repo, commit_sha):
self.repo = repo
self.commit_sha = commit_sha
def get_source(self, environment, template):
try:
data = get_repo_blob_data_cached(self.repo, template, self.commit_sha)
except ObjectDoesNotExist:
raise TemplateNotFound(template)
source = data.decode('utf-8')
def is_up_to_date():
# There's not much point to caching here, because we create
# a new loader for every request anyhow...
return False
return source, None, is_up_to_date
class YamlBlockEscapingGitTemplateLoader(GitTemplateLoader):
# https://github.com/inducer/relate/issues/130
def get_source(self, environment, template):
source, path, is_up_to_date = \
super(YamlBlockEscapingGitTemplateLoader, self).get_source(
environment, template)
from os.path import splitext
_, ext = splitext(template)
ext = ext.lower()
if ext in [".yml", ".yaml"]:
source = process_yaml_for_expansion(source)
return source, path, is_up_to_date
def expand_yaml_macros(repo, commit_sha, yaml_str):
if isinstance(yaml_str, six.binary_type):
yaml_str = yaml_str.decode("utf-8")
from jinja2 import Environment, StrictUndefined
jinja_env = Environment(
loader=YamlBlockEscapingGitTemplateLoader(repo, commit_sha),
undefined=StrictUndefined)
# {{{ process explicit [JINJA] tags (deprecated)
def compute_replacement(match):
template = jinja_env.from_string(match.group(1))
return template.render()
yaml_str, count = JINJA_YAML_RE.subn(compute_replacement, yaml_str)
if count:
# The file uses explicit [JINJA] tags. Assume that it doesn't
# want anything else processed through YAML.
return yaml_str
# }}}
jinja_str = process_yaml_for_expansion(yaml_str)
template = jinja_env.from_string(jinja_str)
yaml_str = template.render()
return yaml_str
# }}}
# {{{ repo yaml getting
def get_raw_yaml_from_repo(repo, full_name, commit_sha):
"""Return decoded YAML data structure from
the given file in *repo* at *commit_sha*.
:arg commit_sha: A byte string containing the commit hash
from six.moves.urllib.parse import quote_plus
cache_key = "%RAW%%2".join((
quote_plus(repo.controldir()), quote_plus(full_name), commit_sha.decode(),
))
result = None
# Memcache is apparently limited to 250 characters.
if len(cache_key) < 240:
result = def_cache.get(cache_key)
result = load_yaml(
expand_yaml_macros(
repo, commit_sha,
get_repo_blob(repo, full_name, commit_sha).data))
def_cache.add(cache_key, result, None)
return result
def get_yaml_from_repo(repo, full_name, commit_sha, cached=True):
"""Return decoded, struct-ified YAML data structure from
the given file in *repo* at *commit_sha*.
from six.moves.urllib.parse import quote_plus
cache_key = "%%%2".join(
(quote_plus(repo.controldir()), quote_plus(full_name),
import django.core.cache as cache
def_cache = cache.caches["default"]
result = None
# Memcache is apparently limited to 250 characters.
if len(cache_key) < 240:
result = def_cache.get(cache_key)
if result is not None:
return result
expanded = expand_yaml_macros(
repo, commit_sha,
get_repo_blob(repo, full_name, commit_sha).data)
result = dict_to_struct(load_yaml(expanded))
if cached:
def_cache.add(cache_key, result, None)
# }}}
# {{{ markup
def _attr_to_string(key, val):
if val is None:
return key
elif "\"" in val:
return "%s='%s'" % (key, val)
else:
return "%s=\"%s\"" % (key, val)
class TagProcessingHTMLParser(html_parser.HTMLParser):
def __init__(self, out_file, process_tag_func):
self.out_file = out_file
self.process_tag_func = process_tag_func
def handle_starttag(self, tag, attrs):
attrs = dict(attrs)
attrs.update(self.process_tag_func(tag, attrs))
self.out_file.write("<%s %s>" % (tag, " ".join(
_attr_to_string(k, v) for k, v in six.iteritems(attrs))))
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
def handle_endtag(self, tag):
self.out_file.write("</%s>" % tag)
def handle_startendtag(self, tag, attrs):
attrs = dict(attrs)
attrs.update(self.process_tag_func(tag, attrs))
self.out_file.write("<%s %s/>" % (tag, " ".join(
_attr_to_string(k, v) for k, v in attrs.iteritems())))
def handle_data(self, data):
self.out_file.write(data)
def handle_entityref(self, name):
self.out_file.write("&%s;" % name)
def handle_charref(self, name):
self.out_file.write("&#%s;" % name)
def handle_comment(self, data):
self.out_file.write("<!--%s-->" % data)
def handle_decl(self, decl):
self.out_file.write("<!%s>" % decl)
def handle_pi(self, data):
raise NotImplementedError(
_("I have no idea what a processing instruction is."))
def unknown_decl(self, data):
self.out_file.write("<![%s]>" % data)
def __init__(self, md, course, commit_sha, reverse_func):
self.commit_sha = commit_sha
def get_course_identifier(self):
if self.course is None:
return "bogus-course-identifier"
else:
return self.course.identifier
try:
if url.startswith("course:"):
course_id = url[7:]
if course_id:
return self.reverse_func("relate-course_page",
args=(course_id,))
else:
return self.reverse_func("relate-course_page",
args=(self.get_course_identifier(),))
elif url.startswith("flow:"):
flow_id = url[5:]
return self.reverse_func("relate-view_start_flow",
args=(self.get_course_identifier(), flow_id))
elif url.startswith("staticpage:"):
page_path = url[11:]
return self.reverse_func("relate-content_page",
args=(self.get_course_identifier(), page_path))
elif url.startswith("media:"):
media_path = url[6:]
return self.reverse_func("relate-get_media",
args=(
self.get_course_identifier(),
self.commit_sha,
media_path))
elif url.startswith("repo:"):
path = url[5:]
return self.reverse_func("relate-get_repo_file",
args=(
self.get_course_identifier(),
self.commit_sha,
path))
elif url.startswith("repocur:"):
path = url[8:]
return self.reverse_func("relate-get_current_repo_file",
args=(
self.get_course_identifier(),
path))
elif url.strip() == "calendar:":
return self.reverse_func("relate-view_calendar",
except NoReverseMatch:
from base64 import b64encode
message = ("Invalid character in RELATE URL: " + url).encode("utf-8")
return "data:text/plain;base64,"+b64encode(message).decode()
return None
def process_tag(self, tag_name, attrs):
changed_attrs = {}
if tag_name == "table":
changed_attrs["class"] = "table table-condensed"
if tag_name in ["a", "link"] and "href" in attrs:
new_href = self.process_url(attrs["href"])
if new_href is not None:
changed_attrs["href"] = new_href
elif tag_name == "img" and "src" in attrs:
new_src = self.process_url(attrs["src"])
if new_src is not None:
changed_attrs["src"] = new_src
elif tag_name == "object" and "data" in attrs:
new_data = self.process_url(attrs["data"])
if new_data is not None:
changed_attrs["data"] = new_data
return changed_attrs
def process_etree_element(self, element):
changed_attrs = self.process_tag(element.tag, element.attrib)
for key, val in six.iteritems(changed_attrs):
element.set(key, val)
def walk_and_process_tree(self, root):
self.process_etree_element(root)
self.walk_and_process_tree(child)
def run(self, root):
self.walk_and_process_tree(root)
# root through and process Markdown's HTML stash (gross!)
for i, (html, safe) in enumerate(self.md.htmlStash.rawHtmlBlocks):
parser = TagProcessingHTMLParser(outf, self.process_tag)
parser.feed(html)
self.md.htmlStash.rawHtmlBlocks[i] = (outf.getvalue(), safe)
def __init__(self, course, commit_sha, reverse_func):
self.course = course
self.commit_sha = commit_sha
LinkFixerTreeprocessor(md, self.course, self.commit_sha,
reverse_func=self.reverse_func)
def remove_prefix(prefix, s):
if s.startswith(prefix):
return s[len(prefix):]
else:
return s
JINJA_PREFIX = "[JINJA]"
Andreas Klöckner
committed
def markup_to_html(course, repo, commit_sha, text, reverse_func=None,
validate_only=False, jinja_env={}):
if reverse_func is None:
from django.core.urlresolvers import reverse
reverse_func = reverse
if course is not None and not jinja_env:
try:
import django.core.cache as cache
except ImproperlyConfigured:
cache_key = None
else:
import hashlib
cache_key = ("markup:v3:%d:%s:%s"
% (course.id, str(commit_sha),
hashlib.md5(text.encode("utf-8")).hexdigest()))
def_cache = cache.caches["default"]
result = def_cache.get(cache_key)
if result is not None:
return result
if text.lstrip().startswith(JINJA_PREFIX):
text = remove_prefix(JINJA_PREFIX, text.lstrip())
else:
cache_key = None
# {{{ process through Jinja
from jinja2 import Environment, StrictUndefined
env = Environment(
loader=GitTemplateLoader(repo, commit_sha),
undefined=StrictUndefined)
template = env.from_string(text)
text = template.render(**jinja_env)
Andreas Klöckner
committed
if validate_only:
return
from course.mdx_mathjax import MathJaxExtension
LinkFixerExtension(course, commit_sha, reverse_func=reverse_func),
"markdown.extensions.extra",
"markdown.extensions.codehilite",
],
output_format="html5")
if cache_key is not None:
def_cache.add(cache_key, result, None)
return result
TITLE_RE = re.compile(r"^\#+\s*(\w.*)", re.UNICODE)
def extract_title_from_markup(markup_text):
lines = markup_text.split("\n")
for l in lines[:10]:
match = TITLE_RE.match(l)
if match is not None:
return match.group(1)
return None
# {{{ datespec processing
DATE_RE = re.compile(r"^([0-9]+)\-([01][0-9])\-([0-3][0-9])$")
TRAILING_NUMERAL_RE = re.compile(r"^(.*)\s+([0-9]+)$")
class InvalidDatespec(ValueError):
def __init__(self, datespec):
ValueError.__init__(self, str(datespec))
self.datespec = datespec
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
AT_TIME_RE = re.compile(r"^(.*)\s*@\s*([0-2]?[0-9])\:([0-9][0-9])\s*$")
class AtTimePostprocessor(object):
def __init__(self, hour, minute, second=0):
self.hour = hour
self.minute = minute
self.second = second
@classmethod
def parse(cls, s):
match = AT_TIME_RE.match(s)
if match is not None:
hour = int(match.group(2))
minute = int(match.group(3))
if not (0 <= hour < 24):
raise InvalidDatespec(s)
if not (0 <= minute < 60):
raise InvalidDatespec(s)
return match.group(1), AtTimePostprocessor(hour, minute)
else:
return s, None
def apply(self, dtm):
from pytz import timezone
server_tz = timezone(settings.TIME_ZONE)
return dtm.astimezone(server_tz).replace(
hour=self.hour,
minute=self.minute,
second=self.second)
PLUS_DELTA_RE = re.compile(r"^(.*)\s*([+-])\s*([0-9]+)\s+"
"(weeks?|days?|hours?|minutes?)$")
class PlusDeltaPostprocessor(object):
def __init__(self, count, period):
self.count = count
self.period = period
@classmethod
def parse(cls, s):
match = PLUS_DELTA_RE.match(s)
if match is not None:
count = int(match.group(3))
if match.group(2) == "-":
count = -count
period = match.group(4)
return match.group(1), PlusDeltaPostprocessor(count, period)
else:
return s, None
def apply(self, dtm):
if self.period.startswith("week"):
d = datetime.timedelta(weeks=self.count)
elif self.period.startswith("day"):
d = datetime.timedelta(days=self.count)
elif self.period.startswith("hour"):
d = datetime.timedelta(hours=self.count)
elif self.period.startswith("minute"):
d = datetime.timedelta(minutes=self.count)
else:
raise InvalidDatespec(_("invalid period: %s" % self.period))
return dtm + d
DATESPEC_POSTPROCESSORS = [
AtTimePostprocessor,
PlusDeltaPostprocessor,
]
Andreas Klöckner
committed
def parse_date_spec(course, datespec, vctx=None, location=None):
Andreas Klöckner
committed
orig_datespec = datespec
def localize_if_needed(d):
if d.tzinfo is None:
from relate.utils import localize_datetime
return localize_datetime(d)
return d
if isinstance(datespec, datetime.datetime):
return localize_if_needed(datespec)
if isinstance(datespec, datetime.date):
return localize_if_needed(
datetime.datetime.combine(datespec, datetime.time.min))
datespec = datespec.strip()
# {{{ parse postprocessors
postprocs = []
while True:
parsed_one = False
for pp_class in DATESPEC_POSTPROCESSORS:
datespec, postproc = pp_class.parse(datespec)
if postproc is not None:
parsed_one = True
postprocs.insert(0, postproc)
break
datespec = datespec.strip()
if not parsed_one:
break
# }}}
def apply_postprocs(dtime):
for postproc in postprocs:
dtime = postproc.apply(dtime)
return dtime
match = DATE_RE.match(datespec)
result = datetime.date(
int(match.group(1)),
int(match.group(2)),
int(match.group(3)))
result = localize_if_needed(
datetime.datetime.combine(result, datetime.time.min))
return apply_postprocs(result)
match = TRAILING_NUMERAL_RE.match(datespec)
Andreas Klöckner
committed
if vctx is not None:
from course.validation import validate_identifier
Andreas Klöckner
committed
validate_identifier(vctx, "%s: event kind" % location,
Andreas Klöckner
committed
match.group(1))
if course is None:
return now()
from course.models import Event
return apply_postprocs(
Event.objects.get(
course=course,
kind=match.group(1),
ordinal=int(match.group(2))).time)
except ObjectDoesNotExist:
Andreas Klöckner
committed
if vctx is not None:
vctx.add_warning(
location,
_("unrecognized date/time specification: '%s' "
"(interpreted as 'now')")
% orig_datespec)
return now()
if vctx is not None:
from course.validation import validate_identifier
Andreas Klöckner
committed
validate_identifier(vctx, "%s: event kind" % location, datespec)
Andreas Klöckner
committed
if course is None:
return now()
from course.models import Event
return apply_postprocs(
Event.objects.get(
course=course,
kind=datespec,
ordinal=None).time)
except ObjectDoesNotExist:
Andreas Klöckner
committed
if vctx is not None:
vctx.add_warning(
location,
_("unrecognized date/time specification: '%s' "
"(interpreted as 'now')")
% orig_datespec)
return now()
# }}}
# {{{ page chunks
def compute_chunk_weight_and_shown(course, chunk, role, now_datetime,
facilities):
if not hasattr(chunk, "rules"):
return 0, True
if hasattr(rule, "if_has_role"):
if role not in rule.if_has_role:
if hasattr(rule, "if_after"):
start_date = parse_date_spec(course, rule.if_after)
if now_datetime < start_date:
continue
if hasattr(rule, "if_before"):
end_date = parse_date_spec(course, rule.if_before)
if end_date < now_datetime:
continue
if hasattr(rule, "if_in_facility"):
if rule.if_in_facility not in facilities:
if hasattr(rule, "roles"):
if role not in rule.roles:
continue
start_date = parse_date_spec(course, rule.start)
end_date = parse_date_spec(course, rule.end)
shown = True
if hasattr(rule, "shown"):
shown = rule.shown
return rule.weight, shown
def get_processed_page_chunks(course, repo, commit_sha,
page_desc, role, now_datetime, facilities):
for chunk in page_desc.chunks:
chunk.weight, chunk.shown = \
compute_chunk_weight_and_shown(
facilities)
chunk.html_content = markup_to_html(course, repo, commit_sha, chunk.content)
if not hasattr(chunk, "title"):
from course.content import extract_title_from_markup
chunk.title = extract_title_from_markup(chunk.content)
page_desc.chunks.sort(key=lambda chunk: chunk.weight, reverse=True)
return [chunk for chunk in page_desc.chunks
# }}}
# {{{ repo desc getting
def normalize_page_desc(page_desc):
if hasattr(page_desc, "content"):
content = page_desc.content
from relate.utils import struct_to_dict, Struct
d = struct_to_dict(page_desc)
del d["content"]
d["chunks"] = [Struct({"id": "main", "content": content})]
return Struct(d)
return page_desc
def get_staticpage_desc(repo, course, commit_sha, filename):
page_desc = get_yaml_from_repo(repo, filename, commit_sha)
page_desc = normalize_page_desc(page_desc)
return page_desc
def get_course_desc(repo, course, commit_sha):
return get_staticpage_desc(repo, course, commit_sha, course.course_file)
def normalize_flow_desc(flow_desc):
if hasattr(flow_desc, "pages"):
pages = flow_desc.pages
from relate.utils import struct_to_dict, Struct
d = struct_to_dict(flow_desc)
del d["pages"]
d["groups"] = [Struct({"id": "main", "pages": pages})]
return Struct(d)
Andreas Klöckner
committed
if hasattr(flow_desc, "rules"):
rules = flow_desc.rules
if not hasattr(rules, "grade_identifier"):
# Legacy content with grade_identifier in grading rule,
# move first found grade_identifier up to rules.
Andreas Klöckner
committed
rules.grade_identifier = None
rules.grade_aggregation_strategy = None
Andreas Klöckner
committed
for grule in rules.grading:
if grule.grade_identifier is not None: