aiohttp web方式提供文件下載服務(wù)

python3.6
使用了aiohttp, aiofiles庫，異步操作
服務(wù)端代碼如下，啟動(dòng)服務(wù)后，默認(rèn)監(jiān)聽0.0.0.0:8080
沒有做任何異常處理，適用于小文件，僅供參考

file_server.py:

import aiofiles
import asyncio
import os
from aiohttp.web import Response
from aiohttp import web

def query_parse(req):
    obj = req.query_string
    queryitem = []
    if obj:
        query = req.query.items()
        for item in query:
            queryitem.append(item)
        return dict(queryitem)
    else:
        return None

async def download(request):
    query = query_parse(request)
    filename = query.get('file')
        # 沒有作任何異常處理，只有服務(wù)器上存在/tmp/files目錄，并且請(qǐng)求的文件在該目錄下，才能正常下載
    file_dir = '/tmp/files'
    file_path = os.path.join(file_dir, filename)
    if os.path.exists(file_path):
        async with aiofiles.open(file_path, 'rb') as f:
            content = await f.read()
        if content:
            response = Response(
                content_type='application/octet-stream',
                headers={'Content-Disposition': 'attachment;filename={}'.format(filename)},
                body=content
                            )
            return response

        else:
            return
    else:
        return

loop = asyncio.get_event_loop()
app = web.Application(loop=loop)
app.router.add_route(method='get', path='/download', handler=download)
web.run_app(app)
loop.close()

python3 file_server.py 運(yùn)行服務(wù)端

客戶端代碼如下：
download.py:

import aiohttp
import asyncio
import aiofiles

# 訪問服務(wù)端download接口函數(shù)，下載服務(wù)器上指定目錄（/tmp/files/）下的文件l5u.jpg
url = 'http://localhost:8080/download'
params = {'file': 'l5u.jpg'}

async def download():
    async with aiohttp.ClientSession() as session:
        async with session.get(url, params=params) as resp:
            content = await resp.content.read()
        async with aiofiles.open('/tmp/test.jpg', 'wb') as f:
            await f.write(content)

loop = asyncio.get_event_loop()
loop.run_until_complete(download())

執(zhí)行python3 download.py，查看/tmp/test.jpg是否存在，打開是否正常
或者直接使用瀏覽器訪問http://localhost:8080/download?file=l5u.jpg， 下載下來的文件名稱與請(qǐng)求發(fā)送的文件名稱（服務(wù)器上存在的文件名稱）一致