-
Notifications
You must be signed in to change notification settings - Fork 0
/
Menu.py
65 lines (48 loc) · 1.98 KB
/
Menu.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
__author__ = 'federico mateucci'
from Database import Database
from models.Blog import Blog
class Menu(object):
def __init__(self):
self.user = input("Enter your author name : ")
self.user_blog = None
if self._user_has_account():
print("Welcome back {}".format(self.user))
else:
self._prompt_user_for_account()
def _user_has_account(self):
datainfoblog = Database.find_one('blogs', {'author': self.user}) # Check in database if user exists, and throws a boolean
if datainfoblog is not None:
self.user_blog = Blog.get_from_mongo(datainfoblog['id'])
return True
else:
return False
def _prompt_user_for_account(self):
tittle = input("enter blog title : ")
description = input("Enter blog description")
blog = Blog(author=self.user,
tittle=tittle,
description=description)
blog.save_to_mongo()
self.user_blog = blog
def run_menu(self):
read_or_write = input("Do you want to read (R) or write (W) blogs?")
antiCrashDecision = read_or_write.upper()
if antiCrashDecision == 'R':
self._list_blogs()
self._view_blog()
elif antiCrashDecision == 'W':
self.user_blog.new_post()
else:
print("Thansk You For Blogging !")
def _prompt_write_post(self):
self.user_blog.new_post()
def _list_blogs(self):
blogs = Database.find(collection='blogs', query={})
for blog in blogs:
print("ID:{}, Tittle: {}, Author: {}".format(blog['id'], blog['tittle'], blog['author']))
def _view_blog(self):
blog_to_see = input("Enter ID of your blog you like it to read")
blog = Blog.get_from_mongo(blog_to_see)
posts = blog.get_posts()
for post in posts:
print("Date : {}, Tittle: {}\n\n{}".format(post['date'], post['tittle'], post['content']))