def post_link(): check_user_is_logged() # POST request if request.method == 'POST': # Create the link link = Link(request.form['url']) link.comment = request.form['comment'] link.user = g.user # Update db db_session.add(link) db_session.commit() return redirect(url_for('home')) # GET Request else: return render_template('post.html', user=g.user)
def main(): # Command-line parser = argparse.ArgumentParser( description='Populate a database from a text file') parser.add_argument('input_file', type=argparse.FileType('r'), default=sys.stdin, help='a list of users') args = parser.parse_args() # Delete the db dump if it exists if os.path.exists(db_file_name): os.remove(db_file_name) # # Populate the database # db_init() try: db_data = json.load(args.input_file) except ValueError as e: sys.stderr.write('Error while reading "%s"\n' % args.input_file.name) sys.stderr.write('%s\n' % e) sys.exit(0) # Add the users for user_data in db_data['users']: # Create the user instance user = User(user_data['name'], user_data['email']) user.set_password(user_data['pwd']) # Add each link if 'links' in user_data: for link_data in user_data['links']: link = Link(link_data['url']) if 'comment' in link_data: link.comment = link_data['comment'] user.links.append(link) # Add the user db_session.add(user) # Job done, commiy all db_session.commit()