From 71ce702d755c67839f7d8e7abbc54dc44a8d7062 Mon Sep 17 00:00:00 2001 From: Liam Driscoll <driscolll4@cardiff.ac.uk> Date: Fri, 2 Dec 2022 05:57:12 +0000 Subject: [PATCH] Replacing .csv code with more appropriate database code. --- project_server.py | 13 ++++++++----- static/Manage_Coworking_Spaces.js | 1 - 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/project_server.py b/project_server.py index af115bc..0b8c9b0 100644 --- a/project_server.py +++ b/project_server.py @@ -1,8 +1,9 @@ import os -import csv +import sqlite3 from flask import Flask, redirect, request, render_template, jsonify app = Flask(__name__) +DATABASE = "project_db.db" ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif']) @@ -26,10 +27,12 @@ def addRecord(): recordCheckinInstructions = request.form['recordCheckinInstructions'] recordData = [recordName, recordAddress, recordMainPhotos, recordAdditionalPhotos, recordDescription, recordWebsite, recordEmail, recordPhoneNumber, recordOpeningHours, recordCheckinInstructions] - with open('coworking_spaces.csv', 'a') as addToFile: - csvWriter = csv.writer(addToFile) - csvWriter.writerow(recordData) - addToFile.close() + conn = sqlite3.connect(DATABASE) + cur = conn.cursor() + cur.execute("INSERT INTO coworking_spaces ('Name', 'Address', 'Main_Photo', 'Additional_Photos', 'Description', 'Website', 'Email', 'Phone_Number', 'Opening_Hours', 'Checkin_Instructions')\ + VALUES (?,?,?,?,?,?,?,?,?,?)", (recordName, recordAddress, recordMainPhotos, recordAdditionalPhotos, recordDescription, recordWebsite, recordEmail, recordPhoneNumber, recordOpeningHours, recordCheckinInstructions)) + conn.commit() + conn.close() infoMessage = (f'{recordData} Record added.') diff --git a/static/Manage_Coworking_Spaces.js b/static/Manage_Coworking_Spaces.js index 3a9e607..b0860b5 100644 --- a/static/Manage_Coworking_Spaces.js +++ b/static/Manage_Coworking_Spaces.js @@ -3,7 +3,6 @@ function onSubmitLoad() { document.getElementById('addButton').addEventListener('click',addRecord); } - function headerLoad() { let headerDiv = document.querySelector('#headerDiv'); let headerDivHeight = headerDiv.offsetHeight; -- GitLab