def create_schema():
creation_script_file = 'data/db_schema/01_create_schema.sql'
with open(creation_script_file) as schema_script:
with establish_connection() as conn, \
conn.cursor() as cursor:
try:
sql_to_run = schema_script.read()
cursor.execute(sql_to_run)
print("Database schema created")
except Exception as ex:
print("Schema creation failed")
print(ex.args)
def init_db():
init_conn = get_connection_data('postgres')
db_name = os.environ.get('MY_PSQL_DBNAME')
with establish_connection(connection_data=init_conn) as conn:
with conn.cursor() as cursor:
try:
drop_statement = 'DROP DATABASE IF EXISTS "{}";'.format(
db_name)
create_statement = 'CREATE DATABASE "{}";'.format(db_name)
cursor.execute(drop_statement)
cursor.execute(create_statement)
print("Database created")
except Exception as ex:
print("Database creation failed")
print(ex.args)
def registration():
try:
if request.method == "POST":
name = request.form['name']
email = request.form['email']
username = request.form['username']
password = request.form['password']
confirm_pw = request.form['confirm']
if password == confirm_pw:
password = pw_hash.hash_pw(password)
data_manager.reg_new_user(data_manager.establish_connection(), name, email, username, password)
else:
message = "Passwords don't match. Try again."
return render_template('main.html', message=message)
except psycopg2.IntegrityError:
return redirect('/')
return render_template('main.html')
def youtube_link(show_id):
link = request.form["youtube_link"]
link = link.replace("watch?v=", "embed/")
data_manager.youtube_link(data_manager.establish_connection(), link, show_id)
return redirect(url_for("detailed", show_id=show_id))
def restore_deleted_show():
show_id = request.form['deletedShowId']
data_manager.restore(data_manager.establish_connection(), show_id)
return redirect(url_for("deleted_page"))
def delete_show(show_id):
data_manager.delete(data_manager.establish_connection(), show_id)
return redirect("/")