mirror of
https://github.com/AmanTahiliani/PeerNotes.git
synced 2026-08-07 11:55:12 -04:00
Peer to Peer code along with test script (#18)
* Initial Peer service code * Updated peer service * Updated peer service * Updated file service * Updated User Ip registration * Updated File moving function * test * test1 again * new --------- Co-authored-by: Sahej Panag <spanag3@gatech.edu>
This commit is contained in:
111
file_peer/peer_service.py
Normal file
111
file_peer/peer_service.py
Normal file
@@ -0,0 +1,111 @@
|
||||
from flask import Flask, send_file, request
|
||||
import requests
|
||||
import socket
|
||||
import shutil
|
||||
import os
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
|
||||
@app.route('/copy-file', methods = ['POST'])
|
||||
def copy_file():
|
||||
try:
|
||||
data = request.json
|
||||
source_file = data['file_path']
|
||||
id = data['file_id']
|
||||
if not os.path.exists(source_file):
|
||||
return f"Source file does not exist at {source_file}", 400
|
||||
filename = os.path.basename(source_file)
|
||||
destination_path = os.path.join('./uploads/', str(id) +'_' + filename)
|
||||
shutil.copyfile(source_file, destination_path)
|
||||
return "File Moved successfully", 200
|
||||
except Exception as e:
|
||||
print(f'An error occured: {e}')
|
||||
return "An error occured", 400
|
||||
|
||||
|
||||
@app.route('/ip', methods=["GET"])
|
||||
def get_internal_ip():
|
||||
internal_ip = None
|
||||
try:
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
s.connect(("8.8.8.8", 80))
|
||||
internal_ip = s.getsockname()[0]
|
||||
s.close()
|
||||
except Exception as e:
|
||||
print("Error:", e)
|
||||
return internal_ip, 200
|
||||
|
||||
@app.route('/send', methods=['GET'])
|
||||
def send():
|
||||
file_id = request.args.get('id')
|
||||
filename = request.args.get('filename')
|
||||
|
||||
file_path = './uploads/' + file_id + '_' + filename
|
||||
|
||||
return send_file(file_path, as_attachment=True, download_name = str(file_id)+'_'+filename)
|
||||
|
||||
@app.route('/request', methods=['GET'])
|
||||
def request_file():
|
||||
data = request.json
|
||||
file_id = data['id']
|
||||
filename = data['filename']
|
||||
ip = data['ip']
|
||||
|
||||
|
||||
url = f"http://{ip}:8080/send"
|
||||
params = {
|
||||
'id': file_id,
|
||||
'filename': filename,
|
||||
}
|
||||
|
||||
response = requests.get(url, params=params)
|
||||
file_path = './uploads/' + str(file_id) + '_' + filename
|
||||
|
||||
if response.status_code == 200:
|
||||
if os.path.exists(file_path):
|
||||
return "File with name already exists", 200
|
||||
|
||||
with open(file_path, 'wb') as f:
|
||||
f.write(response.content)
|
||||
print("File downloaded successfully")
|
||||
return f'File Downloaded to location to location {file_path}', 200
|
||||
else:
|
||||
error = response.text
|
||||
print("Error:", error)
|
||||
return error, response.status_code
|
||||
|
||||
@app.route('/request-tests', methods=['GET'])
|
||||
def request_test_():
|
||||
data = request.json
|
||||
file_id = data['id']
|
||||
filename = data['filename']
|
||||
ip = data['ip']
|
||||
new_filename = data['new_filename']
|
||||
|
||||
|
||||
url = f"http://{ip}:8080/send"
|
||||
params = {
|
||||
'id': file_id,
|
||||
'filename': filename,
|
||||
}
|
||||
|
||||
response = requests.get(url, params=params)
|
||||
file_path = './uploads/' + str(file_id) + '_' + new_filename
|
||||
|
||||
if response.status_code == 200:
|
||||
if os.path.exists(file_path):
|
||||
return "File with name already exists", 200
|
||||
|
||||
with open(file_path, 'wb') as f:
|
||||
f.write(response.content)
|
||||
print("File downloaded successfully")
|
||||
return f'File Downloaded to location to location {file_path}', 200
|
||||
else:
|
||||
error = response.text
|
||||
print("Error:", error)
|
||||
return error, response.status_code
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(debug=True, host= '0.0.0.0', port=8080)
|
||||
2
file_peer/uploads/1_abcd.txt
Normal file
2
file_peer/uploads/1_abcd.txt
Normal file
@@ -0,0 +1,2 @@
|
||||
Hello World!
|
||||
Flask Notes Transfer Test
|
||||
2
file_peer/uploads/2_bcdef.txt
Normal file
2
file_peer/uploads/2_bcdef.txt
Normal file
@@ -0,0 +1,2 @@
|
||||
Hello World!
|
||||
Flask Notes Transfer Test 2
|
||||
Reference in New Issue
Block a user