def rewrite_dataset(self): from ..orm import Dataset # Now patch up the Dataset object ds = Dataset(**self.bundle.identity.to_dict()) ds.name = self.bundle.identity.name ds.vname = self.bundle.identity.vname self.session.merge(ds)
def _create(self): """Create the database from the base SQL""" from databundles.orm import Dataset, Partition, Table, Column, File from ..identity import new_identity from sqlalchemy.orm import sessionmaker tables = [ Dataset, Partition, Table, Column, File ] for table in tables: table.__table__.create(bind=self.engine) # Create the Dataset record session = self.creation_session ds = Dataset(**self.bundle.config.identity) ident = new_identity(self.bundle.config.identity) ds.name = ident.name ds.vname = ident.vname session.add(ds) session.commit()