diff --git a/server.py b/server.py index 4a23911..a68e5ee 100644 --- a/server.py +++ b/server.py @@ -1,3 +1,5 @@ +import time + from flask import Flask, request, render_template from wtforms import Form, StringField @@ -16,12 +18,17 @@ def index(): form = AnalyzeForm(request.form) results = {} error = None - try: - results = {'friends': analyze_friends(form.user_id), - 'followers': analyze_followers(form.user_id)} - except Exception as exc: - error = exc if request.method == 'POST' and form.validate() and form.user_id.data: + if app.config['DRY_RUN']: + time.sleep(2) + results = {'friends': (150, 25, 200), + 'followers': (200, 20, 250)} + else: + try: + results = {'friends': analyze_friends(form.user_id), + 'followers': analyze_followers(form.user_id)} + except Exception as exc: + error = exc return render_template('index.html', form=form, results=results, error=error) @@ -30,8 +37,10 @@ def index(): import argparse parser = argparse.ArgumentParser() parser.add_argument('--debug', action='store_true') + parser.add_argument('--dry-run', action='store_true') parser.add_argument('port', nargs=1, type=int) args = parser.parse_args() [port] = args.port + app.config['DRY_RUN'] = args.dry_run app.run(port=port, debug=args.debug)