# Copyright (C) 2000-2002  Bastian Kleineidam
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

from StandardLogger import StandardLogger
import time, linkcheck, linkcheck.Config
from linkcheck.log import strtime

class SQLLogger (StandardLogger):
    """ SQL output for PostgreSQL, not tested"""
    def __init__ (self, **args):
        apply(StandardLogger.__init__, (self,), args)
        self.dbname = args['dbname']
        self.separator = args['separator']

    def init (self):
        if self.fd is None: return
        self.starttime = time.time()
        if self.logfield("intro"):
            self.fd.write("-- "+(linkcheck._("created by %s at %s\n") % (linkcheck.Config.AppName,
                       strtime(self.starttime))))
            self.fd.write("-- "+(linkcheck._("Get the newest version at %s\n") % linkcheck.Config.Url))
            self.fd.write("-- "+(linkcheck._("Write comments and bugs to %s\n\n") % \
	                linkcheck.Config.Email))
            self.fd.flush()

    def newUrl (self, urlData):
        if self.fd is None: return
        self.fd.write("insert into %s(urlname,recursionlevel,parentname,"
              "baseref,errorstring,validstring,warningstring,infoString,"
	      "valid,url,line,name,checktime,downloadtime,cached) values "
              "(%s,%d,%s,%s,%s,%s,%s,%s,%d,%s,%d,%s,%d,%d,%d)%s\n" % \
	      (self.dbname,
	       linkcheck.StringUtil.sqlify(urlData.urlName),
               urlData.recursionLevel,
	       linkcheck.StringUtil.sqlify(urlData.parentName),
               linkcheck.StringUtil.sqlify(urlData.baseRef),
               linkcheck.StringUtil.sqlify(urlData.errorString),
               linkcheck.StringUtil.sqlify(urlData.validString),
               linkcheck.StringUtil.sqlify(urlData.warningString),
               linkcheck.StringUtil.sqlify(urlData.infoString),
               urlData.valid,
               linkcheck.StringUtil.sqlify(urlData.url),
               urlData.line,
               linkcheck.StringUtil.sqlify(urlData.name),
               urlData.checktime,
               urlData.downloadtime,
               urlData.cached,
	       self.separator))
        self.fd.flush()

    def endOfOutput (self, linknumber=-1):
        if self.fd is None: return
        if self.logfield("outro"):
            self.stoptime = time.time()
            duration = self.stoptime - self.starttime
            name = linkcheck._("seconds")
            self.fd.write("-- "+linkcheck._("Stopped checking at %s") % \
	              strtime(self.stoptime))
            if duration > 60:
                duration = duration / 60
                name = linkcheck._("minutes")
            if duration > 60:
                duration = duration / 60
                name = linkcheck._("hours")
            self.fd.write("	(%.3f %s)\n" % (duration, name))
        self.fd.flush()
        self.fd = None