唠唠闲话
之前介绍了『droppy 文件共享服务的搭建』。但在一些场景中，我们可能需要在命令行或在 Python 代码中，临时上传和下载文件。这时可以用一个更简单的策略：使用 flask 编写一个临时的 API。
服务端配置
以下是一个简单的 Flask 应用程序代码示例，用于处理文件的上传和下载。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
from flask import Flask, request, send_from_directory, jsonify
import os
os.makedirs('uploads', exist_ok=True)
app = Flask(__name__)
@app.route('/upload', methods=['POST'])
def upload_file():
if 'file' not in request.files:
return 'No file part', 400
file = request.files['file']
if file.filename == '':
return 'No selected file', 400
file.save(os.path.join('uploads', file.filename))
return 'File uploaded successfully', 200
@app.route('/files', methods=['GET'])
def list_files():
files = os.listdir('uploads')
return jsonify(files), 200
@app.route('/download/<filename>', methods=['GET'])
def download_file(filename):
try:
return send_from_directory('uploads', filename, as_attachment=True)
except FileNotFoundError:
return 'File not found', 404
if __name__ == '__main__':
app.run(host='0.0.0.0', port=5000)
|
运行 python server.py 命令即可启动该服务。
客户端交互
在代码中上传：将 file_path 填写为实际需要上传的文件，并相应修改 host 地址。
1
2
3
4
5
6
7
8
9
10
| import requests
file_path = 'path_to_your_file.txt'
with open(file_path, 'rb') as f:
files = {'file': f}
response = requests.post('http://server1_ip:5000/upload', files=files)
print(response.text)
|
也可以写成 Python 脚本形式，编写命令行工具 client.py，这里包括三个函数：
upload 上传文件list_files 查看文件列表download 下载文件
完整代码：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
| import requests
import click
from loguru import logger
@click.group()
def cli():
"""文件管理客户端."""
pass
@cli.command()
@click.option('--file', 'file_path', required=True, type=click.Path(exists=True), help='要上传的文件路径')
@click.option('--server-host', 'server_host', required=True, type=str, help='服务器的主机地址（包含IP和端口）')
def upload(file_path, server_host):
"""上传文件到指定服务器."""
url = f"http://{server_host}/upload"
logger.info(f"正在上传 {file_path} 到 {url}")
with open(file_path, 'rb') as f:
files = {'file': f}
try:
response = requests.post(url, files=files)
response.raise_for_status()
logger.info(f"服务器返回: {response.text}")
except requests.RequestException as e:
logger.error(f"发生错误: {e}")
@cli.command()
@click.option('--server-host', 'server_host', required=True, type=str, help='服务器的主机地址（包含IP和端口）')
def list_files(server_host):
"""查看服务器上的文件列表."""
url = f"http://{server_host}/files"
logger.info(f"正在从 {url} 检索文件列表")
try:
response = requests.get(url)
response.raise_for_status()
files = response.json()
if files:
logger.info("服务器上的文件列表:")
for file in files:
logger.info(f"- {file}")
else:
logger.info("服务器上没有可用文件。")
except requests.RequestException as e:
logger.error(f"发生错误: {e}")
@cli.command()
@click.option('--filename', 'filename', required=True, type=str, help='要下载的文件名')
@click.option('--server-host', 'server_host', required=True, type=str, help='服务器的主机地址（包含IP和端口）')
def download(filename, server_host):
"""从服务器下载文件."""
url = f"http://{server_host}/download/{filename}"
logger.info(f"正在下载 {filename} 从 {url}")
try:
response = requests.get(url, stream=True)
response.raise_for_status()
with open(filename, 'wb') as f:
for chunk in response.iter_content(chunk_size=8192):
f.write(chunk)
logger.info(f"{filename} 下载成功")
except requests.RequestException as e:
logger.error(f"发生错误: {e}")
if __name__ == '__main__':
cli()
|
使用示例
在命令行中执行以下命令来上传、查看和下载文件：
上传文件:
1
| python client.py upload --file path_to_your_file --server-host localhost:5000
|
查看文件列表:
1
| python client.py list_files --server-host localhost:5000
|
下载文件:
1
| python client.py download --filename file_name --server-host localhost:5000
|
以上，我们用 Python 的 flask 和 click 搭建一个简单的命令行文件共享服务。
wechat
alipay