setup_db.py

import sqlalchemy
from sqlalchemy import *
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
import malware_sample_model
import utils
import os, sys



# Load configuration
config = utils.get_configuration("malwarehouse.cfg")
uri = config.get('database', 'uri')
engine = create_engine(uri)
engine.echo = True  

# Create the root directory if it does not exist
if engine.name == "sqlite":
    database_path = os.path.dirname(engine.url.database)
    if not os.path.exists(database_path):
        try:
            os.makedirs(database_path)
        except Exception, err:
            print err
            sys.exit(1)
try:
    metadata = MetaData(engine)
    Base = malware_sample_model.Base
    Base.metadata.create_all(engine)
    print "%s successfully created"%uri
except Exception, err:
    print "Failed to create database -%s"%(err)