Exemplo n.º 1
0
def shell(options):
    '''
    Interactive shell with database access
    '''
    from peewee import find_subclasses
    from models import (
                        Profile,
                        COMaster,
                        IED,
                        VarSys,
                        DI,
                        AI,
                        MV,
                        Event,
                        Energy,
                        BaseModel, )
    #for comaster in Profile.by_name(options.profile).comaster_set():
    #    src = 'co%s = comaster' % comaster.addres[-3:]
    #    exec(src)
    from datetime import datetime, date
    from IPython import embed
    models = find_subclasses(BaseModel)
    for model in models:
        print "Importing model %s" % model.__name__
    embed()
Exemplo n.º 2
0
def rebuilddb(options):
    '''
    Rebuild DB
    '''
    from peewee import drop_model_tables, create_model_tables, find_subclasses
    from models import BaseModel, populate_tables
    print "Reconsruyendo la base de datos..."
    models = find_subclasses(BaseModel)
    print "Drop tables..."
    drop_model_tables(models)
    print "Cereate tables..."
    create_model_tables(models)
    CONFIG = {
            'default':[
                    dict(address='192.168.1.97', description="Placa de prueba 1 "),
                    dict(address='192.168.1.98', description="Placa de prueba 2"),
            ],
            'test':[
                dict(address='127.0.0.1', description=u"Conexión con localhost"),
            ]
        }
    print "Populating..."
    for name, list_of_comaster_cfg in CONFIG.iteritems():
        populate_tables(name, list_of_comaster_cfg)