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()