df$pdZddlZddlZddlZddlZddlmZmZddl m Z dZ dZ dZ dZGd d Zy) zServe GridFS files with Motor and aiohttp. Requires Python 3.5 or later and aiohttp 3.0 or later. See the :doc:`/examples/aiohttp_gridfs_example`. N)AsyncIOMotorDatabaseAsyncIOMotorGridFSBucket) _hash_gridoutc$|j|S)a&Override to choose a GridFS file to serve at a URL. By default, if a URL pattern like ``/fs/{filename}`` is mapped to this :class:`AIOHTTPGridFS`, then the filename portion of the URL is used as the filename, so a request for "/fs/image.png" results in a call to :meth:`.AsyncIOMotorGridFSBucket.open_download_stream_by_name` with "image.png" as the ``filename`` argument. To customize the mapping of path to GridFS file, override ``get_gridfs_file`` and return a :class:`asyncio.Future` that resolves to a :class:`~motor.motor_asyncio.AsyncIOMotorGridOut`. For example, to retrieve the file by ``_id`` instead of filename:: def get_gridfile_by_id(bucket, filename, request): # "filename" is interpreted as _id instead of name. # Return a Future AsyncIOMotorGridOut. return bucket.open_download_stream(file_id=filename) client = AsyncIOMotorClient() gridfs_handler = AIOHTTPGridFS(client.my_database, get_gridfs_file=get_gridfile_by_id) :Parameters: - `bucket`: An :class:`~motor.motor_asyncio.AsyncIOMotorGridFSBucket` - `filename`: A string, the URL portion matching {filename} in the URL pattern - `request`: An :class:`aiohttp.web.Request` )open_download_stream_by_name)bucketfilenamerequests O/home/api-vastappli/venv/lib/python3.12/site-packages/motor/aiohttp/__init__.pyget_gridfs_filer "s<  . .x 88cy)a5Override to customize cache control behavior. Return a positive number of seconds to trigger aggressive caching or 0 to mark resource as cacheable, only. 0 is the default. For example, to allow image caching:: def image_cache_time(filename, modified, mime_type): if mime_type.startswith('image/'): return 3600 return 0 client = AsyncIOMotorClient() gridfs_handler = AIOHTTPGridFS(client.my_database, get_cache_time=image_cache_time) :Parameters: - `filename`: A string, the URL portion matching {filename} in the URL pattern - `modified`: A datetime, when the matching GridFS file was created - `mime_type`: The file's type, a string like "application/octet-stream" r)r modified mime_types r get_cache_timerCs0 r cy)a,Override to modify the response before sending to client. For example, to allow image caching:: def gzip_header(response, gridout): response.headers['Content-Encoding'] = 'gzip' client = AsyncIOMotorClient() gridfs_handler = AIOHTTPGridFS(client.my_database, set_extra_headers=gzip_header) :Parameters: - `response`: An :class:`aiohttp.web.Response` - `gridout`: The :class:`~motor.motor_asyncio.AsyncIOMotorGridOut` we will serve to the client Nr)responsegridouts r set_extra_headersr^sr c |jjjjd}d|z}t jj|d#tt f$rd|z}Y7wxYw)N formatterz>@M LyX ++ - -3 - 7TA n %B7'ABs6AA0/A0c,eZdZdZdeeefdZdZdZ y) AIOHTTPGridFSa.Serve files from `GridFS`_. This class is a :ref:`request handler ` that serves GridFS files, similar to aiohttp's built-in static file server. .. code-block:: python client = AsyncIOMotorClient() gridfs_handler = AIOHTTPGridFS(client.my_database) app = aiohttp.web.Application() # The GridFS URL pattern must have a "{filename}" variable. resource = app.router.add_resource("/fs/{filename}") resource.add_route("GET", gridfs_handler) resource.add_route("HEAD", gridfs_handler) app_handler = app.make_handler() server = loop.create_server(app_handler, port=80) By default, requests' If-Modified-Since headers are honored, but no specific cache-control timeout is sent to clients. Thus each request for a GridFS file requires a quick check of the file's ``uploadDate`` in MongoDB. Pass a custom :func:`get_cache_time` to customize this. :Parameters: - `database`: An :class:`AsyncIOMotorDatabase` - `get_gridfs_file`: Optional override for :func:`get_gridfs_file` - `get_cache_time`: Optional override for :func:`get_cache_time` - `set_extra_headers`: Optional override for :func:`set_extra_headers` .. _GridFS: https://www.mongodb.com/docs/manual/core/gridfs/ fsct|tstd|z||_t |j||_||_||_||_y)NzDFirst argument to AIOHTTPGridFS must be AsyncIOMotorDatabase, not %r) isinstancer TypeError _databaser_bucket_get_gridfs_file_get_cache_time_set_extra_headers)selfdatabaseroot_collectionr rrs r __init__zAIOHTTPGridFS.__init__sX($89VYaa "/P /-"3r cK |jd}|jdvr-tj j |jddh |j|j|d{}tj j}t|}|j|j||||j!|||j"}|Z|j%|j&j(}|j&j%d } || k\r|j+d |S|j,j/d } | '| j1d |k(r|j+d |S|j2|_|j7|d{|jdk(rkd } | |j2krZ|j9|j:d{} |j=| d{| t?| z } | |j2krZ|S#t$rt|Y#wxYw7#tj$r0}tj j|j|d}~wwxYw777w)z&Send filepath to client using request.r )GETHEADr6r7)methodallowed_methodsNrtzinfor) microsecondi0z If-None-Match") rrr%r8r!r"HTTPMethodNotAllowedr.r-gridfsNoFile HTTPNotFoundpathStreamResponser_set_standard_headersr0if_modified_sincereplace upload_dater; set_statusheadersgetstriplengthcontent_lengthprepareread chunk_sizewritelen) r1r r rerespchecksum ims_valueif_sinceretagwrittenchunks r __call__zAIOHTTPGridFS.__call__sJ #))*5H >> 0++22~~v3  E 11$,,'RRG{{))+!) ""7<<wI g.--  !((0C0C0J0J(KH**22q2AH8#$ ""?3   38 ; OOC K%nnll7### >>U "GGNN*%ll7+=+=>>jj'''3u:% GNN*  g # ' " #S}} E++** *=1 D E@ $ ?'sJ H;J  H=/H:0H=4D*J JAJ !J"J :J;!J J H73J 6H77J :H==J+I;;JJ J J cJ|j|_|j}|tj|\}}|r||_d|z|j d<|j ||j|j}|dkDrtjjtjjjdtj|zjd|j d<dt|z|j d <yd |j d <y) Nz"%s"Etagrr:)secondsz%a, %d %b %Y %H:%M:%S GMTExpireszmax-age=z Cache-Controlpublic)rG last_modified content_type mimetypes guess_typerIr/datetimenowtimezoneutcrF timedeltastrftimestr)r1rBrTrrUrbencoding cache_times r rDz#AIOHTTPGridFS._set_standard_headerss$00++  %.%9%9$%? "L(  ,D %0 V))$0C0CWEYEYZ >!!%%h&7&7&;&;<DDDDQ$$Z89h23 LL # -7Z,HDLL ),4DLL )r N) __name__ __module__ __qualname____doc__r rrr4r[rDrr r r'r'|s& J'%+ 4&7r5r r')rqrerc aiohttp.webr!r?motor.motor_asynciorrmotor.motor_gridfsrr rrr%r'rr r rus@ N, 9B 6&BE5E5r