Esempio n. 1
0
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:
        for line in args.input_file:
            line = line.strip()
            if len(line) == 0:
                continue

            tokens = line.split()
            user = User(tokens[0], tokens[1])
            user.set_password(tokens[2])
            db_session.add(user)
    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)

    db_session.commit()
Esempio n. 2
0
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:
        for line in args.input_file:
            line = line.strip()
            if len(line) == 0:
                continue

            tokens = line.split()
            user = User(tokens[0], tokens[1])
            user.set_password(tokens[2])
            db_session.add(user)
    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)

    db_session.commit()
Esempio n. 3
0
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)
Esempio n. 4
0
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)
Esempio n. 5
0
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)	
Esempio n. 6
0
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()
Esempio n. 7
0
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()