Mercurial > public > sg101

annotate membermap/views.py @ 1044:a5048a9bc2a8

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Added John Blair's Southern California Surf Music book to store page.
author Brian Neal <bgneal@gmail.com>
date Sat, 06 Feb 2016 14:49:17 -0600
parents e1c03da72818
children 26f2b83e7468
rev   line source
gremmie@1 1 """
gremmie@1 2 Views for the membermap application.
bgneal@679 3
gremmie@1 4 """
bgneal@679 5 import json
bgneal@679 6
bgneal@1031 7 from django.shortcuts import render
bgneal@266 8 from django.template.loader import render_to_string
gremmie@1 9 from django.http import HttpResponse
gremmie@1 10 from django.http import HttpResponseBadRequest
gremmie@1 11 from django.http import HttpResponseForbidden
gremmie@1 12 from django.views.decorators.http import require_POST
bgneal@266 13 from django.core.cache import cache
gremmie@1 14
bgneal@1024 15 from core.html import ImageCheckError
bgneal@1024 16 from core.html import image_check
bgneal@1024 17 from core.markup import site_markup
gremmie@1 18 from membermap.models import MapEntry
gremmie@1 19 from membermap.forms import MapEntryForm
bgneal@266 20
bgneal@266 21 CACHE_KEY = 'membermap_json'
bgneal@266 22 CACHE_TIMEOUT = 5 * 60
gremmie@1 23
gremmie@1 24
gremmie@1 25 def index(request):
gremmie@1 26 entry = None
gremmie@1 27 if request.user.is_authenticated():
gremmie@1 28 try:
gremmie@1 29 entry = MapEntry.objects.get(user=request.user)
gremmie@1 30 except MapEntry.DoesNotExist:
gremmie@1 31 pass
gremmie@1 32 if entry is not None:
gremmie@1 33 form = MapEntryForm(initial={
gremmie@1 34 'location': entry.location,
gremmie@1 35 'message': entry.message})
gremmie@1 36 else:
gremmie@1 37 form = MapEntryForm()
gremmie@1 38
bgneal@1031 39 return render(request, 'membermap/index.html', {
gremmie@1 40 'form': form,
bgneal@1031 41 })
gremmie@1 42
gremmie@1 43
gremmie@1 44 def query(request):
gremmie@1 45 """
gremmie@1 46 This view is called by AJAX. If the user is logged in, return
gremmie@1 47 a JSON object that consists of:
gremmie@1 48 "users" : array of user objects
gremmie@1 49 "recent" : array of usernames recently modified
gremmie@1 50 """
bgneal@266 51 if not request.user.is_authenticated():
bgneal@266 52 return HttpResponseForbidden('You must be logged in.')
gremmie@1 53
bgneal@266 54 # Do we have all the JSON cached?
bgneal@266 55 s = cache.get(CACHE_KEY)
bgneal@266 56 if s:
gremmie@1 57 return HttpResponse(s, content_type='application/json')
gremmie@1 58
bgneal@266 59 # Compute JSON for the map
bgneal@791 60 entries = MapEntry.objects.select_related('user', 'user__profile').\
bgneal@791 61 order_by('user__username')
bgneal@266 62 users = []
bgneal@791 63 avatar_urls = []
bgneal@266 64 recent = []
bgneal@266 65 for entry in entries.iterator():
bgneal@266 66 users.append(dict(name=entry.user.username,
bgneal@266 67 lat=entry.lat,
bgneal@266 68 lon=entry.lon,
bgneal@266 69 message=entry.html,
bgneal@266 70 ))
bgneal@791 71 avatar = entry.user.profile.avatar
bgneal@791 72 if avatar and avatar.url:
bgneal@791 73 avatar_urls.append(avatar.url)
bgneal@791 74 else:
bgneal@791 75 avatar_urls.append(None)
bgneal@266 76 recent.append((entry.date_updated, entry.user.username))
bgneal@266 77
bgneal@266 78 # Render the messages that go in the balloons
bgneal@791 79 for user, avatar_url in zip(users, avatar_urls):
bgneal@266 80 user['message'] = render_to_string('membermap/balloon.html',
bgneal@791 81 dict(user=user, avatar_url=avatar_url))
bgneal@266 82
bgneal@266 83 # Produce the list of recent updates
bgneal@266 84 recent.sort(reverse=True)
bgneal@266 85 del recent[10:]
bgneal@266 86 recent = [entry[1] for entry in recent]
bgneal@266 87
bgneal@266 88 # Create the JSON for the map
bgneal@266 89 result = dict(users=users, recent=recent)
bgneal@266 90 s = json.dumps(result, ensure_ascii=False)
bgneal@266 91
bgneal@266 92 cache.set(CACHE_KEY, s, CACHE_TIMEOUT)
bgneal@266 93 return HttpResponse(s, content_type='application/json')
gremmie@1 94
gremmie@1 95
gremmie@1 96 @require_POST
gremmie@1 97 def add(request):
gremmie@1 98 """
gremmie@1 99 This view is called by AJAX to add/update the user to the map.
gremmie@1 100 It returns the new JSON representation of the user.
gremmie@1 101 """
gremmie@1 102 if not request.user.is_authenticated():
gremmie@1 103 return HttpResponseForbidden('You must be logged in.')
gremmie@1 104
gremmie@1 105 loc = request.POST.get('loc', None)
gremmie@1 106 lat = request.POST.get('lat', None)
gremmie@1 107 lon = request.POST.get('lon', None)
bgneal@1024 108 msg = request.POST.get('msg', '').strip()
gremmie@1 109
gremmie@1 110 if loc is None or lat is None or lon is None:
gremmie@1 111 return HttpResponseBadRequest('Missing parameters')
gremmie@1 112
gremmie@1 113 try:
gremmie@1 114 lat = float(lat)
gremmie@1 115 lon = float(lon)
gremmie@1 116 except ValueError:
gremmie@1 117 return HttpResponseBadRequest('Invalid lat/lon')
gremmie@1 118
bgneal@1024 119 html = ''
bgneal@1024 120 if msg:
bgneal@1024 121 html = site_markup(msg)
bgneal@1024 122 try:
bgneal@1024 123 image_check(html)
bgneal@1024 124 except ImageCheckError as ex:
bgneal@1024 125 return HttpResponseBadRequest(str(ex))
bgneal@1024 126
gremmie@1 127 try:
gremmie@1 128 entry = MapEntry.objects.get(user=request.user)
gremmie@1 129 except MapEntry.DoesNotExist:
gremmie@1 130 entry = MapEntry(user=request.user)
gremmie@1 131
gremmie@1 132 entry.location = loc
gremmie@1 133 entry.lat = lat
gremmie@1 134 entry.lon = lon
gremmie@1 135 entry.message = msg
bgneal@1024 136 entry.save(html=html)
gremmie@1 137
bgneal@266 138 cache.delete(CACHE_KEY)
bgneal@266 139
bgneal@266 140 avatar_url = None
bgneal@789 141 profile = entry.user.profile
bgneal@266 142 if profile.avatar and profile.avatar.url:
bgneal@266 143 avatar_url = profile.avatar.url
bgneal@266 144
bgneal@266 145 u = dict(name=entry.user.username,
bgneal@266 146 lat=entry.lat,
bgneal@266 147 lon=entry.lon,
bgneal@266 148 message=entry.html)
bgneal@266 149
bgneal@266 150 u['message'] = render_to_string('membermap/balloon.html',
bgneal@266 151 dict(user=u, avatar_url=avatar_url))
bgneal@266 152
bgneal@266 153 result = json.dumps(u, ensure_ascii=False)
bgneal@266 154 return HttpResponse(result, content_type='application/json')
gremmie@1 155
gremmie@1 156
gremmie@1 157 @require_POST
gremmie@1 158 def delete(request):
gremmie@1 159 """
gremmie@1 160 This view is called by AJAX to delete the user from the map.
gremmie@1 161 """
gremmie@1 162 if not request.user.is_authenticated():
gremmie@1 163 return HttpResponseForbidden('You must be logged in.')
gremmie@1 164
gremmie@1 165 try:
gremmie@1 166 entry = MapEntry.objects.get(user=request.user)
gremmie@1 167 except MapEntry.DoesNotExist:
gremmie@1 168 pass
gremmie@1 169 else:
gremmie@1 170 entry.delete()
bgneal@266 171 cache.delete(CACHE_KEY)
gremmie@1 172
gremmie@1 173 return HttpResponse('')