
#  IATI Updates, IATI Registry API augmented
#  by Mark Brough
#
#  Copyright (C) 2013 Publish What You Fund
#
#  This programme is free software; you may redistribute and/or modify
#  it under the terms of the GNU Affero General Public License v3.0

from flask import Flask, render_template, flash, request, Markup, \
    session, redirect, url_for, escape, Response, abort, send_file

from iatiupdates import app
from iatiupdates import db
import registry
import os

@app.route("/")
def home():
    num_publishers = registry.getNumRealPublishers()
    frequency_totals = registry.getFrequencyTotals()
    frequencies = registry.FREQUENCIES
    return render_template("dashboard.html",
            num_publishers=num_publishers,
            frequency_totals=frequency_totals,
            frequencies=frequencies)

@app.route("/freq/")
def freq():
    return registry.calculate_frequency()

@app.route("/publisher/")
def publisher():
    publishers = registry.publishers()
    frequencies = registry.FREQUENCIES
    return render_template("publisher.html", 
                    publishers=publishers,
                    frequencies=frequencies,
                    orgtypes=dict(registry.ORGANIZATION_TYPES))

@app.route("/publisher/<packagegroup_name>/updates/")
def publisher_updates(packagegroup_name):
    publisher = registry.publishers(packagegroup_name)
    updated_dates = registry.getUpdatedDates(packagegroup_name)
    return render_template("publisher_updates.html",
            publisher = publisher,
            updated_dates = updated_dates)

@app.route("/package/")
def package():
    packages = registry.packages()
    return render_template("package.html", 
                    packages=packages)

@app.route("/revision/")
def revision():
    revisions = registry.revisions()
    return render_template("revision.html", 
                    revisions=revisions)

@app.route("/update/")
def update():
    print "getting packagegroups..."
    registry.get_packagegroups()
    print "getting packages..."
    registry.get_packages()
    print "getting revisions..."
    registry.get_revisions()
    return "Complete"

@app.route("/update/revisions/")
def update_revisions():
    print "getting revisions..."
    registry.get_revisions()
    return "Complete"

@app.route("/update/packagegroups/")
def update_packagegroups():
    print "getting packagegroups..."
    registry.get_packagegroups()
    return "Complete"

@app.route("/update/packages/")
def update_packages():
    print "getting packages..."
    registry.get_packages()
    return "Complete"

@app.errorhandler(404)
def page_not_found(error):
    return render_template('404.html'), 404

"""@app.errorhandler(500)
def internal_server_error(e):
    return render_template('500.html'), 500"""

@@ -1263,148 +1263,175 @@
 /%22)%0A
-def publisher_updates(packagegroup_name):%0A    publisher = registry.publishers(packagegroup_name)%0A    updated_dates = registry.getUpdatedDate
+@app.route(%22/publisher/%3Cpackagegroup_name%3E/updates/%3Cmessage_method%3E%22)%0Adef publisher_updates(packagegroup_name, message_method=None):%0A    publisher = registry.publisher
 s(pa
@@ -1551,37 +1551,39 @@
     
-updated_dates = updated_dates
+message_method = message_method
 )%0A%0A@

Allow message method to be provided, don't need to query updated dates here