def test_dbconnect():
localcon = dbconnect(databasename='components1.db')
print 'Test for missing database.'
assert localcon is False, 'Database does exist'
localcon = dbconnect()
print 'Test for default database.'
assert localcon is not False, 'Default database does not exist'
localcon.close()
localcon = dbconnect(databasename='components1.db', readonly=False)
print 'Test empty database create.'
if not os.path.exists('components1.db'):
assert False, 'Unable to create database file.'
else:
localcon.close()
os.remove('components1.db')
Base.metadata.bind = engine
DBSession = sessionmaker()
DBSession.bind = engine
session = DBSession()
app = Flask(__name__)
# Load default config and override config from an environment variable
app.config.update(dict(
DATABASE=os.path.join(app.root_path, 'database/components.db'),
DEBUG=True,
SECRET_KEY='development key',
))
app.config.from_envvar('APP_SETTINGS', silent=True)
con = dbconnect()
@app.route('/')
def index():
return render_template('index.html')
def create_query(filtered=1):
'''
Build a query object depending on whether it is a filtered query or not
:param filtered:
:return: orm query object
'''
if filtered == '1':
query_obj = session.query(Components.ID, Components.Name, Components.CurrentStock, Components.ReorderLevel,
import csv
from componentsmodule import Component, dbconnect
import pdb
#pdb.set_trace()
with dbconnect() as conn:
with open(r'testdata.csv','r') as fp:
component_details = csv.reader(fp)
titles_row = component_details.next()
for component_row in component_details:
comp_object = Component(zip(titles_row,component_row),conn)
print component_row