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()
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)