source: trunk/Monitoring/smon/local_repo.py @ 871

# -*- coding: ISO-8859-1 -*-

##################################
#  @program        smon
#  @description    simulation monitor
#  @copyright      Copyright “(c)2009 Centre National de la Recherche Scientifique CNRS. 
#                             All Rights Reserved”
#  @svn_file       $Id: repo_io.py 2599 2013-03-24 19:01:23Z jripsl $
#  @version        $Rev: 2599 $
#  @lastrevision   $Date: 2013-03-24 20:01:23 +0100 (Sun, 24 Mar 2013) $
#  @license        CeCILL (http://dods.ipsl.jussieu.fr/jripsl/smon/LICENSE)
##################################

import sys
import sqlite3

import types

_conn=None
timeout=1

def connect():
        global _conn

        if _conn is not None:
                raise Exception()

        _conn=sqlite3.connect(":memory:",timeout)

        create_tables()

def free():
        global _conn

        if _conn is None:
                raise Exception()

        _conn.close()

        _conn=None

def create_tables():

        _conn.execute("create table simulation (id INTEGER PRIMARY KEY, name TEXT)")
        _conn.execute("create unique index idx_simulation on simulation (name)")

        _conn.execute("create table message (id INTEGER PRIMARY KEY, simulation_id TEXT)") # TODO: check how to use INT datatype for simulation_id column

def populate_tables_with_sample():

        rows = [('SIMU-001',),
                        ('SIMU-002',),
                        ('SIMU-003',),]
        _conn.executemany('INSERT INTO simulation (name) VALUES (?)', rows)

def retrieve_simulations():
        li=[]
        c=_conn.cursor()

        c.execute("select name,id from simulation")

        rs=c.fetchone()
        while rs is not None:
                li.append(types.Simulation(name=rs[0],id=rs[1]))
                rs=c.fetchone()

        return li

def retrieve_simulation(name):
        c=_conn.cursor()

        c.execute("select name,id from simulation where name = ?",(name,))

        rs=c.fetchone()

        if rs is None:
                raise Exception()

        return types.Simulation(id=rs[1])
    
def delete_simulation(simulation):
        _conn.execute("delete from simulation where name = ?",(simulation.name,))

def create_simulation(simulation):
        _conn.execute("insert into simulation (name) values (?)",(simulation.name,))
    
def update_simulation_status(simulation):
        _conn.execute("update simulation set status=? where name = ?",(simulation.status,simulation.name))

def retrieve_messages(simulation):
        li=[]

        c=_conn.cursor()

        _conn.execute("select id from message where simulation_id = ?",(simulation.id,))

        rs=c.fetchone()
        while rs is not None:
                li.append(types.Message(id=rs[0]))
                rs=c.fetchone()

        return li

def delete_messages(simulation):
        _conn.execute("delete from message where simulation_id = ?",(simulation.id,))

def create_message(message):
        _conn.execute("insert into message (simulation_id) values (?)",(message.simuid,))

def retrieve_last_message(simulation):
        c=_conn.cursor()

        _conn.execute("select id, content from message where simulation_id=? order by id desc limit 1",(simulation.id,))

        rs=c.fetchone()

        if rs is None:
                raise Exception()

        return types.Message(id=rs[0])