#!/usr/bin/env python

"""
This script imports the popcon data into the database
See http://popcon.debian.org/
"""

import aux
import sys
import gzip
from gatherer import gatherer

def get_gatherer(connection, config):
  return popcon_gatherer(connection, config)

class popcon_gatherer(gatherer):
  def __init__(self, connection, config):
    gatherer.__init__(self, connection, config)

  def run(self, source):
    try:
      my_config = self.config[source]
    except:
      raise

    if not 'path' in my_config:
      raise aux.ConfigException, "path not configured for source " + source

    if not 'distribution' in my_config:
      raise aux.ConfigException, "distribution not configured for source " + source

    cur = self.cursor()

    cur.execute("PREPARE pop_insert AS INSERT INTO popcon (name, vote, olde, recent, nofiles, distribution) VALUES ($1, $2, $3, $4, $5, '%s')" % my_config['distribution'])

    popcon = gzip.open(my_config['path'])

    cur.execute("DELETE FROM popcon WHERE distribution = '%s'" % my_config['distribution'])

    for line in popcon.readlines():
      name, data = line.split(None, 1)
      if name == "Submissions:":
	cur.execute("INSERT INTO popcon (name, vote, distribution) VALUES ('_submissions', %s, '%s')" % (data, my_config['distribution']))
      try:
	(name, vote, old, recent, nofiles) = data.split()
	query = "EXECUTE pop_insert('%s', %s, %s, %s, %s)" %\
	    (name, vote, old, recent, nofiles)
	cur.execute(query)
      except ValueError:
	continue

    cur.execute("DEALLOCATE pop_insert")

if __name__ == '__main__':
  main()