udd/udd/pts_gatherer.py

# Last-Modified: <Sun Aug 10 12:16:12 2008>

# This file is a part of the Ultimate Debian Database Project

from gatherer import gatherer
from aux import ConfigException, quote
from time import strptime

ZERO_DATE = '0000-01-01'

def get_gatherer(config, connection, source):
  return pts_gatherer(config, connection, source)


class pts_gatherer(gatherer):
  """This class imports PTS subscribers"""

  def __init__(self, connection, config, source):
    gatherer.__init__(self, connection, config, source)
    self.assert_my_config('path')

  def run(self):
    src_cfg = self.my_config

    c = self.connection.cursor()

    c.execute("DELETE FROM pts")

    c.execute("PREPARE pts_insert AS INSERT INTO pts (source, email) VALUES ($1, $2)")
      
    f = open(src_cfg['path'])
    for line in f:
      (package, subs) = line.split("\t")
      for sub in subs.split(", "):
        sub = sub.strip()
        c.execute("EXECUTE pts_insert(%s, %s)", (package, sub))

    c.execute("DEALLOCATE pts_insert")
    c.execute("ANALYZE pts")

1	# Last-Modified: <Sun Aug 10 12:16:12 2008>
2
3	# This file is a part of the Ultimate Debian Database Project
4
5	from gatherer import gatherer
6	from aux import ConfigException, quote
7	from time import strptime
8
9	ZERO_DATE = '0000-01-01'
10
11	def get_gatherer(config, connection, source):
12	return pts_gatherer(config, connection, source)
13
14
15	class pts_gatherer(gatherer):
16	"""This class imports PTS subscribers"""
17
18	def __init__(self, connection, config, source):
19	gatherer.__init__(self, connection, config, source)
20	self.assert_my_config('path')
21
22	def run(self):
23	src_cfg = self.my_config
24
25	c = self.connection.cursor()
26
27	c.execute("DELETE FROM pts")
28
29	c.execute("PREPARE pts_insert AS INSERT INTO pts (source, email) VALUES ($1, $2)")
30
31	f = open(src_cfg['path'])
32	for line in f:
33	(package, subs) = line.split("\t")
34	for sub in subs.split(", "):
35	sub = sub.strip()
36	c.execute("EXECUTE pts_insert(%s, %s)", (package, sub))
37
38	c.execute("DEALLOCATE pts_insert")
39	c.execute("ANALYZE pts")
40