def user_handler(uid=None):
if not is_login(session):
flash('Please log in before', 'error')
return render_template('login.html')
if uid is None or len(uid) <= 0:
uid = session['_user']
user = User.find_user(db, uid=uid)
if is_current_user(session['_user'], uid):
return render_template('user.html', user=user)
return render_template('_profile.html', user=user)
def profile_handler(uid=None):
if not is_login(session):
return render_template('login.html')
if request.method == 'GET':
user = User.find_user(db, uid=uid)
if is_current_user(session['_user'], uid):
logger.debug('current user %s,uid user %s', session['_user'], uid)
if user.path is None:
return render_template('profile.html', user=user)
logger.info("Find the user %s", user.__dict__)
return render_template('user.html', user=user)
return render_template('_profile.html', user=user)
f = request.files['file']
fn = secure_filename(f.filename)
if not os.path.exists(os.path.join(UPLOAD, uid)):
os.makedirs(os.path.join(UPLOAD, uid))
p = os.path.join(os.path.join(UPLOAD, uid), fn)
f.save(p)
db.update({'uid': uid}, path=p, _company=request.form['_company'], _address=request.form['_address'])
return render_template('user.html', user=User.find_user(db, uid=uid))
def login():
if request.method == 'GET':
return render_template('login.html')
if is_login(session):
user = User.find_user(db, _user=request.form['_user'])
if getattr(user, 'path', None) is not None:
return render_template('user.html', user=user)
flash('Sign up ok!Please compelete your infromation', 'warning')
return render_template('profile.html', user=user)
user = validate_user(request.form['_user'], request.form['_password'])
if user is None:
flash("Invalid username/password or the user does not exist", 'error')
return render_template('login.html')
session['_user'] = user.uid
if getattr(user, 'path', None) is None:
return render_template('profile.html', user=user)
return render_template('user.html', user=user)
def find_user(username, password):
user = User.find_user(db, _user=username, _password=password)
return user
