# Last-Modified: <Sat Jul 26 18:20:06 2008>
# This file is part of the Ultimate Debian Database Project

from gatherer import gatherer
import aux
from glob import glob
import gzip
import psycopg2
import sys

date_translation = {
    'Deb': 'Feb',
    'Augl': 'Aug',
    'Fev': 'Feb' }

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

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

  def run(self, source):
    if not 'path' in self.config[source]:
      raise aux.ConfigException('path not specified for source ' + source)
    path = self.config[source]['path']

    cursor = self.cursor()

    cursor.execute("DELETE FROM upload_history")

    cursor.execute("PREPARE uh_insert AS INSERT INTO upload_history VALUES \
	($1, $2, $3, $4, $5, $6, $7, $8)")

    for name in glob(path + '/debian-devel-*'):
      print name
      f = None
      if name.endswith(".gz"):
	f = gzip.open(name)
      else:
	f = open(name)
      
      current = {}
      last_field = None
      line_count = 0
      for line in f.readlines():
	line_count += 1
	line = line.strip()
	# Stupid multi-line maintainer fields *grml*
	if line == '':
	  query = "EXECUTE uh_insert(%(Source)s, %(Version)s, %(Date)s, %(Changed-By)s, \
	      %(Maintainer)s, %(NMU)s, %(Key)s, %(Signed-By)s)" % current
	  try:
	    cursor.execute(query)
	  except psycopg2.ProgrammingError, s:
	    print "Error at line %d of file %s" % (line_count, name)
	    raise
	  current = {}
	  last_field = None
	  continue

	if line.find(':') == -1:
	  if not last_field:
	    raise Exception, "Format error on line " + line_count + "of file " + name
	  current[last_field] += line
	  continue


	(field, data) = line.split(':', 1)
	data = data.strip()

	if field != 'NMU':
	  current[field] = aux.quote(data)
	else:
	  current[field] = data
	
	last_field = field
    
    cursor.execute("DEALLOCATE uh_insert")

    sys.exit(1)