Added function to connect to database

pull/3/head
assar 8 years ago
parent b8dff4a817
commit 10adc15885

@ -1,9 +1,32 @@
import json
import sqlalchemy import sqlalchemy
from flask import Flask from flask import Flask
from flask import render_template from flask import render_template
app = Flask(__name__) app = Flask(__name__)
db_engine = {}
db_metadata = {}
@app.route('/') @app.route('/')
def index(): def index():
return render_template('partsearch.html') return render_template('partsearch.html')
def connect(user, password, db, host='localhost', port=5432):
'''Returns a connection and a metadata object'''
# We connect with the help of the PostgreSQL URL
url = 'postgresql://{}:{}@{}:{}/{}'
url = url.format(user, password, host, port, db)
# The return value of create_engine() is our connection object
con = sqlalchemy.create_engine(url, client_encoding='utf8')
# We then bind the connection to MetaData()
meta = sqlalchemy.MetaData(bind=con, reflect=True)
return con, meta
if __name__ == '__main__':
with open('admin.json') as f:
postgres_credentials = json.load(f)
db_engine, db_metadata = connect(postgres_credentials['username'], postgres_credentials['password'], 'parts')

Loading…
Cancel
Save