def add_entry(): if not current_user.is_authenticated(): abort(401) tag_arr=request.form['tags'].split(',') tag_list_out=[] for index, a in enumerate(tag_arr): a=a.strip() if a: tag_list_out.append(Tag(a)) last_entry_object = Entry.query.order_by(Entry.id.desc()).first() if last_entry_object: entry_content_object=EntryContent(tag_list_out, last_entry_object.id+1) else: entry_content_object=EntryContent(tag_list_out, 1) entry_object=Entry(entry_content_object, request.form['title'], request.form['text']) db_session.add(entry_content_object) db_session.add(entry_object) db_session.commit() flash('New post was successfully added') return redirect(url_for('show_entries'))
def setUp(self): """Before each test, set up a blank database""" self.db_fd, library.core.config.DB_URI = tempfile.mkstemp() library.core.config.TESTING = True self.app = blogheaven.app.test_client() library.core.database.init_db() last_user_object = User.query.order_by(User.id.desc()).first() if last_user_object: admin = User(last_user_object.id+1, last_user_object.id+1, 'default') else: admin = User(1, 'leiksu', 'default') db_session.add(admin) db_session.commit()
from library.core.database import init_db init_db() from library.core.database import db_session from library.core.models import User admin = User(1,'leiksu', 'default') db_session.add(admin) db_session.commit()