fix erroneous commit
[pithos] / pithos / im / views.py
index b3c72c2..36d5aae 100644 (file)
@@ -34,6 +34,7 @@
 import json
 import logging
 import socket
+import csv
 
 from datetime import datetime
 from functools import wraps
@@ -50,6 +51,8 @@ from django.utils.http import urlencode
 from django.utils.translation import ugettext as _
 from django.core.urlresolvers import reverse
 
+from urllib import quote
+
 from pithos.im.models import User, Invitation
 from pithos.im.util import isoformat
 
@@ -73,7 +76,6 @@ def requires_login(func):
         return func(request, *args)
     return wrapper
 
-
 def requires_admin(func):
     @wraps(func)
     def wrapper(request, *args):
@@ -89,7 +91,11 @@ def requires_admin(func):
 
 
 def index(request):
-    return render_response('index.html', next=request.GET.get('next', ''))
+    kwargs = {'standard_modules':settings.IM_STANDARD_MODULES,
+              'other_modules':settings.IM_OTHER_MODULES}
+    return render_response('index.html',
+                           next=request.GET.get('next', ''),
+                           **kwargs)
 
 
 @requires_admin
@@ -133,23 +139,6 @@ def users_list(request):
                             prev=prev,
                             next=next)
 
-
-@requires_admin
-def users_create(request):
-    if request.method == 'GET':
-        return render_response('users_create.html')
-    if request.method == 'POST':
-        user = User()
-        user.uniq = request.POST.get('uniq')
-        user.realname = request.POST.get('realname')
-        user.is_admin = True if request.POST.get('admin') else False
-        user.affiliation = request.POST.get('affiliation')
-        user.quota = int(request.POST.get('quota') or 0) * (1024 ** 3)  # In GiB
-        user.renew_token()
-        user.save()
-        return redirect(users_info, user.id)
-
-
 @requires_admin
 def users_info(request, user_id):
     user = User.objects.get(id=user_id)
@@ -197,12 +186,15 @@ def generate_invitation_code():
             return code
 
 
-def send_invitation(inv):
-    url = settings.INVITATION_LOGIN_TARGET % inv.code
+def send_invitation(baseurl, inv):
+    url = settings.INVITATION_LOGIN_TARGET % (baseurl, inv.code, quote(baseurl))
     subject = _('Invitation to Pithos')
     message = render_to_string('invitation.txt', {
                 'invitation': inv,
-                'url': url})
+                'url': url,
+                'baseurl': baseurl,
+                'service': settings.SERVICE_NAME,
+                'support': settings.DEFAULT_CONTACT_EMAIL})
     sender = settings.DEFAULT_FROM_EMAIL
     send_mail(subject, message, sender, [inv.uniq])
     logging.info('Sent invitation %s', inv)
@@ -226,7 +218,7 @@ def invite(request):
                 defaults={'code': code, 'realname': realname})
             
             try:
-                send_invitation(invitation)
+                send_invitation(request.build_absolute_uri('/').rstrip('/'), invitation)
                 if created:
                     inviter.invitations = max(0, inviter.invitations - 1)
                     inviter.save()
@@ -234,7 +226,7 @@ def invite(request):
                 message = _('Invitation sent to %s' % uniq)
             except (SMTPException, socket.error) as e:
                 status = 'error'
-                message = e.strerror
+                message = getattr(e, 'strerror', '')
         else:
             status = 'error'
             message = _('No invitations left')
@@ -252,3 +244,243 @@ def invite(request):
             'status': status,
             'message': message})
     return HttpResponse(html)
+
+def send_verification(baseurl, user):
+    url = settings.ACTIVATION_LOGIN_TARGET % (baseurl,
+                                              quote(user.auth_token),
+                                              quote(baseurl))
+    message = render_to_string('activation.txt', {
+            'user': user,
+            'url': url,
+            'baseurl': baseurl,
+            'service': settings.SERVICE_NAME,
+            'support': settings.DEFAULT_CONTACT_EMAIL})
+    sender = settings.DEFAULT_FROM_EMAIL
+    send_mail('Pithos account activation', message, sender, [user.email])
+    logging.info('Sent activation %s', user)
+
+def local_create(request):
+    if request.method == 'GET':
+        return render_response('local_create.html')
+    elif request.method == 'POST':
+        username = request.POST.get('uniq')
+        realname = request.POST.get('realname')
+        email = request.POST.get('email')
+        password = request.POST.get('password')
+        status = 'success'
+        cookie_value = None
+        if not username:
+            status = 'error'
+            message = 'No username provided'
+        elif not password:
+            status = 'error'
+            message = 'No password provided'
+        elif not email:
+            status = 'error'
+            message = 'No email provided'
+        
+        if status == 'success':
+            username = '%s@local' % username
+            try:
+                user = User.objects.get(uniq=username)
+                status = 'error'
+                message = 'Username is not available'
+            except User.DoesNotExist:
+                user = User()
+                user.uniq = username 
+                user.realname = realname
+                user.email = request.POST.get('email')
+                user.password = request.POST.get('password')
+                user.is_admin = False
+                user.quota = 0
+                user.state = 'UNVERIFIED'
+                user.level = 1
+                user.renew_token()
+                try:
+                    send_verification(request.build_absolute_uri('/').rstrip('/'), user)
+                    message = _('Verification sent to %s' % user.email)
+                    user.save()
+                except (SMTPException, socket.error) as e:
+                    status = 'error'
+                    name = 'strerror'
+                    message = getattr(e, name) if hasattr(e, name) else e
+        
+        html = render_to_string('local_create.html', {
+                'status': status,
+                'message': message})
+        response = HttpResponse(html)
+        return response
+
+def send_password(baseurl, user):
+    url = settings.PASSWORD_RESET_TARGET % (baseurl,
+                                            quote(user.uniq),
+                                            quote(baseurl))
+    message = render_to_string('password.txt', {
+            'user': user,
+            'url': url,
+            'baseurl': baseurl,
+            'service': settings.SERVICE_NAME,
+            'support': settings.DEFAULT_CONTACT_EMAIL})
+    sender = settings.DEFAULT_FROM_EMAIL
+    send_mail('Pithos password recovering', message, sender, [user.email])
+    logging.info('Sent password %s', user)
+
+def reclaim_password(request):
+    if request.method == 'GET':
+        return render_response('reclaim.html')
+    elif request.method == 'POST':
+        username = request.POST.get('uniq')
+        username = '%s@local' % username
+        try:
+            user = User.objects.get(uniq=username)
+            try:
+                send_password(request.build_absolute_uri('/').rstrip('/'), user)
+                status = 'success'
+                message = _('Password reset sent to %s' % user.email)
+                user.status = 'UNVERIFIED'
+                user.save()
+            except (SMTPException, socket.error) as e:
+                status = 'error'
+                name = 'strerror'
+                message = getattr(e, name) if hasattr(e, name) else e
+        except User.DoesNotExist:
+            status = 'error'
+            message = 'Username does not exist'
+        
+        html = render_to_string('reclaim.html', {
+                'status': status,
+                'message': message})
+        return HttpResponse(html)
+
+@requires_admin
+def invitations_list(request):
+    invitations = Invitation.objects.order_by('id')
+    
+    filter = request.GET.get('filter', '')
+    if filter:
+        if filter.startswith('-'):
+            invitations = invitations.exclude(uniq__icontains=filter[1:])
+        else:
+            invitations = invitations.filter(uniq__icontains=filter)
+    
+    try:
+        page = int(request.GET.get('page', 1))
+    except ValueError:
+        page = 1
+    offset = max(0, page - 1) * settings.ADMIN_PAGE_LIMIT
+    limit = offset + settings.ADMIN_PAGE_LIMIT
+    
+    npages = int(ceil(1.0 * invitations.count() / settings.ADMIN_PAGE_LIMIT))
+    prev = page - 1 if page > 1 else None
+    next = page + 1 if page < npages else None
+    return render_response('invitations_list.html',
+                            invitations=invitations[offset:limit],
+                            filter=filter,
+                            pages=range(1, npages + 1),
+                            page=page,
+                            prev=prev,
+                            next=next)
+
+@requires_admin
+def invitations_export(request):
+    # Create the HttpResponse object with the appropriate CSV header.
+    response = HttpResponse(mimetype='text/csv')
+    response['Content-Disposition'] = 'attachment; filename=invitations.csv'
+
+    writer = csv.writer(response)
+    writer.writerow(['ID',
+                     'Uniq',
+                     'Real Name',
+                     'Code',
+                     'Inviter Uniq',
+                     'Inviter Real Name',
+                     'Is_accepted',
+                     'Created',
+                     'Accepted',])
+    invitations = Invitation.objects.order_by('id')
+    for inv in invitations:
+        writer.writerow([inv.id,
+                         inv.uniq.encode("utf-8"),
+                         inv.realname.encode("utf-8"),
+                         inv.code,
+                         inv.inviter.uniq.encode("utf-8"),
+                         inv.inviter.realname.encode("utf-8"),
+                         inv.is_accepted,
+                         inv.created,
+                         inv.accepted])
+
+    return response
+
+
+@requires_admin
+def users_export(request):
+    # Create the HttpResponse object with the appropriate CSV header.
+    response = HttpResponse(mimetype='text/csv')
+    response['Content-Disposition'] = 'attachment; filename=users.csv'
+
+    writer = csv.writer(response)
+    writer.writerow(['ID',
+                     'Uniq',
+                     'Real Name',
+                     'Admin',
+                     'Affiliation',
+                     'State',
+                     'Quota (GiB)',
+                     'Updated',])
+    users = User.objects.order_by('id')
+    for u in users:
+        writer.writerow([u.id,
+                         u.uniq.encode("utf-8"),
+                         u.realname.encode("utf-8"),
+                         u.is_admin,
+                         u.affiliation.encode("utf-8"),
+                         u.state.encode("utf-8"),
+                         u.quota,
+                         u.updated])
+
+    return response
+
+@requires_admin
+def users_create(request):
+    if request.method == 'GET':
+        return render_response('users_create.html')
+    if request.method == 'POST':
+        user = User()
+        user.uniq = request.POST.get('uniq')
+        user.realname = request.POST.get('realname')
+        user.is_admin = True if request.POST.get('admin') else False
+        user.affiliation = request.POST.get('affiliation')
+        user.quota = int(request.POST.get('quota') or 0) * (1024 ** 3)  # In GiB
+        user.renew_token()
+        user.save()
+        return redirect(users_info, user.id)
+
+@requires_login
+def users_profile(request):
+    next = request.GET.get('next')
+    user = User.objects.get(uniq=request.user)
+    states = [x[0] for x in User.ACCOUNT_STATE]
+    return render_response('users_profile.html',
+                            user=user,
+                            states=states,
+                            next=next)
+
+@requires_login
+def users_edit(request):
+    user = User.objects.get(uniq=request.user)
+    user.realname = request.POST.get('realname')
+    user.affiliation = request.POST.get('affiliation')
+    user.is_verified = True
+    user.save()
+    next = request.POST.get('next')
+    if next:
+        return redirect(next)
+    
+    status = 'success'
+    message = _('Profile has been updated')
+    html = render_to_string('users_profile.html', {
+            'user': user,
+            'status': status,
+            'message': message})
+    return HttpResponse(html)
+