示例#1
0
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'))
示例#2
0
	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()
示例#3
0
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()