Guest User

Untitled

a guest
Jun 6th, 2020
96
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.87 KB | None | 0 0
  1. import os
  2. import psycopg2
  3.  
  4. from flask import Flask, session, render_template
  5. from flask_session import Session
  6. from sqlalchemy import create_engine
  7. from sqlalchemy.orm import scoped_session, sessionmaker
  8.  
  9.  
  10. app = Flask(__name__)
  11.  
  12. # Check for environment variable
  13. if not os.getenv("DATABASE_URL"):
  14. raise RuntimeError("DATABASE_URL is not set")
  15.  
  16. DATABASE_URL = os.environ['DATABASE_URL']
  17. conn = psycopg2.connect(DATABASE_URL, sslmode='require')
  18.  
  19. # Configure session to use filesystem
  20. app.config["SESSION_PERMANENT"] = False
  21. app.config["SESSION_TYPE"] = "filesystem"
  22. Session(app)
  23.  
  24. # Set up database
  25. engine = create_engine(os.getenv("DATABASE_URL"))
  26. db = scoped_session(sessionmaker(bind=engine))
  27.  
  28.  
  29. @app.route("/")
  30. def index():
  31. return "Project 1: TODO"
  32.  
  33.  
  34.  
  35. try:
  36. connection = psycopg2.connect(user = "zpztmceixgfwqvs",
  37. password = "16671705ed2cd07673xf836a831c7855f36cb8c05a5188abc7fd823c850e46625",
  38. host = "ec2-54-81-37-11x5.compute-1.amazonaws.com",
  39. port = "5432",
  40. database = "d8n5q9qux7r43ut")
  41.  
  42. cursor = connection.cursor()
  43.  
  44.  
  45. except (Exception, psycopg2.Error) as error :
  46. print ("Error while connecting to PostgreSQL", error)
  47. finally:
  48. #closing database connection.
  49. if(connection):
  50. cursor.close()
  51. connection.close()
  52. print("PostgreSQL connection is closed")
  53.  
  54.  
  55.  
  56. @app.route("/register", methods=["GET", "POST"])
  57. def register():
  58. return render_template("register.html")
  59. password = request.form.get("password")
  60. username = request.form.get("username")
  61. session.clear()
  62. db.execute("INSERT INTO users (username, password) VALUES (:username, :password")
  63. db.commit()
Add Comment
Please, Sign In to add comment