def signup_handler(request):
method = request.request.method
if is_authorised(request):
request.redirect(r'/')
return
if method == 'GET':
request.write(render_template('signup.html', {'disp':False, 'is_user' : is_authorised(request), 'location' : '/user/create', 'title' : "Sign Up" }))
elif method == 'POST':
print("running post")
username = request.get_field('username')
password = request.get_field('password')
repeat_password = request.get_field('repeat_password')
user = User.get(username)
if user is None and password == repeat_password and len(password) > 1:
user = User(username, password)
print("creating user : {}".format(user))
user.add()
l = List("", user.id)
l.add()
try:
os.makedirs('static/img/list/{}'.format(user.name))
except:
pass
try:
os.makedirs('static/img/user/{}'.format(user.name))
except:
pass
request.set_secure_cookie('user_id', str(user.id))
request.redirect(r'/user/edit')
else:
request.write(render_template('signup.html', {'user_issue': user is not None, 'disp':True, 'is_user' : is_authorised(request), 'location' : '/user/create', 'title' : "Sign Up" }))
def profile_edit_handler(request):
method = request.request.method
if not is_authorised(request):
request.redirect(r'/')
return
user = User.get_by_id(int(request.get_secure_cookie('user_id')))
if method == "GET":
request.write(render_template('profile.html', {'user_obj' : user, 'disp' : False, 'is_user' : is_authorised(request), 'user_id' : user.id, 'user' : user.name, 'title' : "Edit Profile" }))
if method == "POST":
email = request.get_field('email')
if not email:
request.write(render_template('profile.html', {'user_obj' : user, 'disp' : True, 'is_user' : is_authorised(request), 'user_id' : user.id, 'user' : user.name, 'title' : "Edit Profile" }))
return
rname = request.get_field('rname')
fname, contype, body = request.get_file('profile-img')
user = User.get_by_id(int(request.get_secure_cookie('user_id')))
user.email = email
user.real_name = rname
if fname:
user.image = '/static/img/user/{}/profile-img.{}'.format(user.name, fname.split('.')[-1])
with open(user.image[1:], 'wb') as f:
f.write(body)
user.update()
request.redirect(r'/timeline/{}'.format(user.id))
def index_handler(request):
cookie = request.get_secure_cookie('user_id')
if cookie != None:
user = User.get_by_id(int(cookie))
if user:
ls = List.get_user_lists(user)
names = user.get_newsfeed()
user_list = user.get_lists()[0]
request.write(render_template('news-feed.html', {'user_id':str(request.get_secure_cookie('user_id'))[2:-1],
'user_id': user.id, 'names':names, 'is_user' : is_authorised(request),
'title' : 'News Feed', 'user' : user.name, 'user_list': user_list.id}))
if not cookie or not user:
request.write(render_template('homepage.html', {'is_user' : is_authorised(request), 'title' : "Home Page"}))
def list_creation_handler(request, list_id):
method = request.request.method
if not is_authorised(request):
request.redirect(r'/login')
return
user = User.get_by_id(int(request.get_secure_cookie('user_id')))
if method == 'GET':
user_list = user.get_lists()[0]
request.write(render_template('create.html', {'user_id':str(request.get_secure_cookie('user_id'))[2:-1], 'user' : user.name, 'is_user' : is_authorised(request), 'title' : 'Create A List', 'user_list': user_list.id, 'list_id':list_id}))
elif method == 'POST':
textdesc = request.get_field('description')
##Get the User
user = User.get_by_id(int(request.get_secure_cookie('user_id')))
head, contype, body = request.get_file('file_upload')
#Get a new Item object
ls = List.get(int(list_id))
if ls and (head or textdesc):
item = Item(ls.id, text=textdesc)
item.add()
if head:
item.image = head
head = head.split('.')[-1]
filename = 'static/img/list/{}/item{}.{}'.format(user.name, item.id, head)
item.image = '/'+filename
with open(filename, 'wb') as f:
f.write(body)
item.update()
request.redirect(r'/list/{}'.format(item.list_id))
else:
pass
def login_handler(request):
method = request.request.method
if is_authorised(request):
request.redirect(r'/')
return
if method == 'GET':
request.write(render_template('login.html', {'disp' : False, 'is_user' : is_authorised(request), 'location' : '/login', 'title' : "Login" }))
elif method == 'POST':
username = request.get_field('username')
password = request.get_field('password')
m = hashlib.sha256()
password = password.encode()
m.update(password)
password = m.hexdigest()
user = User.get(username)
if user is not None and password == user.password:
request.set_secure_cookie('user_id', str(user.id))
request.redirect(r'/login')
else:
request.write(render_template('login.html', {'disp' : True, 'is_user' : is_authorised(request), 'location' : '/login', 'title' : "Login" }))
def list_display_handler(request, list_id):
method = request.request.method
if not is_authorised(request):
request.redirect(r'/')
return
if method == 'GET':
ls = List.get(int(list_id))
if ls:
user = User.get_by_id(ls.userid)
user_list = user.get_lists()[0]
user2 = User.get_by_id(int(request.get_secure_cookie('user_id')))
bucket = [a.id for a in ls.get_items()]
comments = Comment.get_comments_for_list(int(list_id))
items = {}
for item in bucket:
items[item] = Item.get(item)
request.write(render_template('my_bucket_list.html',
{'comments': comments, 'user_id':str(request.get_secure_cookie('user_id'))[2:-1],
'logged_in_username' : user2.name, 'bucket' : bucket,
'items':items, 'user' : user.name, 'is_user' : is_authorised(request),
'list_title' : ls.title, 'user_name' : user.name, 'list_id' : ls.id,
'title' : "{}\'s Bucket List\'".format(user.name), 'user_list': user_list.id}))
else:
error404_handler(request)
return
elif method == 'POST':
text = request.get_field('comment')
save = request.get_field('save') == ''
if not save:
if text:
user = int(request.get_secure_cookie('user_id'))
c = Comment(user, text, list_id)
c.add()
request.redirect(r'/list/{}'.format(list_id))
return
ls = List.get(int(list_id))
for i in [a.id for a in ls.get_items()]:
checked = request.get_field("check{}".format(i))
item = Item.get(i)
item.completed = bool(checked)
item.update()
request.redirect(r'/')
return
def list_edit_handler(request, list_id):
method = request.request.method
if not is_authorised(request):
request.redirect(r'/')
return
if method == "GET":
ls = List.get(int(list_id))
if ls:
user = User.get_by_id(ls.userid)
user_list = user.get_lists()[0]
user2 = User.get_by_id(int(request.get_secure_cookie('user_id')))
if not user.name == user2.name:
request.redirect("/list/{}".format(list_id))
return
bucket = [a.id for a in ls.get_items()]
comments = Comment.get_comments_for_list(int(list_id))
items = {}
for item in bucket:
items[item] = Item.get(item)
request.write(
render_template('edit_list.html',
{'nfeed':False, 'comments' : comments, 'user_id' : str(request.get_secure_cookie('user_id'))[2:-1],
'logged_in_username' : user2.name, 'bucket' : bucket, 'items' : items,
'user' : user.name, 'is_user' : is_authorised(request), 'list_id' : ls.id,
'title' : 'Edit List', 'user_list': user_list.id}))
else:
error404_handler(request)
return
elif method == "POST":
ls = List.get(int(list_id))
for i in [a.id for a in ls.get_items()]:
checked = request.get_field("check{}".format(i))
if bool(checked):
item = Item.get(i)
item.delete()
request.redirect(r'/list/{}'.format(list_id))
return
def timeline_handler(request, uid):
method = request.request.method
if not is_authorised(request):
request.redirect(r'/')
return
id = request.get_secure_cookie('user_id')
if method == "GET":
user = User.get_by_id(int(id))
user_obj = User.get_by_id(int(uid))
lists = [a for a in user.get_lists()]
items = [a.get_items()[0] if a.get_items() != [] else Empty() for a in lists]
request.write(render_template('timeline.html',
{'title':'Timeline', 'lists':lists,
'objs':items, 'user_obj':user_obj,
'user':user.name, 'user_id':str(id)[2:-1],
'is_user':is_authorised(request)}))
if method == "POST":
l = List(request.get_field("title"), int(id))
l.add()
request.redirect(r'/list/'+str(l.id))
def error404_handler(request):
request.write(render_template('error404.html', {'title' : "Error 404" }))