def deleteComponentBucket():
ver = request.form['ver']
fileName = request.form['Fname']
if normal_db_functions.check_duplicate(database_address, fileName, ver):
url = normal_db_functions.get_URL(database_address, fileName, ver)
key = url[0][0].split("/")[-1]
my_bucket = get_bucket()
my_bucket.Object(key).delete()
normal_db_functions.delete_component(database_address, fileName, ver)
flash('File deleted successfully')
return redirect(url_for('files'))
def retrieve():
ver = request.args.get('ver')
fileName = request.args.get('Fname')
connection = normal_db_functions.create_connection(database_address)
if normal_db_functions.check_duplicate(database_address, fileName, ver):
url = normal_db_functions.get_URL(database_address, fileName, ver)
key = url[0][0].split("/")[-1]
print(key)
my_bucket = get_bucket()
file_obj = my_bucket.Object(key).get()
return Response(file_obj['Body'].read(),
mimetype='text/plain',
headers={
"Content-Disposition":
"attachment;filename={}".format(key)
})
else:
return jsonify("DOESN'T EXIST"), 404
def component():
file = request.files['file']
filetype = file.filename.split(".")[1]
ver = request.args.get('ver')
fileName = request.args.get('Fname')
URL = "https://capprojteam3.s3-ap-southeast-2.amazonaws.com/" + fileName + "".join(
ver.split('.')) + '.' + filetype
now = datetime.now() # current date and time
date_time = now.strftime("%d/%m/%Y, %H:%M:%S")
connection = normal_db_functions.create_connection(database_address)
if normal_db_functions.check_duplicate(database_address, fileName, ver):
return jsonify("DUPLICATE"), 400
normal_db_functions.create_component(connection,
(fileName, ver, date_time, URL))
my_bucket = get_bucket()
my_bucket.Object(fileName + "".join(ver.split('.')) + '.' +
filetype).put(Body=file)
# flash('File uploaded successfully')
return redirect(url_for('files'))
def test_check_duplicate_2(self):
duplicate = normal_db_functions.check_duplicate(db_file, "Bob", "1.2.3.4")
self.assertEqual(duplicate, False)
def test_check_duplicate_1(self):
duplicate = normal_db_functions.check_duplicate(db_file, "Thomas", "1.2.3.4")
self.assertEqual(duplicate, True)