企业商务网站的技术,网站o2o,学校门户网站建设的好处,北航做网站公司之前介绍了 droppy 文件共享服务的搭建。但在一些场景中#xff0c;我们需要在命令行或在 Python 代码中#xff0c;临时上传和下载文件。这时可以用一个更简单的策略#xff1a;使用 flask 编写一个临时的 API。
服务端配置
以下是一个简单的 Flask 应用程序代码示例我们需要在命令行或在 Python 代码中临时上传和下载文件。这时可以用一个更简单的策略使用 flask 编写一个临时的 API。
服务端配置
以下是一个简单的 Flask 应用程序代码示例用于处理文件的上传和下载。
# server.py
from flask import Flask, request, send_from_directory, jsonify
import os# 创建 uploads 目录如果不存在
os.makedirs(uploads, exist_okTrue)app Flask(__name__)app.route(/upload, methods[POST])
def upload_file():if file not in request.files:return No file part, 400file request.files[file]if file.filename :return No selected file, 400# 保存文件到服务器file.save(os.path.join(uploads, file.filename))return File uploaded successfully, 200app.route(/files, methods[GET])
def list_files():# 获取 uploads 目录中的文件列表files os.listdir(uploads)return jsonify(files), 200app.route(/download/filename, methods[GET])
def download_file(filename):# 下载指定的文件try:return send_from_directory(uploads, filename, as_attachmentTrue)except FileNotFoundError:return File not found, 404if __name__ __main__:app.run(host0.0.0.0, port5000)运行 python server.py 命令启动服务。
客户端交互
在代码中上传将 file_path 填写为实际需要上传的文件并相应修改 host 地址。
import requests# 要上传的文件路径
file_path path_to_your_file.txtwith open(file_path, rb) as f:files {file: f}response requests.post(http://server1_ip:5000/upload, filesfiles)print(response.text)也可以写成 Python 脚本形式编写命令行工具 client.py这里包括三个函数
upload 上传文件list_files 查看文件列表download 下载文件
完整代码
import requests
import click
from loguru import loggerclick.group()
def cli():文件管理客户端.passcli.command()
click.option(--file, file_path, requiredTrue, typeclick.Path(existsTrue), help要上传的文件路径)
click.option(--server-host, server_host, requiredTrue, typestr, help服务器的主机地址包含IP和端口)
def upload(file_path, server_host):上传文件到指定服务器.url fhttp://{server_host}/uploadlogger.info(f正在上传 {file_path} 到 {url})with open(file_path, rb) as f:files {file: f}try:response requests.post(url, filesfiles)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, requiredTrue, typestr, help服务器的主机地址包含IP和端口)
def list_files(server_host):查看服务器上的文件列表.url fhttp://{server_host}/fileslogger.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, requiredTrue, typestr, help要下载的文件名)
click.option(--server-host, server_host, requiredTrue, typestr, help服务器的主机地址包含IP和端口)
def download(filename, server_host):从服务器下载文件.url fhttp://{server_host}/download/{filename}logger.info(f正在下载 {filename} 从 {url})try:response requests.get(url, streamTrue)response.raise_for_status()with open(filename, wb) as f:for chunk in response.iter_content(chunk_size8192):f.write(chunk)logger.info(f{filename} 下载成功)except requests.RequestException as e:logger.error(f发生错误: {e})if __name__ __main__:cli()使用示例
在命令行中执行以下命令来上传、查看和下载文件
上传文件:
python client.py upload --file path_to_your_file --server-host localhost:5000查看文件列表:
python client.py list_files --server-host localhost:5000下载文件:
python client.py download --filename file_name --server-host localhost:5000以上我们用 Python 的 flask 和 click 搭建一个简单的命令行文件共享服务。
