#!/usr/bin/env python
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.


import sys
import logging
import datetime
import psycopg2

from socorro.lib.datetimeutil import utc_now

logger = logging.getLogger('dailyMatviews')
logger.addHandler(logging.StreamHandler(sys.stderr))


def update(config, targetDate):
    functions = (
      # function name, parmeters, dependencies
      ('update_product_versions', [], []),
      ('update_signatures', [targetDate], []),
      ('update_os_versions', [targetDate], []),
      ('update_tcbs', [targetDate], []),
      ('update_adu', [targetDate], []),
      ('update_rank_compare', [targetDate], []),
      ('update_nightly_builds', [targetDate], []),
      ('update_build_adu', [targetDate], []),
      ('update_crashes_by_user', [targetDate], ['update_adu']),
      ('update_crashes_by_user_build', [targetDate], []),
      ('update_correlations', [targetDate], []),
      ('update_home_page_graph', [targetDate], ['update_adu']),
      ('update_home_page_graph_build', [targetDate], ['update_build_adu']),
      ('update_tcbs_build', [targetDate], []),
      ('update_explosiveness', [targetDate],
       ['update_tcbs', 'update_build_adu']),
    )

    failed = set()
    # returning false is ok in some cases, see bug 790022
    failureOk = ('update_adu',)
    databaseDSN = ""
    if 'databaseHost' in config:
        databaseDSN += 'host=%(databaseHost)s '
    if 'databaseName' in config:
        databaseDSN += 'dbname=%(databaseName)s '
    if 'databaseUserName' in config:
        databaseDSN += 'user=%(databaseUserName)s '
    if 'databasePassword' in config:
        databaseDSN += 'password=%(databasePassword)s'
    dsn = databaseDSN % config
    connection = psycopg2.connect(dsn)
    cursor = connection.cursor()
    for funcname, parameters, deps in functions:
        if set(deps) & failed:
            # one of the deps previously failed, so skip this one
            logger.warn("For %r, dependency %s failed so skipping"
                         % (funcname, ', '.join(set(deps) & failed)))
            continue
        logger.info('Running %s' % funcname)
        failureMessage = None
        success = False
        # Set the temp_buffers for this session
        databaseTempbuffers = '8MB' # default
        if 'databaseTempbuffers' in config:
            databaseTempbuffers = config.databaseTempbuffers
        cursor.execute(""" SET TEMP_BUFFERS = %s """, databaseTempbuffers);
        try:
            cursor.callproc(funcname, parameters)
            # fetchone() returns a tuple of length 1
            result = cursor.fetchone()
            if result and result[0] or funcname in failureOk:
                success = True
            else:
                # "expected" error
                logger.warn('%r failed' % funcname)
                failureMessage = '%s did not return true' % funcname
        except psycopg2.InternalError:
            # unexpected error
            logger.error('%r failed' % funcname, exc_info=True)
            import sys  # don't assume that this has been imported
            __, error_value = sys.exc_info()[:2]
            failureMessage = str(error_value)
        if success:
            connection.commit()
        else:
            connection.rollback()
            failed.add(funcname)

    return len(failed)
