#!/usr/bin/env python
# BioXHitDBInterface.py
#
#   Copyright (C) 2007 CCLRC, Graeme Winter
#
#   This code is distributed under the BSD license, a copy of which is
#   included in the root directory of this package.
#
# An interface to the CCP4i / BioXHit database handler, to allow recording
# of progress during data reduction.
#

import os
import sys
import exceptions

class _BioXHitDBInterfaceReal(object):
  '''A class to mediate transactions with the CCP4i database.'''

  def __init__(self):

    self._user = ccp4i.GetUserId()
    if not self._user:
      self._user = 'unknown'

    dbClientAPI.DbStartHandler()
    self._connection = dbClientAPI.handlerconnection()
    self._connection.DbRegister(self._user, 'dummy', True)

    self._project = None

    return

  def create_project(self, project, directory):
    '''Create a new CCP4i project in directory.'''

    result = self._connection.CreateDatabase(
        project, directory)

    status = result[0]
    result = result[1:]

    if not status == 'OK':
      raise RuntimeError, 'failed to create project %s' % project

    self._project = project


  def create_job(self, job,
                 input_files = [],
                 output_files = [],
                 log_file = None):
    '''Create a job within this project.'''

    if not self._project:
      raise RuntimeError, 'no current project'

    result = self._connection.NewRecord(self._project)

    status = result[0]
    result = result[1:]

    if not status == 'OK':
      raise RuntimeError, 'error creating job in project %s' % \
            self._project

    job_id = result[0]

    for f in input_files:
      self._connection.AddInputFile(project, job_id, f)

    for f in output_files:
      self._connection.AddOutputFile(project, job_id, f)

    if log_file:
      self._connection.SetLogfile(project, job_id, log_file)

    # set the job name in here...
    # want to set TITLE, TASKNAME, STATUS

    self._connection.SetData(self._project, job_id, 'TASKNAME', job)

    return

  def __del__(self):
    self._connection.HandlerDisconnect()


class _BioXHitDBInterfaceFake(object):
  '''A class to do nothing if ccp4i db handler not available.'''

  def __init__(self):
    pass

  def create_project(self, project, directory):
    pass

  def create_job(self, job,
                 input_files = [],
                 output_files = [],
                 log_file = None):
    pass


try:

  sys.path.append('/home/graeme/bioxhit/dbccp4i-0.1/dbccp4i')
  sys.path.append('/home/graeme/bioxhit/dbccp4i-0.1/ClientAPI')

  import ccp4i
  import dbClientAPI

  BioXHitDBInterface = _BioXHitDBInterfaceReal()

except exceptions.Exception, e:

  print '%s' % str(e)

  BioXHitDBInterface = _BioXHitDBInterfaceFake()

if __name__ == '__main__':

  import time

  project = 'demo%d' % int(time.time())

  BioXHitDBInterface.create_project(project, os.getcwd())
  BioXHitDBInterface.create_job('foo', log_file = 'arse.log')