Ben Homework

.gitignore

@@ -68,3 +68,5 @@ docs/_build/
 # PyBuilder
 target/
 
+# PyCharm
+.idea/

Procfile

@@ -0,0 +1 @@
+web: gunicorn urly_bird.wsgi --log-file -

requirements.txt

@@ -1,8 +1,27 @@
+dj-database-url==0.3.0
+dj-static==0.0.6
 Django==1.8.2
 django-bootstrap3==5.4.0
 django-debug-toolbar==1.3.0
 django-extensions==1.5.5
+django-filter==0.10.0
+django-toolbelt==0.0.1
+djangorestframework==3.1.3
 fake-factory==0.5.1
+gnureadline==6.3.3
+gunicorn==19.3.0
+hashID==3.1.4
+hashids==1.1.0
 ipython==3.1.0
+matplotlib==1.4.3
+nose==1.3.7
+numpy==1.9.2
+pandas==0.16.2
 psycopg2==2.6.1
+pyparsing==2.0.3
+python-dateutil==2.4.2
+pytz==2015.4
+six==1.9.0
+sqlparse==0.1.15
+static3==0.6.1
 Werkzeug==0.10.4

runtime.txt

@@ -0,0 +1 @@
+python-3.4.3

urly_bird/accounts/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

urly_bird/accounts/forms.py

@@ -0,0 +1,10 @@
+from django import forms
+from django.contrib.auth.models import User
+
+
+class UserForm(forms.ModelForm):
+    password = forms.CharField(widget=forms.PasswordInput())
+
+    class Meta:
+        model = User
+        fields = ('username', 'email', 'password',)

urly_bird/accounts/models.py

@@ -0,0 +1,3 @@
+from django.db import models
+
+# Create your models here.

urly_bird/accounts/templates/registration/login.html

@@ -0,0 +1,31 @@
+{% extends "base.html" %}
+{% load bootstrap3 %}
+
+{% block content %}
+
+    {% if form.errors %}
+        <p>Your username and password didn't match. Please try again.</p>
+    {% endif %}
+
+    {% if next %}
+        {% if user.is_authenticated %}
+            <p>Your account doesn't have access to this page. To proceed,
+                please login with an account that has access.</p>
+        {% else %}
+            <p>Please login to see this page.</p>
+        {% endif %}
+    {% endif %}
+    <div class="container">
+        <form method="post" action="{% url 'django.contrib.auth.views.login' %}">
+            {% csrf_token %}
+            <div class="form-group form-custom-narrow">
+                {% bootstrap_form form %}
+            </div>
+            <input type="submit" class="btn btn-default" value="login"/>
+            <input type="hidden" name="next" value="{{ next }}"/>
+        </form>
+    </div>
+    {# Assumes you setup the password_reset view in your URLconf #}
+    {#<p><a href="{% url 'password_reset' %}">Lost password?</a></p>#}
+
+{% endblock %}

urly_bird/accounts/templates/registration/register.html

@@ -0,0 +1,18 @@
+{% extends "base.html" %}
+{% load bootstrap3 %}
+{% block content %}
+    <h2>Register</h2>
+
+    <div class="container">
+        <form method="POST" action="{% url 'user_register' %}">
+
+            {% csrf_token %}
+            <div class="form-group form-custom-narrow">
+                {% bootstrap_form user_form %}
+            </div>
+            <input type="submit" class="btn btn-default" value="Register"/>
+            <input type="hidden" name="next" value="{{ next }}"/>
+        </form>
+    </div>
+
+{% endblock %}

urly_bird/accounts/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

urly_bird/accounts/views.py

@@ -0,0 +1,30 @@
+from django.shortcuts import render, redirect
+from django.contrib.auth import authenticate, login
+from django.contrib import messages
+from accounts.forms import UserForm
+
+# Create your views here.
+def register_rater(request):
+    if request.method == "POST":
+        user_form = UserForm(request.POST)
+        if user_form.is_valid():
+            user = user_form.save()
+
+            password = user.password
+            user.set_password(password)
+            user.save()
+
+            user = authenticate(username=user.username,
+                                password=password)
+
+            login(request, user)
+
+            messages.add_message(
+                request,
+                messages.SUCCESS,
+                "Welcome, {}. You have successfully created an account and are now logged in".format(user.username))
+
+            return redirect("all_bookmarks")
+    else:
+        user_form = UserForm()
+    return render(request, "registration/register.html", {'user_form': user_form})

urly_bird/api/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

urly_bird/api/models.py

@@ -0,0 +1,3 @@
+from django.db import models
+
+# Create your models here.

urly_bird/api/permissions.py

@@ -0,0 +1,20 @@
+from rest_framework import permissions
+
+
+class IsOwnerOrReadOnly(permissions.BasePermission):
+    def has_object_permission(self, request, view, obj):
+        if request.method in permissions.SAFE_METHODS:
+            return True
+        else:
+            return request.user == obj.user
+
+
+class IsSameUser(permissions.BasePermission):
+    def has_permission(self, request, view):
+        if request.method == "POST":
+            return request.user.is_anonymous()
+        else:
+            return True
+
+    def has_object_permission(self, request, view, obj):
+        return request.user == obj

urly_bird/api/serializers.py

@@ -0,0 +1,96 @@
+from django.contrib.auth import authenticate, login
+from django.contrib.auth.models import User
+from django.db.models import Count
+from django.utils import timezone
+from links.models import Bookmark, Click
+from rest_framework import serializers
+from hashids import Hashids
+from rest_framework.reverse import reverse
+
+
+class ClickSerializer(serializers.ModelSerializer):
+    user = serializers.PrimaryKeyRelatedField(read_only=True)
+    bookmark = serializers.PrimaryKeyRelatedField(read_only=True)
+    timestamp = serializers.SerializerMethodField()
+
+    class Meta:
+        model = Click
+
+    def create(self, validated_data):
+        """Automatically sets the timestamp for a bookmark."""
+        click = Click.objects.create(**validated_data)
+        click.timestamp = timezone.now()
+        click.save()
+        return click
+
+
+    def get_timestamp(self, obj):
+        return obj.timestamp
+
+
+#######################################################################################################################
+
+class BookmarkSerializer(serializers.HyperlinkedModelSerializer):
+    user = serializers.PrimaryKeyRelatedField(read_only=True)
+    api_url = serializers.HyperlinkedIdentityField(view_name='bookmark-detail')
+    code = serializers.SerializerMethodField()
+    timestamp = serializers.SerializerMethodField()
+    click_number = serializers.IntegerField(source='num_clicks', read_only=True)
+    click_count = serializers.SerializerMethodField()
+    _links = serializers.SerializerMethodField()
+
+    class Meta:
+        model = Bookmark
+        fields = (
+            'id', 'api_url', 'code', 'url', 'title', 'description', 'timestamp', 'user', 'click_number', 'click_count',
+            '_links')
+
+    def get_code(self, obj):
+        return obj.code
+
+    def get_timestamp(self, obj):
+        return obj.timestamp
+
+    def get__links(self, obj):
+        links = {
+            "clicks": reverse('create_click', kwargs=dict(bookmark_id=obj.id),
+                              request=self.context.get('request'))}
+        return links
+
+    def get_click_count(self, obj):
+        bookmark = Bookmark.objects.filter(pk=obj.id).annotate(click_num=Count('click'))
+        return bookmark[0].click_num
+
+    def create(self, validated_data):
+        """Automatically sets the short url code and the timestamp for a bookmark."""
+        bookmark = Bookmark.objects.create(**validated_data)
+        hashids = Hashids(salt="Hopefully the URLyBird does not get any worms")
+        code = hashids.encode(bookmark.id)
+        bookmark.code = code
+        bookmark.timestamp = timezone.now()
+        bookmark.save()
+        return bookmark
+
+
+#######################################################################################################################
+
+class UserSerializer(serializers.HyperlinkedModelSerializer):
+    total_clicks = serializers.IntegerField(source='num_clicks', read_only=True)
+    bookmark_set = BookmarkSerializer(many=True, read_only=True)
+
+    class Meta:
+        model = User
+        fields = ('id', 'url', 'username', 'email', 'total_clicks', 'bookmark_set', 'password')
+        write_only_fields = ('password',)
+
+    def create(self, validated_data):
+        user = super().create(validated_data)
+        user.set_password(validated_data['password'])
+        user.save()
+
+        user = authenticate(username=user.username,
+                            password=validated_data['password'])
+
+        login(self.context.get('request'), user)
+
+        return user

urly_bird/api/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

urly_bird/api/views.py

@@ -0,0 +1,58 @@
+from django.contrib.auth.models import User
+from django.db.models import Count
+from rest_framework import viewsets, permissions, generics, filters
+from api.serializers import BookmarkSerializer, ClickSerializer, UserSerializer
+from links.models import Bookmark, Click
+from api.permissions import IsOwnerOrReadOnly, IsSameUser
+import django_filters
+
+
+class BookmarkFilter(django_filters.FilterSet):
+    title = django_filters.CharFilter(name="title", lookup_type="icontains")
+    description = django_filters.CharFilter(name="description", lookup_type="icontains")
+    user = django_filters.NumberFilter(name='user', lookup_type="exact")
+
+    class Meta:
+        model = Bookmark
+        fields = ['title', 'description', 'user']
+
+
+class BookmarkViewSet(viewsets.ModelViewSet):
+    """Add the following search terms to the url to filter results:
+    title=your+title+search
+    description=your+description+search
+    user=your+user+id+search
+    Example: http://host/api/bookmarks/?title=nest&description=birds+live"""
+    queryset = Bookmark.objects.all().annotate(num_clicks=Count('click'))
+    serializer_class = BookmarkSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,
+                          IsOwnerOrReadOnly)
+    filter_backends = (filters.DjangoFilterBackend,)
+    filter_class = BookmarkFilter
+
+    def perform_create(self, serializer):
+        serializer.save(user=self.request.user)
+
+
+class ClickView(generics.ListCreateAPIView):
+    serializer_class = ClickSerializer
+
+    def get_queryset(self):
+        return Click.objects.filter(bookmark__id=self.kwargs['bookmark_id'])
+
+    def perform_create(self, serializer):
+        bookmark = Bookmark.objects.filter(pk=self.kwargs['bookmark_id'])[0]
+        if self.request.user.is_authenticated():
+            serializer.save(user=self.request.user, bookmark=bookmark)
+        else:
+            serializer.save(user=User.objects.get(pk=103), bookmark=bookmark)
+
+
+class UserViewSet(viewsets.ModelViewSet):
+    serializer_class = UserSerializer
+    permission_classes = (IsSameUser,)
+
+    def get_queryset(self):
+        queryset = User.objects.all().annotate(num_clicks=Count('bookmark__click')).filter(
+            pk=self.request.user.id)
+        return queryset

urly_bird/links/admin.py

@@ -0,0 +1,12 @@
+from django.contrib import admin
+from links.models import Bookmark, Click
+
+class BookmarkAdmin(admin.ModelAdmin):
+    list_display = ["id", "title", "description", "url", "code", "timestamp", "user"]
+
+class ClickAdmin(admin.ModelAdmin):
+    list_display = ["timestamp", "bookmark"]
+
+# Register your models here.
+admin.site.register(Bookmark, BookmarkAdmin)
+admin.site.register(Click, ClickAdmin)

urly_bird/links/forms.py

@@ -0,0 +1,20 @@
+from django import forms
+from links.models import Bookmark
+from django.utils import timezone
+
+
+
+class BookmarkForm(forms.ModelForm):
+    url = forms.URLField()
+    title = forms.CharField()
+    description = forms.CharField(widget=forms.Textarea, required=False)
+    code = forms.CharField(widget=forms.HiddenInput(), required=False)
+    timestamp = forms.DateTimeField(widget=forms.HiddenInput(), initial=timezone.now)
+
+    class Meta:
+        model = Bookmark
+        fields = ("url", "title", "description", "code", "timestamp")
+
+
+class EditBookmarkForm(BookmarkForm):
+    code = forms.CharField()