Refactor members views

members/views/utils.py

@@ -0,0 +1,167 @@
+from django.shortcuts import render
+from django.contrib.auth.decorators import permission_required
+from django.views.decorators.http import require_http_methods
+from django.views.decorators.csrf import ensure_csrf_cookie
+from django.http import HttpResponse, HttpResponseRedirect
+from django.core.mail import send_mail
+from django.conf import settings
+from django.utils.translation import ugettext as _
+from django.forms.models import model_to_dict
+
+# REST framework
+from members.serializers import MemberSerializer
+from rest_framework import generics
+from rest_framework import permissions
+from rest_framework.throttling import UserRateThrottle, AnonRateThrottle
+
+from members.models import Member, Request, Payment
+
+
+# Can be used to retrieve single member information via REST API
+class MemberDetail(generics.RetrieveAPIView):
+    """Member detail rest API view."""
+
+    queryset = Member.objects.all()
+    serializer_class = MemberSerializer
+    permission_classes = (permissions.IsAdminUser, )
+    throttle_classes = (UserRateThrottle, AnonRateThrottle, )
+
+
+def error_view(request, message):
+    return render(request, 'error.html', {'error': str(message)})
+
+
+def validate_recaptcha(response):
+    """
+    Recaptcha is used in member applications.
+
+    :param response:
+    :return: Boolean, success or not
+    """
+    values = {
+        'secret': settings.GOOGLE_RECAPTCHA_SECRET_KEY,
+        'response': response,
+    }
+    url = "https://www.google.com/recaptcha/api/siteverify"
+    headers = {'Content-type': 'application/x-www-form-urlencoded'}
+    resp = requests.post(url, values, headers=headers)
+    try:
+        result = json.loads(resp.text)
+        logging.info('Recaptcha response: {}'.format(result))
+        return result["success"]
+    except:
+        return False
+
+
+def send_mail_wrapper(subject, message):
+    """Call send_mail function."""
+    send_mail(subject,
+              message,
+              'no-reply@sahkoinsinoorikilta.fi',
+              ['viestintamestari@sahkoinsinoorikilta.fi'],
+              fail_silently=False)
+
+
+def convert_table_to_html(table, request):
+    """
+    Convert table to html.
+
+    This is a horrible hack for converting a table object to raw html.
+    Even with extensive research I wasn't able to find a way to add a path
+    prefix "e.g. /members/list" to the query strings "e.g. ?sort=foo", so I
+    did it manually with string.replace.
+
+    Note: When adding the html to a page, you need to run it through
+    the "safe" filter. E.g. "{{ table|safe }}"
+
+    :param table: Table object from members.tables
+    :param request:  HttpRequest
+    :return: Raw html string
+    """
+    table_as_html = table.as_html(request)
+    path = request.path
+
+    fixed = table_as_html.replace(r'href="?', r'href="{}?'.format(path))
+    return fixed
+
+
+@ensure_csrf_cookie
+@require_http_methods(["GET"])
+@permission_required('members.change_member', login_url='/login')
+def settings_page(request, *args, **kwargs):
+    """Render member app settings page."""
+    return render(request, 'settings.html', {})
+
+
+@ensure_csrf_cookie
+@require_http_methods(["POST"])
+@permission_required('members.change_member', login_url='/login')
+def import_csv(request, *args, **kwargs):
+    """Get csv data imported to page and create members based on that."""
+    try:
+        data = request.POST['textfield']
+        payment_source = request.POST['payment_source']
+    except:
+        return render(request,
+                      'error.html',
+                      {'error': _('Missing "textfield" POST request field')})
+
+    try:
+        result = MemberForm.csv_to_models(data, payment_source=payment_source)
+    except CSVValidationError as ex:
+        logging.exception('Model validation error')
+        return error_view(request, ex.form_errors)
+    except Exception as ex:
+        logging.exception('Other error in CSV import')
+        return error_view(request, ex)
+
+    member_table = MemberTable(result.members,
+                               request=request,
+                               exclude=['id', 'options'],
+                               attrs={'class': 'table table-bordered table-hover'})
+
+    member_table_html = convert_table_to_html(member_table, request)
+
+    payment_table = PaymentTable(result.payments,
+                                 request=request,
+                                 exclude=['id', 'options'],
+                                 attrs={'class': 'table table-bordered table-hover'})
+
+    payment_table_html = convert_table_to_html(payment_table, request)
+
+    request.session['models'] = result
+    context = {
+        'members': member_table_html,
+        'payments': payment_table_html
+    }
+    return render(request, 'member_add_many_confirm.html', context)
+
+
+@ensure_csrf_cookie
+@require_http_methods(["GET"])
+@permission_required('members.change_member', login_url='/login')
+def export_csv(request, *args, **kwargs):
+    """Export members as csv."""
+    response = HttpResponse()
+    response['Content-type'] = 'text/csv'
+    response['Accept'] = 'text/csv'
+    response['Content-Disposition'] = 'filename; filename=members.csv'
+    writer = csv.writer(response, csv.excel)
+    # BOM (optional...Excel needs it to open UTF-8 file properly)
+    response.write(u'\ufeff'.encode('utf8'))
+    for obj in Member.objects.all():
+        data = obj.as_array()
+        field_list = map(lambda d: str(d), data)
+
+        writer.writerow(field_list)
+
+    return response
+
+
+def send_mail_wrapper(subject, message, email_to):
+    """Send mail to default email."""
+    send_mail(subject,
+              message,
+              settings.DEFAULT_EMAIL_FROM,
+              [email_to],
+              fail_silently=False)