#!/usr/bin/python

#TODO: Flickr photos out to check to see if captions have been updated

#TODO: MySQL warnings are just going to standard output right now

#TODO: non-existent attributes are set to the empty string to simplify MySQL...

#TODO: queries could be optimized by checking multiple entries for updates 

#   simulatenously (SELECT all entries that need to be updated...)

#TODO: check style

#TODO: Move MySQL connection info to another file, not under svn

from getopt import gnu_getopt

from sys import argv

from xml.dom.minidom import Document

import MySQLdb

import feed

import os.path, sys

#TODO: refactor this

# Specifies whether or not to force updates

force = False

# Controls how much to output

verbose = False

options, args = gnu_getopt(argv[1:], 'vf', ['force', 'verbose'])

for o, a in options:

  if o == '--force' or o == '-f':

    force = True

  elif o == '--verbose' or o == '-v':

    verbose = True

"""Parse a list to feeds to check for updates and populate a central database

for other applications to use.

"""

# Connect to database

db_settings_path = os.path.dirname(__file__) + '/config.txt'

if not os.path.isfile(db_settings_path):

  print 'ERROR: Please create config.txt for db settings'

  exit()

db_settings = {}

for line in open(db_settings_path):

  if not line.startswith(('#', '//')):

    setting = line.strip().split(' ',1)

    db_settings[setting[0]] = setting[1]

db_host = db_settings['db_host']

db_name = db_settings['db_name']

db_user = db_settings['db_user']

db_passwd = db_settings['db_passwd']

conn = MySQLdb.Connection(host=db_host, db=db_name, user=db_user,

  passwd=db_passwd, charset='utf8')

cursor = conn.cursor()

# Minidom object

doc = Document()

# Import the feeds

import_list = os.path.dirname(__file__) + '/feeds.list'

if os.path.isfile(import_list):

  for line in open(import_list):

    # Only act on non-empty, non-comment lines

    if line.strip() and not line.startswith(('#', '//')):

      # Split line into feed and attributes

      f = line.split()

      # Check for requested subclass

      if len(f) > 1 and hasattr(feed, f[1]) \

          and callable(getattr(feed, f[1])) \

          and issubclass(getattr(feed, f[1]), feed.Feed):

        this_type = getattr(feed, f[1])

      # Otherwise, use default object

      else:

        if verbose:

          print "Warning: class %s not found -- using standard Feed class" % (f[1])

        this_type = feed.Feed

      if verbose:

        print "Checking %s (%s)..." % (f[0], this_type.__name__)

      else:

        print "Checking %s..." % this_type.__name__

      this_feed = this_type(f[0])

      if not this_feed.parse():      

        sys.stderr.write(this_type.__name__ + ' updated failed.\n')

      this_feed.db_check(cursor, verbose, force)

else:

  print "No feeds to check."