diff --git a/example/app.py b/example/app.py index 77c688e..5e5d9ad 100644 --- a/example/app.py +++ b/example/app.py @@ -22,17 +22,15 @@ db = SQLAlchemy(app) toolbar = DebugToolbarExtension(app) +with app.app_context(): + db.create_all() + class ExampleModel(db.Model): __tablename__ = 'examples' value = db.Column(db.String(100), primary_key=True) -@app.before_first_request -def setup(): - db.create_all() - - @app.route('/') def index(): app.logger.info("Hello there") diff --git a/src/flask_debugtoolbar/panels/profiler.py b/src/flask_debugtoolbar/panels/profiler.py index 3294e4a..66f22aa 100644 --- a/src/flask_debugtoolbar/panels/profiler.py +++ b/src/flask_debugtoolbar/panels/profiler.py @@ -94,7 +94,7 @@ def process_response(self, request, response): def title(self): if not self.is_active: return "Profiler not active" - return 'View: %.2fms' % (float(self.stats.total_tt)*1000,) + return 'View: %.2fms' % (float(self.stats.total_tt) * 1000,) def nav_title(self): return 'Profiler' @@ -102,7 +102,7 @@ def nav_title(self): def nav_subtitle(self): if not self.is_active: return "in-active" - return 'View: %.2fms' % (float(self.stats.total_tt)*1000,) + return 'View: %.2fms' % (float(self.stats.total_tt) * 1000,) def url(self): return '' diff --git a/src/flask_debugtoolbar/utils.py b/src/flask_debugtoolbar/utils.py index 267a2e3..76957e1 100644 --- a/src/flask_debugtoolbar/utils.py +++ b/src/flask_debugtoolbar/utils.py @@ -88,6 +88,7 @@ def format_sql(query, args): SqlLexer(), HtmlFormatter(noclasses=True, style=PYGMENT_STYLE))) + def gzip_compress(data, compresslevel=6): buff = io.BytesIO() with gzip.GzipFile(fileobj=buff, mode='wb', compresslevel=compresslevel) as f: diff --git a/test/basic_app.py b/test/basic_app.py index 6ec1ddd..98dac60 100644 --- a/test/basic_app.py +++ b/test/basic_app.py @@ -4,10 +4,12 @@ from flask_debugtoolbar import DebugToolbarExtension app = Flask('basic_app') +app.config['DEBUG'] = True app.config['SECRET_KEY'] = 'abc123' app.config['SQLALCHEMY_RECORD_QUERIES'] = True app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////tmp/test.db' -# This is no longer needed for Flask-SQLAlchemy 3.0+, if you're using 2.X you'll want to define this: +# This is no longer needed for Flask-SQLAlchemy 3.0+, +# if you're using 2.X you'll want to define this: # app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False # make sure these are printable in the config panel @@ -17,17 +19,15 @@ toolbar = DebugToolbarExtension(app) db = SQLAlchemy(app) +with app.app_context(): + db.create_all() + class Foo(db.Model): __tablename__ = 'foo' id = db.Column(db.Integer, primary_key=True) -@app.before_first_request -def setup(): - db.create_all() - - @app.route('/') def index(): Foo.query.filter_by(id=1).all() diff --git a/test/test_toolbar.py b/test/test_toolbar.py index fd04ddc..8624ee3 100644 --- a/test/test_toolbar.py +++ b/test/test_toolbar.py @@ -7,7 +7,6 @@ def load_app(name): app = __import__(name).app - app.config['TESTING'] = True return app.test_client() diff --git a/tox.ini b/tox.ini index 6483179..324525d 100644 --- a/tox.ini +++ b/tox.ini @@ -13,7 +13,7 @@ commands = deps = pycodestyle commands = - pycodestyle flask_debugtoolbar test + pycodestyle src/flask_debugtoolbar test --ignore=E731,W503,W504 [pycodestyle] max-line-length = 100