X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/577bb2b84d936d65bf072ef0a6b898db9d6e77ab..5913c54d19b8f6775633176032161d49f9b2f1aa:/apps/wiki/helpers.py?ds=inline diff --git a/apps/wiki/helpers.py b/apps/wiki/helpers.py deleted file mode 100644 index 3e0267ed..00000000 --- a/apps/wiki/helpers.py +++ /dev/null @@ -1,194 +0,0 @@ -from django import http -from django.db.models import Count -from django.utils import simplejson as json -from django.utils.functional import Promise -from datetime import datetime -from functools import wraps - - -class ExtendedEncoder(json.JSONEncoder): - - def default(self, obj): - if isinstance(obj, Promise): - return unicode(obj) - - if isinstance(obj, datetime): - return datetime.ctime(obj) + " " + (datetime.tzname(obj) or 'GMT') - - return json.JSONEncoder.default(self, obj) - - -# shortcut for JSON reponses -class JSONResponse(http.HttpResponse): - - def __init__(self, data={}, **kwargs): - # get rid of mimetype - kwargs.pop('mimetype', None) - - data = json.dumps(data, cls=ExtendedEncoder) - super(JSONResponse, self).__init__(data, mimetype="application/json", **kwargs) - - -# return errors -class JSONFormInvalid(JSONResponse): - def __init__(self, form): - super(JSONFormInvalid, self).__init__(form.errors, status=400) - - -class JSONServerError(JSONResponse): - def __init__(self, *args, **kwargs): - kwargs['status'] = 500 - super(JSONServerError, self).__init__(*args, **kwargs) - - -def ajax_login_required(view): - @wraps(view) - def authenticated_view(request, *args, **kwargs): - if not request.user.is_authenticated(): - return http.HttpResponse("Login required.", status=401, mimetype="text/plain") - return view(request, *args, **kwargs) - return authenticated_view - - -def ajax_require_permission(permission): - def decorator(view): - @wraps(view) - def authorized_view(request, *args, **kwargs): - if not request.user.has_perm(permission): - return http.HttpResponse("Access Forbidden.", status=403, mimetype="text/plain") - return view(request, *args, **kwargs) - return authorized_view - return decorator - -import collections - -def recursive_groupby(iterable): - """ -# >>> recursive_groupby([1,2,3,4,5]) -# [1, 2, 3, 4, 5] - - >>> recursive_groupby([[1]]) - [1] - - >>> recursive_groupby([('a', 1),('a', 2), 3, ('b', 4), 5]) - ['a', [1, 2], 3, 'b', [4], 5] - - >>> recursive_groupby([('a', 'x', 1),('a', 'x', 2), ('a', 'x', 3)]) - ['a', ['x', [1, 2, 3]]] - - """ - - def _generator(iterator): - group = None - grouper = None - - for item in iterator: - if not isinstance(item, collections.Sequence): - if grouper is not None: - yield grouper - if len(group): - yield recursive_groupby(group) - group = None - grouper = None - yield item - continue - elif len(item) == 1: - if grouper is not None: - yield grouper - if len(group): - yield recursive_groupby(group) - group = None - grouper = None - yield item[0] - continue - elif not len(item): - continue - - if grouper is None: - group = [item[1:]] - grouper = item[0] - continue - - if grouper != item[0]: - if grouper is not None: - yield grouper - if len(group): - yield recursive_groupby(group) - group = None - grouper = None - group = [item[1:]] - grouper = item[0] - continue - - group.append(item[1:]) - - if grouper is not None: - yield grouper - if len(group): - yield recursive_groupby(group) - group = None - grouper = None - - return list(_generator(iterable)) - - -def active_tab(tab): - """ - View decorator, which puts tab info on a request. - """ - def wrapper(f): - @wraps(f) - def wrapped(request, *args, **kwargs): - request.wiki_active_tab = tab - return f(request, *args, **kwargs) - return wrapped - return wrapper - - -class ChunksList(object): - def __init__(self, chunk_qs): - self.chunk_qs = chunk_qs.annotate( - book_length=Count('book__chunk')).select_related( - 'book', 'stage__name', - 'user') - - self.book_qs = chunk_qs.values('book_id') - - def __getitem__(self, key): - if isinstance(key, slice): - return self.get_slice(key) - elif isinstance(key, int): - return self.get_slice(slice(key, key+1))[0] - else: - raise TypeError('Unsupported list index. Must be a slice or an int.') - - def __len__(self): - return self.book_qs.count() - - def get_slice(self, slice_): - book_ids = [x['book_id'] for x in self.book_qs[slice_]] - chunk_qs = self.chunk_qs.filter(book__in=book_ids) - - chunks_list = [] - book = None - for chunk in chunk_qs: - if chunk.book != book: - book = chunk.book - chunks_list.append(ChoiceChunks(book, [chunk], chunk.book_length)) - else: - chunks_list[-1].chunks.append(chunk) - return chunks_list - - -class ChoiceChunks(object): - """ - Associates the given chunks iterable for a book. - """ - - chunks = None - - def __init__(self, book, chunks, book_length): - self.book = book - self.chunks = chunks - self.book_length = book_length -