# -*- coding: utf-8 -*-
#===========================================================================
# Copyright (C) 2016  Max-Planck-Institut für Radioastronomie, Bonn, Germany
# 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 3 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, see <http://www.gnu.org/licenses/>.
#===========================================================================
# SVN properties (DO NOT CHANGE)
#
# $Id$
# $HeadURL$
# $LastChangedRevision$
# $Author$
# $LastChangedDate$
#
#============================================================================
from difxdb.model import model
import os

def moduleExists(session, vsn):
    
    if (session.query(model.Module).filter_by(vsn=vsn).count() > 0):
        return(True)
    else:
        return(False)

def getModuleById(session, id):
    
    return(session.query(model.Module).filter_by(id=id).one())
   
   
def getModuleByVSN(session, vsn):
    
    return(session.query(model.Module).filter_by(vsn=vsn).one())

def getModulesByExperimentCode(session, expCode):
    '''
    Returns a list of modules that contain data from an experiment with the give experiment code
    '''

    modules = session.query(model.Module).filter(model.Module.experiments.any(model.Experiment.code==expCode) )
    return(modules)

def getModulesByExperimentId(session, expId):
    '''
    Returns a list of modules that contain data from an experiment with the give experiment Id
    '''

    modules = session.query(model.Module).filter(model.Module.experiments.any(model.Experiment.id==expId) )
    return(modules)
   

def isCheckOutAllowed(session, vsn):
    '''
    Checks whether a module can be checked out from the media library. If any
    of the experiments contained on the module have a status other than "released"
    check-out is not allowed. Also the directory for the module must have been
    parsed at least once in order to allow check-out. However check-out is allowed
    for unscanned modules if an experiment was assigned manually and the experiment
    is released.
    '''
    expCount = 0 
    module = getModuleByVSN(session,vsn)
    
    if (module == None):
        return(False)
    
    # all experiments contained on this module must be released
    for exp in module.experiments:
        expCount += 1
        if exp.status.statuscode < 100:
            return(False)
        
    # Modules that do not have an experiment associated with them must have been
    # scanned at least once in order to allow check-out.
    if module.numScans is None and expCount == 0:
        return(False)

    return(True)
           
def isMark6(vsn):
    '''
    Returns True if this is a Mark6 VSN; False otherwise
    '''
    if '%' in vsn:
        return(True)
    else:
        return(False)

    
def hasDir(vsn, dirPath=None):
    '''
    Checks whether a .dir file (or filelist in case of Mark6 modules)
    exists for the module with the given vsn.
    If the optional dirPath is set the .dir/.filelist file will be searched under that path.
    Otherwise the MARK5_DIR_PATH environment is evaluated.
    '''
    if (dirPath == None):
        dirPath = os.getenv("MARK5_DIR_PATH")
        if (dirPath == None):
            return(False)
             
        if isMark6(vsn):
            if (os.path.isfile(dirPath + "/" + vsn + ".filelist")):
                return(True)
        else:
            if (os.path.isfile(dirPath + "/" + vsn + ".dir")):
                return(True)
        
    return(False)
    
    
def getUnscannedModules(session):
    '''
    Returns a collection of Module objects that have not been scanned 
    '''
    return(session.query(model.Module).filter(model.Module.numScans == None).all())


def getScanCount(session, vsn):
    '''
    Returns the number of scans recorded on the module. None if the module has not been scanned yet.
    '''
    
    try:
        module = getModuleByVSN(session, vsn)
    except:
        return
    
    return(module.numScans)

def retire(session, moduleId):
    '''
    Sets the retired field for a module and removes the module from its slot
    '''
    try:
        module=getModuleById(session, moduleId)
        module.retired = 1
        module.slot = None
        session.commit()
        session.flush()

    except:
        return