82 lines
3.0 KiB
Python
82 lines
3.0 KiB
Python
import requests
|
|
import io
|
|
import os
|
|
import concurrent.futures
|
|
import random
|
|
import string
|
|
|
|
# Endpoint
|
|
endpoint_url = "http://13.212.84.67:8000/api/files/"
|
|
|
|
# Number of files to send
|
|
num_files = 3
|
|
num_threads = 5 # Change the number of threads as needed
|
|
|
|
# Function to generate random file content
|
|
def generate_random_content():
|
|
content_length = random.randint(1024, 20480) # Random content size between 1KB and 20KB
|
|
return ''.join(random.choices(string.ascii_letters + string.digits, k=content_length))
|
|
|
|
# Function to send files in a thread and return the key
|
|
def send_files_thread(thread_id):
|
|
keys = []
|
|
for i in range(num_files):
|
|
# Generate random file content
|
|
content = generate_random_content()
|
|
file_content = io.BytesIO(content.encode())
|
|
|
|
# Define the files dictionary for the POST request
|
|
files = {'file': ('file{}.txt'.format(i), file_content)}
|
|
|
|
# Define any additional data you want to send with the request
|
|
data = {'ttl': 60}
|
|
|
|
# Make the POST request
|
|
response = requests.post(endpoint_url, files=files, data=data)
|
|
|
|
# Check if the request was successful
|
|
if response.status_code == 201:
|
|
items = response.json()
|
|
for item in items:
|
|
key = item.get('key')
|
|
if key:
|
|
keys.append(key)
|
|
print(f"Thread {thread_id}: File {i} uploaded successfully with key {key}.")
|
|
else:
|
|
print(f"Thread {thread_id}: File {i} uploaded, but key not found in the response.")
|
|
else:
|
|
print(f"Thread {thread_id}: File {i} upload failed with status code {response.status_code}.")
|
|
|
|
# Close the file content
|
|
file_content.close()
|
|
|
|
return keys
|
|
|
|
# Function to receive files in a thread using the provided key
|
|
def receive_files_thread(thread_id, file_key):
|
|
# Make the GET request to retrieve the file using the provided key
|
|
response = requests.get(endpoint_url + file_key)
|
|
|
|
# Check if the request was successful
|
|
if response.status_code == 200:
|
|
# Save or process the received file content here
|
|
print(f"Thread {thread_id}: File with key {file_key} received successfully.")
|
|
else:
|
|
print(f"Thread {thread_id}: File with key {file_key} retrieval failed with status code {response.status_code}.")
|
|
|
|
# Create and send/receive files using multiple threads
|
|
with concurrent.futures.ThreadPoolExecutor(max_workers=num_threads) as executor:
|
|
# Start multiple threads for sending files
|
|
send_futures = [executor.submit(send_files_thread, thread_id) for thread_id in range(num_threads)]
|
|
|
|
# Retrieve the keys from the responses
|
|
keys = []
|
|
for future in concurrent.futures.as_completed(send_futures):
|
|
keys.extend(future.result())
|
|
|
|
# Start multiple threads for receiving files using the obtained keys
|
|
receive_futures = [executor.submit(receive_files_thread, thread_id, file_key) for thread_id, file_key in enumerate(keys)]
|
|
|
|
# Wait for all receive threads to complete
|
|
concurrent.futures.wait(receive_futures)
|