def index():
form = URLConverterForm()
if form.validate_on_submit():
url = Url.query.filter_by(long_url=form.url.data).first()
if url is None:
url = Url(long_url=form.url.data)
db.session.add(url)
db.session.flush()
url.short_url = encode_id(url.id)
db.session.add(url)
db.session.commit()
form.url.data = request.base_url + url.short_url
return render_template('index.html', form=form)
parser = argparse.ArgumentParser()
parser.add_argument('-t',
'--generate-test-data',
action='store_true',
help='if present, generate test data for db')
parser.add_argument('-d',
'--drop-tables',
action='store_true',
help='drop all exited tables')
parser.add_argument('-v',
'--verbose',
action='store_true',
help='turn on debug for flask app')
args = parser.parse_args()
if args.drop_tables:
db.drop_all()
db.create_all()
if args.generate_test_data:
habr_urls = [
'https://habrahabr.ru/all/page{}/'.format(i)
for i in range(1, 100)
]
for url in habr_urls:
url = Url(long_url=url)
db.session.add(url)
db.session.flush()
url.short_url = encode_id(url.id)
db.session.add(url)
db.session.commit()
app.run(debug=args.verbose)
