def get_user_data(parent):
"""Make sure user is logged in and get their account data."""
# Get user login status.
user = users.get_current_user()
if user is None:
# User is not logged in... Go log in then.
parent.redirect(users.create_login_url(parent.request.uri))
return None
# User is logged in.
data = models.UserData.get_by_key_name(user.user_id())
if data is None:
# First time logged in. Welcome then.
data = models.init_data(user)
return data
import os
app = Flask(__name__)
login = LoginManager(app)
login.login_view = 'login'
basedir = os.path.abspath(os.path.dirname(__file__))
app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('DATABASE_URL') or \
'sqlite:///' + os.path.join(basedir, 'app.db')
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
app.config['SECRET_KEY'] = os.environ.get(
'SECRET_KEY') or 'you-will-never-guess'
db = SQLAlchemy(app)
import models
from models import User, Note, init_data
init_data()
@app.route('/')
@app.route('/index')
@login_required
def index():
notes = Note.query.all()
notesForUser = []
for note in notes:
if note.receiver == '' or note.receiver == str(current_user.username):
notesForUser.append(note)
return render_template('index.html', title='Home', notes=notesForUser)
@app.route('/changepassform')
def init_db():
db.drop_all()
db.create_all()
init_data()
