223 lines
6.0 KiB
Python
223 lines
6.0 KiB
Python
import datetime
|
|
import hashlib
|
|
import os
|
|
import stripe
|
|
|
|
from accounts.models import UserProfile
|
|
from annoying.decorators import ajax_request, render_to
|
|
from blogs.views import blog_list
|
|
from django.conf import settings
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.contrib.auth.models import User
|
|
from django.core.mail import send_mail
|
|
from django.db.models import Count
|
|
from django.http import HttpResponseRedirect, HttpResponseBadRequest
|
|
from django.shortcuts import render_to_response
|
|
from django.template import RequestContext
|
|
from snipts.models import Snipt
|
|
from snipts.utils import get_lexers_list
|
|
from taggit.models import Tag
|
|
|
|
|
|
@render_to('for-teams.html')
|
|
def for_teams(request):
|
|
if request.user.is_authenticated():
|
|
profile = request.user.profile
|
|
profile.teams_beta_seen = True
|
|
profile.save()
|
|
return {}
|
|
|
|
|
|
@render_to('for-teams-complete.html')
|
|
def for_teams_complete(request):
|
|
|
|
if request.method == 'POST':
|
|
|
|
if request.user.is_authenticated():
|
|
name = request.POST['name']
|
|
members = request.POST['members']
|
|
info = request.POST['info']
|
|
send_mail('[Snipt] New Snipt for Teams beta request.', """
|
|
User: %s (%s)
|
|
Team name: %s
|
|
Team members: %s
|
|
Info:
|
|
|
|
%s
|
|
""" % (request.user.username, request.user.email, name, members,
|
|
info), 'support@snipt.net',
|
|
['nick@nicksergeant.com'], fail_silently=False)
|
|
|
|
profile = request.user.profile
|
|
profile.teams_beta_applied = True
|
|
profile.save()
|
|
else:
|
|
username = request.POST['username']
|
|
email = request.POST['email']
|
|
name = request.POST['name']
|
|
members = request.POST['members']
|
|
info = request.POST['info']
|
|
send_mail('[Snipt] New Snipt for Teams beta request.', """
|
|
User: %s (%s) (not authenticated)
|
|
Team name: %s
|
|
Team members: %s
|
|
Info:
|
|
|
|
%s
|
|
""" % (username, email, name, members, info), 'support@snipt.net',
|
|
['nick@nicksergeant.com'], fail_silently=False)
|
|
|
|
return {}
|
|
|
|
else:
|
|
return HttpResponseBadRequest()
|
|
|
|
|
|
@render_to('homepage.html')
|
|
def homepage(request):
|
|
|
|
if request.blog_user:
|
|
return blog_list(request)
|
|
|
|
coders = []
|
|
|
|
users_with_gravatars = User.objects.filter(
|
|
userprofile__in=UserProfile.objects.filter(has_gravatar=True)
|
|
).order_by('?')
|
|
|
|
for user in users_with_gravatars:
|
|
public_snipts_count = Snipt.objects.filter(
|
|
user=user, public=True).values('pk').count()
|
|
|
|
if public_snipts_count:
|
|
user.email_md5 = hashlib.md5(user.email.lower()).hexdigest()
|
|
coders.append(user)
|
|
|
|
if len(coders) == 35:
|
|
break
|
|
|
|
return {
|
|
'coders': coders,
|
|
'snipts_count': Snipt.objects.all().count(),
|
|
'users_count': User.objects.all().count(),
|
|
}
|
|
|
|
|
|
@ajax_request
|
|
def lexers(request):
|
|
lexers = get_lexers_list()
|
|
objects = []
|
|
|
|
for l in lexers:
|
|
|
|
try:
|
|
filters = l[2]
|
|
except IndexError:
|
|
filters = []
|
|
|
|
try:
|
|
mimetypes = l[3]
|
|
except IndexError:
|
|
mimetypes = []
|
|
|
|
objects.append({
|
|
'name': l[0],
|
|
'lexers': l[1],
|
|
'filters': filters,
|
|
'mimetypes': mimetypes
|
|
})
|
|
|
|
return {'objects': objects}
|
|
|
|
|
|
def login_redirect(request):
|
|
if request.user.is_authenticated():
|
|
return HttpResponseRedirect('/' + request.user.username + '/')
|
|
else:
|
|
return HttpResponseRedirect('/')
|
|
|
|
|
|
@login_required
|
|
@render_to('pro.html')
|
|
def pro(request):
|
|
if request.user.profile.is_pro:
|
|
return HttpResponseRedirect('/' + request.user.username + '/')
|
|
return {}
|
|
|
|
|
|
@login_required
|
|
@render_to('pro-complete.html')
|
|
def pro_complete(request):
|
|
|
|
if request.method == 'POST':
|
|
|
|
token = request.POST['token']
|
|
stripe.api_key = os.environ.get('STRIPE_SECRET_KEY',
|
|
settings.STRIPE_SECRET_KEY)
|
|
|
|
if 'plan' in request.GET:
|
|
plan = request.GET['plan']
|
|
else:
|
|
plan = request.POST['plan']
|
|
|
|
try:
|
|
customer = stripe.Customer.create(card=token,
|
|
plan=plan,
|
|
email=request.user.email)
|
|
except stripe.CardError, e:
|
|
error_message = e.json_body['error']['message']
|
|
return HttpResponseRedirect('/pro/?declined=%s' % error_message or
|
|
'Your card was declined.')
|
|
|
|
profile = request.user.profile
|
|
profile.is_pro = True
|
|
profile.pro_date = datetime.datetime.now()
|
|
profile.stripe_id = customer.id
|
|
profile.save()
|
|
|
|
return {}
|
|
|
|
else:
|
|
return HttpResponseBadRequest()
|
|
|
|
|
|
def sitemap(request):
|
|
|
|
tags = Tag.objects.filter(snipt__public=True)
|
|
tags = tags.annotate(count=Count('taggit_taggeditem_items__id'))
|
|
tags = tags.order_by('-count')[:1000]
|
|
|
|
return render_to_response('sitemap.xml',
|
|
{'tags': tags},
|
|
context_instance=RequestContext(request),
|
|
content_type='application/xml')
|
|
|
|
|
|
@render_to('tags.html')
|
|
def tags(request):
|
|
|
|
all_tags = Tag.objects.filter(snipt__public=True).order_by('name')
|
|
all_tags = all_tags.annotate(count=Count('taggit_taggeditem_items__id'))
|
|
|
|
popular_tags = Tag.objects.filter(snipt__public=True)
|
|
popular_tags = popular_tags.annotate(
|
|
count=Count('taggit_taggeditem_items__id'))
|
|
popular_tags = popular_tags.order_by('-count')[:20]
|
|
popular_tags = sorted(popular_tags, key=lambda tag: tag.name)
|
|
|
|
return {
|
|
'all_tags': all_tags,
|
|
'tags': popular_tags
|
|
}
|
|
|
|
|
|
@ajax_request
|
|
def user_api_key(request):
|
|
|
|
if not request.user.is_authenticated():
|
|
return HttpResponseBadRequest()
|
|
|
|
return {
|
|
'api_key': request.user.api_key.key
|
|
}
|