def get_entry_by_id(user):
entries = JournalEntry.select().where(JournalEntry.author == user)
if not entries:
print("there are no journals")
elif entries:
entry_id = int(input("choose any entry id"))
entry = JournalEntry.select().where(JournalEntry.id == entry_id).get()
print("-" * 20)
print(entry.content)
print("-" * 20)
def get_all_entries(user):
entries = JournalEntry.select().where(JournalEntry.author == user)
print("-" * 20)
for entry in entries:
print(entry.id, entry.timestamp, entry.content[:20], "...")
print("-" * 20)
def delete_journal(user, id):
entry = JournalEntry.select().where(JournalEntry.id == id).get()
print("Delete the journal--", id)
a = input("(y/n)-->")
if a == "y":
entry.delete_instance()
print("Entry deleted")
elif a == "n":
print("Journal not deleted")
return user.get()
def test_journal_entry_creation(self):
"""Test creating a JournalEntry"""
with test_database(TEST_DB, (User, JournalEntry)):
UserModelTestCase.create_users()
user = User.select().get()
JournalEntry.create(
user=user,
title='Testing Journal Entries',
date=datetime.datetime.now().strftime('%Y-%m-%d'),
time_spent='22',
what_i_learned='Hopefully, I learn if this works or not!',
resources_to_remember='teamtreehouse.com')
self.assertEqual(JournalEntry.select().count(), 1)
self.assertEqual(JournalEntry.user, user)
def test_journal_entry_create(self):
"""Test can create new JournalEntry, redirects to homepage, and adds to the database."""
journal_entry_data = {
'title':
'testing new journal entries',
'date':
datetime.datetime.now().strftime('%Y-%m-%d'),
'time_spent':
'2',
'what_i_learned':
'how to test flask views',
'resources_to_remember':
'teamtreehouse.com, stackoverflow.com, flask.pocoo.org'
}
with test_database(TEST_DB, (User, JournalEntry)):
UserModelTestCase.create_users(1)
self.app.post('/login', data=USER_DATA)
journal_entry_data['user'] = User.select().get()
rv = self.app.post('/entry', data=journal_entry_data)
self.assertEqual(rv.status_code, 302)
self.assertEqual(rv.location, 'http://localhost/')
self.assertEqual(JournalEntry.select().count(), 1)
def title_exists(form, field):
if JournalEntry.select().where(JournalEntry.title**field.data).exists():
raise ValidationError('An entry with that title already exists.')
def get_entry_by_id(id):
entry = JournalEntry.select().where(JournalEntry.id == id).get()
print("-" * 20)
print(entry.content)
print("-" * 20)
