Advertisement
Guest User

Untitled

a guest
Jul 20th, 2017
65
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.62 KB | None | 0 0
  1. from flask import Flask, render_template, request, jsonify, redirect, url_for, flash, session
  2. import json, gc
  3. ....
  4.  
  5. app = Flask(__name__)
  6. app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql://postgres:root@localhost/josh'
  7. db = SQLAlchemy(app)
  8.  
  9. class User(db.Model):
  10. __tablename__ = "users"
  11. id = db.Column(db.Integer, primary_key=True)
  12. username = db.Column(db.String(80), unique = True)
  13. email = db.Column(db.String(120), unique=True)
  14. password = db.Column(db.String(100))
  15.  
  16. def __init__(self, email):
  17. self.username = username
  18. self.email = email
  19. self.password = password
  20.  
  21. def __repr__(self):
  22. return '<Username %r>' % self.username
  23.  
  24. @app.route('/register/', methods = ['GET', 'POST'])
  25. def register():
  26. try:
  27. if request.method == 'POST':
  28. username = request.form['username']
  29. email = request.form['email']
  30. password = sha256_crypt.encrypt((str(request.form['password'])))
  31.  
  32. user = db.session.query(User).filter_by(username = username).first()
  33.  
  34. if not user:
  35. result = User(username, email, password)
  36. db.session.add(result)
  37. db.session.commit()
  38.  
  39. flash('Thanks for registering!')
  40. gc.collect()
  41. session['logged_in'] = True
  42. session['username'] = username
  43.  
  44. return redirect(url_for('dashboard'))
  45.  
  46. else:
  47. flash('That username is already taken, please choose another.')
  48.  
  49. return render_template('pages/user_sys/register.html')
  50. except Exception as e:
  51. return (str(e))
  52. ....
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement