通過Python中的http.server搭建文件上傳下載服務(wù)功能
Python搭建文件下載服務(wù)
在Web開發(fā)中，文件下載服務(wù)是一個(gè)常見且基礎(chǔ)的功能。Python的http.server模塊提供了一個(gè)簡單而強(qiáng)大的方式來搭建HTTP服務(wù)器，進(jìn)而實(shí)現(xiàn)文件下載服務(wù)。本文將結(jié)合實(shí)際案例，詳細(xì)介紹如何使用Python的http.server模塊來搭建文件下載服務(wù)。
一、概述
Python的http.server模塊是Python標(biāo)準(zhǔn)庫的一部分，它提供了一個(gè)基本的HTTP服務(wù)器類和請(qǐng)求處理器。通過這個(gè)模塊，我們可以輕松地搭建一個(gè)HTTP服務(wù)器，用于處理客戶端的HTTP請(qǐng)求，包括文件下載請(qǐng)求。
1.1 準(zhǔn)備工作
在開始之前，請(qǐng)確保你的計(jì)算機(jī)上已安裝Python環(huán)境。Python 3.x版本已經(jīng)內(nèi)置了http.server模塊，因此你不需要額外安裝任何庫。
1.2 場(chǎng)景描述
假設(shè)我們有一個(gè)需求：需要搭建一個(gè)HTTP服務(wù)器，用于提供特定目錄下的文件下載服務(wù)。客戶端通過訪問服務(wù)器上的特定URL，即可下載服務(wù)器上的文件。
二、搭建HTTP服務(wù)器
2.1 基礎(chǔ)服務(wù)器搭建
首先，我們將使用http.server模塊搭建一個(gè)基礎(chǔ)的HTTP服務(wù)器。
2.1.1 示例代碼
import http.server
import socketserver
# 設(shè)置端口號(hào)
PORT = 8000
# 創(chuàng)建一個(gè)TCP服務(wù)器
with socketserver.TCPServer(("", PORT), http.server.SimpleHTTPRequestHandler) as httpd:
print("Serving at port", PORT)
# 啟動(dòng)服務(wù)器，使其一直運(yùn)行
httpd.serve_forever()
在上述代碼中，我們導(dǎo)入了http.server和socketserver模塊，并設(shè)置了服務(wù)器監(jiān)聽的端口號(hào)為8000。然后，我們創(chuàng)建了一個(gè)TCPServer實(shí)例，并將http.server.SimpleHTTPRequestHandler作為請(qǐng)求處理器傳入。最后，我們調(diào)用serve_forever()方法啟動(dòng)服務(wù)器，使其持續(xù)運(yùn)行。
2.1.2 運(yùn)行服務(wù)器
將上述代碼保存為一個(gè).py文件，例如server.py。然后，在命令行中導(dǎo)航到該文件所在的目錄，并執(zhí)行以下命令：
python server.py
執(zhí)行后，你將在控制臺(tái)看到“Serving at port 8000”的提示，表示服務(wù)器已成功啟動(dòng)。
2.2 自定義文件目錄
默認(rèn)情況下，SimpleHTTPRequestHandler會(huì)處理服務(wù)器當(dāng)前工作目錄下的文件。然而，我們可能希望服務(wù)器處理特定目錄下的文件。為了實(shí)現(xiàn)這一需求，我們可以創(chuàng)建一個(gè)繼承自SimpleHTTPRequestHandler的類，并重寫translate_path方法。
2.2.1 示例代碼
import http.server
import socketserver
import os
# 定義服務(wù)器端口
PORT = 8080
# 定義文件目錄
DIRECTORY = "/path/to/your/directory"
# 創(chuàng)建一個(gè)處理請(qǐng)求的類
class MyHandler(http.server.SimpleHTTPRequestHandler):
def translate_path(self, path):
# 確保返回的路徑在指定的目錄內(nèi)
path = os.path.normpath(os.path.join(DIRECTORY, path.lstrip('/')))
return path
# 創(chuàng)建服務(wù)器
with socketserver.TCPServer(("", PORT), MyHandler) as httpd:
print(f"Serving at port {PORT}")
# 啟動(dòng)服務(wù)器，使其一直運(yùn)行
httpd.serve_forever()在上述代碼中，我們定義了DIRECTORY變量來指定文件所在的目錄，并創(chuàng)建了一個(gè)MyHandler類，該類繼承自SimpleHTTPRequestHandler并重寫了translate_path方法。在translate_path方法中，我們通過os.path.join和os.path.normpath函數(shù)將請(qǐng)求的路徑與DIRECTORY變量拼接起來，并確保返回的路徑在指定的目錄內(nèi)。
2.2.2 運(yùn)行服務(wù)器
將上述代碼保存為一個(gè).py文件，并替換DIRECTORY變量的值為你希望服務(wù)器處理的文件目錄。然后，按照2.1.2節(jié)中的步驟運(yùn)行服務(wù)器。
2.3 上傳下載服務(wù)
有時(shí)，我們可能需要定制HTTP響應(yīng)頭，以滿足特定的需求。
2.3.1 示例代碼
為了定制響應(yīng)頭，我們繼續(xù)上面的例子，通過重寫do_GET方法來設(shè)置特定的HTTP響應(yīng)頭，比如Content-Disposition以支持文件下載。
import os
import cgi
from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.parse import unquote
class SimpleHTTPRequestHandler(BaseHTTPRequestHandler):
def do_GET(self):
# Parse the URL to get the file path
file_path = unquote(self.path.strip("/"))
if os.path.isfile(file_path):
# If the requested path is a file, serve the file as an attachment
self.send_response(200)
self.send_header('Content-Type', 'application/octet-stream')
# self.send_header('Content-Disposition', f'attachment; filename="{os.path.basename(file_path)}"')
self.end_headers()
with open(file_path, 'rb') as file:
self.wfile.write(file.read())
else:
# Otherwise, show the upload form and file list
self.send_response(200)
self.send_header('Content-type', 'text/html')
self.end_headers()
# List files in the current directory
file_list = os.listdir('.')
file_list_html = ''.join(f'<li><a href="/{file}" rel="external nofollow" >{file}</a></li>' for file in file_list)
self.wfile.write(f'''
<html>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
<body>
<form enctype="multipart/form-data" method="post">
<input type="file" name="file">
<input type="submit" value="Upload">
</form>
<h2>Files in current directory:</h2>
<ul>
{file_list_html}
</ul>
</body>
</html>
'''.encode())
def do_POST(self):
try:
content_type = self.headers['Content-Type']
if 'multipart/form-data' in content_type:
form = cgi.FieldStorage(
fp=self.rfile,
headers=self.headers,
environ={'REQUEST_METHOD': 'POST'}
)
file_field = form['file']
if file_field.filename:
# Save the file
with open(os.path.join('.', file_field.filename), 'wb') as f:
f.write(file_field.file.read())
self.send_response(200)
self.send_header('Content-type', 'text/html')
self.end_headers()
self.wfile.write(b'''
<html>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
<body>
<p>File uploaded successfully.</p>
<script>
setTimeout(function() {
window.location.href = "/";
}, 2000);
</script>
</body>
</html>
''')
else:
self.send_response(400)
self.send_header('Content-type', 'text/html')
self.end_headers()
self.wfile.write(b'''
<html>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
<body>
<p>No file uploaded.</p>
<script>
setTimeout(function() {
window.location.href = "/";
}, 2000);
</script>
</body>
</html>
''')
else:
self.send_response(400)
self.send_header('Content-type', 'text/html')
self.end_headers()
self.wfile.write(b'''
<html>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
<body>
<p>Invalid content type.</p>
<script>
setTimeout(function() {
window.location.href = "/";
}, 2000);
</script>
</body>
</html>
''')
except Exception as e:
self.send_response(500)
self.send_header('Content-type', 'text/html')
self.end_headers()
self.wfile.write(f'''
<html>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
<body>
<p>Error occurred: {str(e)}</p>
<script>
setTimeout(function() {{
window.location.href = "/";
}}, 2000);
</script>
</body>
</html>
'''.encode())
def run(server_class=HTTPServer, handler_class=SimpleHTTPRequestHandler, port=8000):
server_address = ('', port)
httpd = server_class(server_address, handler_class)
# Uncomment the following lines if you need HTTPS support
# httpd.socket = ssl.wrap_socket(httpd.socket,
# keyfile="path/to/key.pem",
# certfile='path/to/cert.pem', server_side=True)
print(f'Starting httpd server on port {port}...')
httpd.serve_forever()
if __name__ == '__main__':
run()在上述代碼中，CustomFileHandler類繼承自之前定義的MyHandler類（或直接從http.server.SimpleHTTPRequestHandler繼承，如果你沒有重寫translate_path方法的話）。我們重寫了do_GET方法，以處理GET請(qǐng)求并定制響應(yīng)頭。
- 首先，我們通過
translate_path方法獲取請(qǐng)求文件的實(shí)際路徑。 - 然后，我們檢查該文件是否存在。
- 如果文件存在，我們使用
mimetypes.guess_type函數(shù)猜測(cè)文件的MIME類型，并設(shè)置相應(yīng)的Content-type響應(yīng)頭。 - 最后，我們打開文件，并使用
copyfile方法將文件內(nèi)容發(fā)送給客戶端。
如果文件不存在，我們通過send_error方法返回404錯(cuò)誤。
瀏覽器訪問http://localhost:8000/
能看到下載列表
點(diǎn)擊選擇文件可以上傳
2.3.2 運(yùn)行服務(wù)器
將上述代碼保存為一個(gè).py文件，并替換DIRECTORY變量的值為你希望服務(wù)器處理的文件目錄。然后，按照之前的步驟運(yùn)行服務(wù)器。
現(xiàn)在，當(dāng)你通過瀏覽器訪問服務(wù)器上的文件時(shí)（例如，http://localhost:8080/example.pdf），瀏覽器將以下載的方式處理該文件，而不是嘗試在瀏覽器中打開它。
三、安全性考慮
雖然http.server模塊提供了一個(gè)快速搭建HTTP服務(wù)器的方法，但它在安全性方面存在一些局限性。例如，默認(rèn)情況下，它會(huì)處理服務(wù)器上所有可讀文件的請(qǐng)求，這可能會(huì)暴露敏感信息。
為了增強(qiáng)安全性，你可以：
- 限制服務(wù)器的訪問權(quán)限，例如，使用防火墻規(guī)則來限制只有特定的IP地址或網(wǎng)絡(luò)才能訪問服務(wù)器。
- 創(chuàng)建一個(gè)白名單，僅允許訪問白名單中指定的文件或目錄。
- 使用更安全的HTTP服務(wù)器框架，如Flask或Django，這些框架提供了更豐富的功能和更好的安全性支持。
四、總結(jié)
通過本教程，我們學(xué)習(xí)了如何使用Python的http.server模塊搭建一個(gè)基本的HTTP服務(wù)器，并實(shí)現(xiàn)文件下載服務(wù)。我們介紹了如何設(shè)置服務(wù)器端口、自定義文件目錄、定制HTTP響應(yīng)頭以及處理GET請(qǐng)求。最后，我們還討論了使用http.server模塊時(shí)需要注意的一些安全性問題。希望這些內(nèi)容對(duì)你有所幫助！
到此這篇關(guān)于通過Python中的http.server搭建文件上傳下載服務(wù)功能的文章就介紹到這了,更多相關(guān)Python http.server文件上傳下載服務(wù)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
- Python 基于http.server模塊實(shí)現(xiàn)簡單http服務(wù)的代碼舉例
- Python中的http.server庫用法詳細(xì)介紹
- 使用Python?http.server模塊共享文件的方法詳解
- Python實(shí)現(xiàn)http服務(wù)器(http.server模塊傳參?接收參數(shù))實(shí)例
- python3之http.server模塊用法
- 詳解用python -m http.server搭一個(gè)簡易的本地局域網(wǎng)
- Python通過命令開啟http.server服務(wù)器的方法
- python中http.server內(nèi)置庫構(gòu)建臨時(shí)文件服務(wù)
相關(guān)文章
python實(shí)現(xiàn)人像動(dòng)漫化的示例代碼
這篇文章主要介紹了python實(shí)現(xiàn)人像動(dòng)漫化的示例代碼，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-05-05
Python實(shí)現(xiàn)網(wǎng)絡(luò)聊天室的示例代碼(支持多人聊天與私聊)
這篇文章主要介紹了Python實(shí)現(xiàn)網(wǎng)絡(luò)聊天室的示例代碼(支持多人聊天與私聊)，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-01-01
python統(tǒng)計(jì)列表中元素出現(xiàn)次數(shù)的三種方法
這篇文章主要介紹了python統(tǒng)計(jì)列表中元素出現(xiàn)次數(shù)的三種方法,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面跟著小編來一起學(xué)習(xí)學(xué)習(xí)吧2024-08-08
python pandas時(shí)序處理相關(guān)功能詳解
這篇文章主要介紹了python pandas時(shí)序處理相關(guān)功能詳解的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了anaconda如何查看并管理python環(huán)境，Anaconda是Python的一個(gè)開源發(fā)行版本，主要面向科學(xué)計(jì)算，預(yù)裝了豐富強(qiáng)大的庫。使用Anaconda可以輕松管理多個(gè)版本的Python環(huán)境,需要的朋友可以參考下2019-07-07
網(wǎng)易2016研發(fā)工程師編程題 獎(jiǎng)學(xué)金（python）
這篇文章主要為大家詳細(xì)介紹了網(wǎng)易2016研發(fā)工程師編程題：獎(jiǎng)學(xué)金（python），具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-06-06
在pycharm中使用pipenv創(chuàng)建虛擬環(huán)境和安裝django的詳細(xì)教程
這篇文章主要介紹了在pycharm中使用pipenv來創(chuàng)建虛擬環(huán)境和安裝django的詳細(xì)教程，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-11-11
Python內(nèi)置方法和屬性應(yīng)用：反射和單例(推薦)
這篇文章主要介紹了Python內(nèi)置方法和屬性應(yīng)用：反射和單例,本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-06-06
這篇文章主要介紹了Python matplotlib可視化實(shí)例解析,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-06-06
Tornado龍卷風(fēng)是一個(gè)開源的網(wǎng)絡(luò)服務(wù)器框架，它是基于社交聚合網(wǎng)站FriendFeed的實(shí)時(shí)信息服務(wù)開發(fā)而來的。這篇文章主要介紹了Python Tornado批量上傳圖片并顯示,需要的朋友可以參考下2020-03-03