import helpers
from config import render, db

class Repo(object):
    def GET(self):
        repo_count = db.query('SELECT REPO,COUNT(DISTINCT IPKEY) AS PACKAGES,\
                COUNT(DISTINCT UUID) AS HOSTS\
                FROM INSTALLED_PACKAGES NATURAL JOIN REPOSITORIES GROUP BY REPO')
        repo_data = dict()
        for t in repo_count:
            repo_data[t['REPO']] = {'HOSTS':t['HOSTS'], 'PACKAGES':t['PACKAGES']}
        if helpers.is_json_request():
            return helpers.serialize(repo_data)
        else:
            return render.repo(repo_data)