def test_json(self):
b = Blog(title='Test', author='Maki')
b.create_post("Post title", "Post content")
b.create_post("Post title2", "Post content2")
post_list = [{
'title': "Post title",
'content': "Post content"
}, {
'title': "Post title2",
'content': "Post content2"
}]
expected = {'title': 'Test', 'author': 'Maki', 'posts': post_list}
self.assertDictEqual(expected, b.json())
def test_json_no_posts(self):
b = Blog(title='Test', author='Maki')
expected = {'title': 'Test', 'author': 'Maki', 'posts': []}
self.assertDictEqual(expected, b.json())
class Menu(object):
def __init__(self, user):
self.user = user
self._account = self._user_has_account()
self.blog_map = {}
self.blog_posts = None
if self._account is not None:
self._blog = self._account
print("Welcome back {}.".format(self.user))
else:
self._prompt_user_for_account()
def _user_has_account(self):
return Database.find_one(collection=constants.BLOG_COLLECTION,
query={'author': self.user})
def _prompt_user_for_account(self):
print(
"\nSeems you do not have an associated account, please create a new blog...."
+ "\n")
title = input("Enter a title for your Blog: ")
description = input("Enter description for your Blog: ")
self._blog = Blog(author=self.user,
description=description,
title=title)
self._blog.save_to_mongo()
self._blog = self._blog.json()
print("[+] Successfully created!!")
def list_all_blogs(self):
table = PrettyTable(['S.No', 'Blog Title', 'Description', 'Author'])
for i, b in enumerate([
d for d in Database.find(collection=constants.BLOG_COLLECTION,
query={})
]):
table.add_row(
[i + 1,
b.get('title'),
b.get('description'),
b.get('author')])
self.blog_map.setdefault(i + 1, b)
print(table)
return self.blog_map
def get_all_posts_for_blog(self, selected_blog):
self.blog_posts = Post.from_blog(selected_blog.get('id'))
post_table = PrettyTable(
['id', 'title', 'content', 'author', 'created_at'])
for i, post in enumerate(self.blog_posts):
post_table.add_row([
post.get('id'),
post.get('title'),
post.get('content'),
post.get('author'),
post.get('created_at'),
])
print("\nPrinting all POSTS for BLOG <{}>".format(
selected_blog.get('title')),
end="\n")
print(post_table)
def run_menu(self):
row = input("Do you wish to read (R) or write (W) the blogs? (R/W) :")
if row.upper() == 'R':
print('Choose a blog:')
blog_map = self.list_all_blogs()
user_selection = input('Enter your selection(S.No.): ')
try:
user_selection = int(user_selection)
if user_selection > len(blog_map):
int('a')
selected_blog = blog_map.get(user_selection)
self.get_all_posts_for_blog(selected_blog)
except ValueError:
print("Invalid value detected, aborting...")
elif row.upper() == 'W':
print("[+]-------" * 5 + "[+]")
new_post = Blog.from_mongo(self._blog.get('id')).new_post()
print(
f"\nPOST <{new_post.get('title')}> created for blog <{self._blog.get('title')}>.\n"
)
else:
print("Thank you for blogging with us!")