PyDatabase.h

Go to the documentation of this file.

/*
    ------------------------------------------------------------------------------------
    LICENSE:
    ------------------------------------------------------------------------------------
    This file is part of EVEmu: EVE Online Server Emulator
    Copyright 2006 - 2021 The EVEmu Team
    For the latest information visit https://evemu.dev
    ------------------------------------------------------------------------------------
    This program is free software; you can redistribute it and/or modify it under
    the terms of the GNU Lesser 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 Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License along with
    this program; if not, write to the Free Software Foundation, Inc., 59 Temple
    Place - Suite 330, Boston, MA 02111-1307, USA, or go to
    http://www.gnu.org/copyleft/lesser.txt.
    ------------------------------------------------------------------------------------
    Author:     Bloody.Rabbit
    Updates:    Allan
*/

#ifndef __PY_DATABASE_H__INCL__
#define __PY_DATABASE_H__INCL__

#include "database/dbcore.h"
#include "python/PyRep.h"

class DBRowDescriptor
: public PyObjectEx_Type1
{
public:
    DBRowDescriptor();

    DBRowDescriptor(PyList* keywords);

    DBRowDescriptor( const DBQueryResult& res );
    DBRowDescriptor( const DBResultRow& row );

    uint32 ColumnCount() const;
    PyString* GetColumnName( uint32 index ) const;
    DBTYPE GetColumnType( uint32 index ) const;
    uint32 FindColumn( const char* name ) const;
    bool VerifyValue( uint32 index, PyRep* value );

    void AddColumn( const char* name, DBTYPE type );

protected:
    virtual ~DBRowDescriptor()    { /* do we need to do anything here? */ }
    // Helper functions:
    PyTuple* _GetColumnList() const;
    PyTuple* _GetColumn(size_t index) const;

    static PyTuple* _CreateArgs();
};

class CRowSet
: public PyObjectEx_Type2
{
public:
    CRowSet( DBRowDescriptor** rowDesc );

    size_t GetRowCount() const { return list().size(); }

    PyPackedRow* GetRow( uint32 index ) const { return list().GetItem( index )->AsPackedRow(); }

    PyPackedRow* NewRow();

protected:
    virtual ~CRowSet()    { /* do we need to do anything here? */ }
    DBRowDescriptor* _GetRowDesc() const;
    //PyList* _GetColumnList() const;

    static PyTuple* _CreateArgs();
    static PyDict* _CreateKeywords(DBRowDescriptor* rowDesc);
};

class CIndexedRowSet
: public PyObjectEx_Type2
{
public:
    CIndexedRowSet( DBRowDescriptor** rowDesc );

    size_t GetRowCount() const { return list().size(); }

    PyPackedRow* GetRow( PyRep* index ) const { return dict().GetItem( index )->AsPackedRow(); }

    PyPackedRow* NewRow( PyRep* key );

protected:
    virtual ~CIndexedRowSet()    { /* do we need to do anything here? */ }
    DBRowDescriptor* _GetRowDesc() const;
    //PyList* _GetColumnList() const;

    static PyTuple* _CreateArgs();
    static PyDict* _CreateKeywords(DBRowDescriptor* rowDesc);
};


class CFilterRowSet
: public PyObjectEx_Type2
{
public:
    CFilterRowSet( DBRowDescriptor** rowDesc );

    size_t GetKeyCount() const { return dict().size(); }

    CRowSet* GetRowset( PyRep *key) const { return (CRowSet*)(dict().GetItem( key )); }

    PyPackedRow* GetRow( PyRep* key, uint32 index )
        const { return GetRowset(key)->GetRow( index )->AsPackedRow(); }

    CRowSet* NewRowset( PyRep* key );

protected:
    virtual ~CFilterRowSet()    { /* do we need to do anything here? */ }
    DBRowDescriptor* _GetRowDesc() const;
    //PyList* _GetColumnList() const;

    static PyTuple* _CreateArgs();
    static PyDict* _CreateKeywords(DBRowDescriptor* rowDesc);
};

#endif /* !__PY_DATABASE_H__INCL__ */