def index():
if request.method == 'POST':
title = request.form['title']
comment = request.form['comment']
post = Post(created=datetime.datetime.now,
title=title,
comment=comment)
post.save()
# get posts for listing
posts = Post.query.descending(Post.created)
blog = Blog.query.first()
brand = Brand.query.first()
pages = Page.query
# NEED TO UPDATE FOR MULTIPLE ADDRESSES
wallets = Wallets.query.filter(
Wallets.username == get_current_user_data()["username"])
# trying to get user login boolean
user = get_current_user_data()
return render_template('index.html',
posts=posts,
user=user,
blog=blog,
brand=brand,
pages=pages,
wallets=wallets)
def addaddress():
myAddress = request.form['address']
myCoin = request.form['coin']
# need to get user mongo_id
userId = getUserId(get_current_user_data()["username"])
address = Wallets(coin=myCoin,
address=myAddress,
userId=userId,
username=get_current_user_data()["username"],
balance=10101010,
error='')
address.save()
return redirect(url_for('index'))
def test_user_expiration(self):
import time
with self.app.test_request_context():
self.app.auth.user_timeout = 0.01
login(self.user)
time.sleep(0.02)
assert get_current_user_data() is None
assert AuthUser.load_current_user() is None
def viewpage(id):
mypage = Page.query.get(id)
brand = Brand.query.first()
pages = Page.query
return render_template('page.html',
page=mypage,
user=get_current_user_data(),
brand=brand,
pages=pages)
def test_user_expiration(self):
import time
with self.app.test_request_context():
self.app.auth.user_timeout = 0.01
login(self.user)
time.sleep(0.02)
assert get_current_user_data() is None
assert AuthUser.load_current_user() is None
def decorator(*args, **kwargs):
data = get_current_user_data()
if data is None:
if request.authorization:
if UserController.authenticate_basic(inject=True):
return f(*args, **kwargs)
return not_logged_in(_not_logged_in, *args, **kwargs)
return f(*args, **kwargs)
def api_user_locale_set():
if 'lang' not in request.form:
return flask.jsonify(**{'fail': 'parameter \'lang\' not given'}), 400
try:
if request.authorization or get_current_user_data():
if request.authorization:
user = UserController.authenticate_basic()
else:
user = User.query.filter(
User.id == get_current_user_data()['id']).one()
if user:
UserController.locale_set(
locale=request.form['lang'], user=user)
return flask.jsonify(**{'success': True}), 201
UserController.locale_set(locale=request.form['lang'])
return flask.jsonify(**{'success': True}), 201
except Exception as ex:
return flask.jsonify(**{'fail': str(ex)}), 400
def viewpost(id):
mypost = Post.query.get(id)
brand = Brand.query.first()
pages = Page.query
comments = Comment.query.filter(Comment.post_id == id)
return render_template('post.html',
post=mypost,
user=get_current_user_data(),
brand=brand,
pages=pages,
comments=comments,
ccount=comments.count())
def profile():
# required for all areas to show brand and pages in menu
brand = Brand.query.first()
pages = Page.query
#if get_current_user_data()["username"] != "admin":
# return redirect(url_for('index'))
if request.method == 'POST':
if request.form['password'] == request.form['confirmpassword']:
#userId = 0
username = get_current_user_data()["username"]
password = request.form['password']
# query mongo for user
myUser = User.query.filter(User.username == username).first()
myAuth = AuthUser(username=username)
myAuth.set_and_encrypt_password(password, salt='1234567')
myUser.password = myAuth.password
myUser.save()
return render_template('profile.html',
brand=brand,
pages=pages,
user=get_current_user_data())
def admin():
if get_current_user_data()["username"] != "admin":
return redirect(url_for('index'))
if request.method == 'POST':
title = request.form['title']
comment = request.form['comment']
time = datetime.datetime.today()
post = Post(created=time, title=title, comment=comment)
post.save()
# get posts for listing
posts = Post.query.descending(Post.created)
brand = Brand.query.first()
blog = Blog.query.first()
pages = Page.query
users = User.query
return render_template('admin.html',
posts=posts,
user=get_current_user_data(),
brand=brand,
blog=blog,
pages=pages,
users=users)
def changepass():
if request.method == 'POST':
if request.form['password'] == request.form['confirmpassword']:
#userId = 0
username = get_current_user_data()["username"]
password = request.form['password']
# query mongo for user
myUser = User.query.filter(User.username == username).first()
myAuth = AuthUser(username=username)
myAuth.set_and_encrypt_password(password, salt='1234567')
myUser.password = myAuth.password
myUser.save()
return redirect(url_for('admin'))
def usignup():
brand = Brand.query.first()
pages = Page.query
# IMPLEMENT SIGNUP CODE
if request.method == 'POST':
if request.form['password'] == request.form['confirmpassword']:
#userId = 0
username = request.form['username']
password = request.form['password']
auth = AuthUser(username=username)
auth.set_and_encrypt_password(password, salt='1234567')
myuser = User(username=username, password=auth.password)
myuser.save()
return render_template('signup.html',
user=get_current_user_data(),
brand=brand,
pages=pages)
def editpage(id):
mypage = Page.query.get(id)
brand = Brand.query.first()
pages = Page.query
# if POST then save new post data
if request.method == 'POST':
title = request.form['title']
content = request.form['content']
time = datetime.datetime.today()
mypage.title = title
mypage.content = content
mypage.created = time
mypage.save()
return redirect(url_for('admin'))
return render_template('pageedit.html',
page=mypage,
user=get_current_user_data(),
brand=brand,
pages=pages)
def render(self, template_path, theme=None, status_code=200, **kwargs):
if not theme:
theme = self.get_active()
# @TO-DO: use a context processor
kwargs["env"] = {z: app.config[z] for z in app.config if z.islower()}
kwargs["env"]["application_root"] = app.config["APPLICATION_ROOT"]
user_context = get_current_user_data()
if user_context:
user = User.query.filter(User.id == user_context["id"]).one()
if not session.get("locale"):
session["locale"] = user.locale
elif session["locale"] != user.locale:
session["locale"] = user.locale
kwargs["user"] = user
return render_template("%s/templates/%s.html" % (theme, template_path), **kwargs), status_code
def ulogin():
# required for all areas to show brand and pages in menu
brand = Brand.query.first()
pages = Page.query
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
user = User.query.filter(User.username == username).first()
if user is not None:
authUser = AuthUser(username=username, salt='1234567')
authUser.password = user.password
#if authUser.authenticate(encrypt(password, salt='1234567')):
if authUser.authenticate(request.form['password']):
return redirect(url_for('index'))
return 'Failure :('
return render_template('login.html',
user=get_current_user_data(),
brand=brand,
pages=pages)
def load_current_user(cls, apply_timeout=True):
data = get_current_user_data(apply_timeout)
if not data:
return None
return cls.query.filter(cls.username==data['username']).one()
def load_current_user(cls, apply_timeout=True):
data = get_current_user_data(apply_timeout)
if not data:
return None
return cls.query.filter(cls.cpf == data['cpf']).one_or_none()
def post():
user = get_current_user_data()
return render_template('newpost.html', user=user)
def test_current_user(self):
with self.app.test_request_context():
login(self.user)
assert get_current_user_data() == self.user.__dict__
assert AuthUser.load_current_user() == self.user
def admin():
user = get_current_user_data()
return render_template('admin.html', user=user)
def admin():
user = get_current_user_data()
return render_template('admin.html', user=user)
def post():
user = get_current_user_data()
return render_template('newpost.html', user=user)
def feed():
user = get_current_user_data()
print user
return render_template('feed.html', user=user)
def test_current_user(self):
with self.app.test_request_context():
login(self.user)
assert get_current_user_data() == self.user.__dict__
assert AuthUser.load_current_user() == self.user
def feed():
user = get_current_user_data()
print user
return render_template('feed.html', user=user)
def refreshAllWallets():
wallets = Wallets.query.filter(
Wallets.username == get_current_user_data()["username"])
for myWallet in wallets:
refreshSingleWallet(myWallet)
