f6d9b1b55d468992ad485676f042d32877c50580
[redakcja.git] / apps / api / utils.py
1 # -*- encoding: utf-8 -*-
2
3 __author__ = "Ɓukasz Rekucki"
4 __date__ = "$2009-09-20 21:48:03$"
5 __doc__ = "Module documentation."
6
7
8 from functools import wraps
9 from piston.emitters import Emitter
10 from piston.utils import rc
11
12 import api.response
13
14 import wlrepo
15 import settings
16
17 class TextEmitter(Emitter):
18     def render(self, request):
19         return unicode(self.construct())
20
21 Emitter.register('raw', TextEmitter, 'text/plain; charset=utf-8')
22 Emitter.register('rawhtml', TextEmitter, 'text/html; charset=utf-8')
23 Emitter.register('rawxml', TextEmitter, 'application/xml; charset=utf-8')
24
25 class DjangoAuth(object):
26
27     def is_authenticated(self, request):
28         return request.user.is_authenticated()
29
30     def challenge(self):
31         return rc.FORBIDDEN
32
33
34 def validate_form(formclass, source='GET'):
35   
36     def decorator(func):
37         @wraps(func)
38         def decorated(self, request, * args, ** kwargs):
39             form = formclass(getattr(request, source), request.FILES)
40
41             if not form.is_valid():
42                 errorlist = [{'field': k, 'errors': str(e)} for k, e in form.errors.items()]
43                 return api.response.BadRequest().django_response(errorlist)
44
45             kwargs['form'] = form
46             return func(self, request, * args, ** kwargs)
47         return decorated
48     return decorator
49     
50 def hglibrary(func):
51     @wraps(func)
52     def decorated(self, *args, **kwargs):
53         l = wlrepo.open_library(settings.REPOSITORY_PATH, 'hg')
54         kwargs['lib'] = l
55         return func(self, *args, **kwargs)
56     return decorated
57
58
59
60 import re
61 import locale
62
63 NAT_EXPR = re.compile(r'(\d+)', re.LOCALE | re.UNICODE)
64 def natural_order(get_key=lambda x: x):
65
66     def getter(key):
67         nkey = get_key(key)
68         if not isinstance(key, unicode):
69             ukey = key.decode('utf-8')
70         else:
71             ukey = nkey
72
73         parts = enumerate( NAT_EXPR.split(ukey))            
74         return [int(x) if n%2 else locale.strxfrm(x.encode('utf-8')) for (n,x) in parts ]
75         
76     return getter
77