pbp_com/apps/almanac/views.py

import csv
import pywapi
from datetime import datetime, date, timedelta
from dateutil.relativedelta import relativedelta
from dateutil.parser import parse

from django import http
from django.template.loader import get_template
from django.template import Context
import ho.pisa as pisa
import cStringIO as StringIO
import cgi

from django.http import HttpResponse
from django.template.context import RequestContext
from django.views.generic import list_detail
from django.shortcuts import render_to_response

from almanac.models import (
    Tide,
    TideDay,
    TideChart,
    Day,
    WeatherConditions,
    WeatherAlert,
    WeatherForecast,
)
from darkroom.models import Webcam
from directory.models import Place, Point, Town


def index(request):
    webcams = Webcam.objects.all()
    towns = Town.objects.filter(in_coverage=True)
    town_weather = []
    for t in towns:
        if t.slug == "castine" or t.slug == "stonington" or t.slug == "blue-hill":
            town_weather.append(pywapi.get_weather_from_yahoo(t.zipcode, "imperial"))

    if datetime.now().hour > 20 or datetime.now().hour < 4:
        moon = "moon"
    else:
        moon = ""

    return render_to_response(
        "almanac/index.html", locals(), context_instance=RequestContext(request)
    )


def tide_index(request):
    charts = TideChart.objects.filter(public=True).order_by("-year")
    months = [
        "jan",
        "feb",
        "mar",
        "apr",
        "may",
        "jun",
        "jul",
        "sep",
        "oct",
        "nov",
        "dec",
    ]
    return render_to_response(
        "almanac/tide_index.html", locals(), context_instance=RequestContext(request)
    )


def tide_week_detail(request, slug, year, month, day):
    chart = TideChart.objects.get(place__slug=slug, year__exact=year)
    start_date = parse(str(year) + "-" + month + "-" + str(day))
    end_date = start_date + timedelta(days=3)
    first_day_set = (
        TideDay.objects.filter(tide_chart__place__slug__exact=slug)
        .exclude(day__date__gt=end_date)
        .filter(day__date__gte=start_date)
    )

    second_start_date = start_date + timedelta(days=4)
    second_end_date = second_start_date + timedelta(days=3)
    second_day_set = (
        TideDay.objects.filter(tide_chart__place__slug__exact=slug)
        .exclude(day__date__gt=second_end_date)
        .filter(day__date__gte=second_start_date)
    )
    return render_to_response(
        "almanac/tide_week_detail.html",
        locals(),
        context_instance=RequestContext(request),
    )


def tide_month_detail(request, slug, year, month):
    chart = TideChart.objects.get(place__slug=slug, year__exact=year)
    days = TideDay.objects.filter(
        tide_chart__place__slug__exact=slug,
        day__date__year=year,
        day__date__month=parse(month).month,
    )
    full_date = datetime(int(year), parse(month).month, 1)
    month = month.capitalize()
    today = date.today()
    return render_to_response(
        "almanac/tide_month_detail.html",
        locals(),
        context_instance=RequestContext(request),
    )


def tide_year_detail(request, slug, year):
    astro_days = Day.objects.filter(chart__year__exact=year)
    chart = TideChart.objects.get(place__slug=slug, year__exact=year)
    return render_to_response(
        "almanac/tide_year_detail.html",
        locals(),
        context_instance=RequestContext(request),
    )


def seasonal_guide(request, slug, year, month, day):
    today = datetime.today()
    chart = TideChart.objects.get(place__slug=slug, year__exact=year)
    start_date = parse(str(year) + "-" + month + "-" + str(day))
    end_date = start_date + relativedelta(months=+1, days=-1)

    days = (
        TideDay.objects.filter(tide_chart__place__slug=slug)
        .exclude(day__date__gt=end_date)
        .filter(day__date__gte=start_date)
    )
    n = len(days) / 2
    first_day_set = days[0:n]
    second_day_set = days[n:]

    # end_date=start_date+timedelta(days=13)
    # first_day_set=TideDay.objects.filter(tide_chart__place__slug__exact=slug).exclude(
    #                       day__date__gt=end_date).filter(
    #                       day__date__gte=start_date)

    # second_start_date=start_date+timedelta(days=14)
    # second_end_date=second_start_date+timedelta(days=13)
    # second_day_set=TideDay.objects.filter(tide_chart__place__slug__exact=slug).exclude(
    #                       day__date__gt=second_end_date).filter(
    #                       day__date__gte=second_start_date)
    noon = datetime(today.year, today.month, today.day, 12, 0)
    return render_to_response(
        "almanac/tide_seasonal_guide.html",
        locals(),
        context_instance=RequestContext(request),
    )


def weekly_dump(request, slug, year, month, day):
    response = HttpResponse(mimetype="text/csv")
    response["Content-Disposition"] = "attachment; filename=tides.csv"
    chart = TideChart.objects.get(place__slug=slug, year__exact=year)
    start_date = parse(str(year) + "-" + month + "-" + str(day))
    end_date = start_date + timedelta(days=7)
    days = (
        TideDay.objects.filter(tide_chart__place__slug=slug)
        .exclude(day__date__gt=end_date)
        .filter(day__date__gte=start_date)
    )

    writer = csv.writer(response)
    for tday in days.all():
        day = tday.day
        tide1 = tday.tide_1
        tide2 = tday.tide_2
        tide3 = tday.tide_3
        tide4 = tday.tide_4
        if tide4:
            writer.writerow(
                [
                    tday.day.date,
                    tide1,
                    tide1.state,
                    tide1.level,
                    tide2,
                    tide2.state,
                    tide2.level,
                    tide3,
                    tide3.state,
                    tide3.level,
                    tide4,
                    tide4.state,
                    tide4.level,
                    day.moonphase,
                    day.moonphase_time,
                    day.moonrise,
                    day.moonset,
                    day.sunrise,
                    day.sunset,
                ]
            )
        else:
            writer.writerow(
                [
                    tday.day.date,
                    tide1,
                    tide1.state,
                    tide1.level,
                    tide2,
                    tide2.state,
                    tide2.level,
                    tide3,
                    tide3.state,
                    tide3.level,
                    "",
                    "",
                    "",
                    day.moonphase,
                    day.moonphase_time,
                    day.moonrise,
                    day.moonset,
                    day.sunrise,
                    day.sunset,
                ]
            )
    return response


def write_pdf(template_src, context_dict):
    template = get_template(template_src)
    context = Context(context_dict)
    html = template.render(context)
    result = StringIO.StringIO()
    pdf = pisa.pisaDocument(StringIO.StringIO(html.encode("UTF-8")), result)
    if not pdf.err:
        return http.HttpResponse(result.getvalue(), mimetype="application/pdf")
    return http.HttpResponse("Gremlin's ate your pdf! %s" % cgi.escape(html))


def seasonal_guide_pdf(request, slug, year, month, day):
    today = datetime.today()
    chart = TideChart.objects.get(place__slug=slug, year__exact=year)
    start_date = parse(str(year) + "-" + month + "-" + str(day))
    end_date = start_date + relativedelta(months=+1, days=-1)

    days = (
        TideDay.objects.filter(tide_chart__place__slug=slug)
        .exclude(day__date__gt=end_date)
        .filter(day__date__gte=start_date)
    )
    n = len(days) / 2
    first_day_set = days[0:n]
    second_day_set = days[n:]

    noon = datetime(today.year, today.month, today.day, 12, 0)

    return write_pdf(
        "almanac/pdf/seasonal_guide.html",
        {
            "pagesize": "letter",
            "today": today,
            "noon": noon,
            "chart": chart,
            "days": days,
            "first_day_set": first_day_set,
            "second_day_set": second_day_set,
        },
    )