################
#  models.py   #
################

from django.db import models
from django.contrib.auth.models import User
from datetime import datetime

class StockTickerSymbol(models.Model):
    StockName = models.CharField(max_length=7, unique=True, default="str8 chillin'")
    user = models.ForeignKey(User)
    added_date = models.DateTimeField('Date Added', default=datetime.now)

    def __unicode__(self):
        return self.StockName

################
#   views.py   #
################

from django.shortcuts import render
from django.contrib.auth.decorators import login_required
from django.shortcuts import render_to_response
from django.template import RequestContext
from favorites.models import StockTickerSymbol
from favorites.forms import FavoritesForm

@login_required(login_url='/login/')
def fav(request):
    context = RequestContext(request)
    favorites_list = StockTickerSymbol.objects.filter(user=request.user).order_by('-added_date')

    if request.method == 'POST':
        form = FavoritesForm(request.POST)

        if form.is_valid():
            form.save(commit=True)
            return fav(request)
        else:
            print form.errors

    else:
        form = FavoritesForm()


    context_dict = {'favorites': favorites_list, 'form':form}

    return render_to_response('favorites/favorites.html', context_dict, context)


################
#   forms.py   #
################

from django import forms
from favorites.models import StockTickerSymbol

class FavoritesForm(forms.ModelForm):
    class Meta:
        model = StockTickerSymbol
        fields = ('StockName', 'user',)


################
#   forms.py   #
################

from django.contrib import admin
from django.contrib.auth.models import User
from favorites.models import StockTickerSymbol

class StockTickerSymbolAdmin(admin.ModelAdmin):

    #This bit of code allows me to succesfully get the default user, but only through the Django admin
    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        if db_field.name == 'user':
            kwargs['initial'] = request.user.id
        return super(StockTickerSymbolAdmin, self).formfield_for_foreignkey(
            db_field, request, **kwargs
        )

admin.site.register(StockTickerSymbol, StockTickerSymbolAdmin)