from psycopg2.extras import DictCursor from psycopg2.pool import ThreadedConnectionPool from digestry.config import config pool = ThreadedConnectionPool( config.getint("database", "pool_minimum"), config.getint("database", "pool_maximum"), dbname=config.get("database", "name"), user=config.get("database", "user"), password=config.get("database", "password"), host=config.get("database", "host"), port=config.getint("database", "port")) class Database(object): def __enter__(self): self.connection = pool.getconn() self.cursor = self.connection.cursor(cursor_factory=DictCursor) def __exit__(self, type, value, traceback): self.connection.commit() self.cursor.close() pool.putconn(self.connection) def select_patron(self, name): data = { "name": name } self.cursor.execute(
#!/usr/bin/env python import flask from digestry.config import config from digestry.routes import routes app = flask.Flask(__name__) app.register_blueprint(routes) app.secret_key = config.get("application", "secret") if __name__ == "__main__": app.run(debug=True)