Package: lazyreader
lazyreader is a Python module for doing lazy reading of file objects.The Python standard library lets you read a file a line-at-a-time, saving you from loading the entire file into memory.
For example:withopen('large_file.txt')asf:forlineinf:print(line)lazyreader lets you do the same thing, but with an arbitrary delimiter, and for any object that presents a.read()method.
For example:fromlazyreaderimportlazyreadwithopen('large_file.txt')asf:fordocinlazyread(f,delimiter=';'):print(doc)This is a snippet of code I spun out from theWellcome Digital Platform.
We have large XML and JSON files stored in S3 – sometimes multiple GBs – but each file is really a series of “documents”, separated by known delimiters.
Downloading and parsing the entire file would be prohibitively expensive, but lazyreader allows us to hold just a single document in memory at a time.Installationlazyreader is available from PyPI:$pipinstalllazyreaderExamplesIf we have a file stored locally, we can open it and split based on any choice of delimiter.
For example, if we had a text file in which record were separated by commas:withopen('lots_of_records.txt')asf:fordocinlazyread(f,delimiter=','):print(doc)Another example: we have a file stored in Amazon S3, and we’d like to read it line-by-line.
Theboto3API gives us a file object for reading from S3:importboto3client=boto3.client('s3')s3_object=client.get_object(Bucket='example-bucket',Key='words.txt')body=s3_object['Body']fordocinlazyread(body,delimiter=b'\n'):print(doc)(This is the use case for which this code was originally written.)One more example: we’re fetching an HTML page, and want to read lines separated by<br>in the underlying HTML.
Like so:importurllib.requestwithurllib.request.urlopen('https://example.org/')asf:fordocinlazyread(f,delimiter=b'<br>'):print(doc)Advanced usagelazyread()returns a generator, which you can wrap to build a pipeline of generators which do processing on the data.First example: we have a file which contains a list of JSON objects, one per line.
(This is the format of output files fromelasticdump.)
What the caller really needs is Python dictionaries, not JSON strings.
We can wraplazyread()like so:importjsondeflazyjson(f,delimiter=b'\n'):fordocinlazyread(f,delimiter=delimiter):# Ignore empty lines, e.g. the last line in a fileifnotdoc.strip():continueyieldjson.loads(doc)Another example: we want to parse a large XML file, but not load it all into memory at once.
We can write the following wrapper:fromlxmlimportetreedeflazyxmlstrings(f,opening_tag,closing_tag):fordocinlazyread(f,delimiter=closing_tag):ifopening_tagnotindoc:continue# We want complete XML blocks, so look for the opening tag and# just return its contentsblock=doc.split(opening_tag)[-1]yieldopening_tag+blockdeflazyxml(f,opening_tag,closing_tag):forxml_stringinlazyxmlstrings(f,opening_tag,closing_tag):yieldetree.fromstring(xml_string)We use both of these wrappers at Wellcome to do efficient processing of large files that are kept in Amazon S3.Isn’t this a bit simple to be a module?Maybe.
There are recipes on Stack Overflow that do very similar, but I find it useful to have in a standalone module.And it’s not completely trivial – at least, not for me.
I made two mistakes when I first wrote this:I was hard-coding the initial running string asrunning=b''That only works if your file object is returning bytestrings.
If it’s returning Unicode strings, you get aTypeError(can’t concat bytes to str) when it first tries to read from the file.
String types are important!After I’d read another 1024 characters from the file, I checked for the delimiter like so:running+=new_dataifdelimiterinrunning:curr,running=running.split(delimiter)yieldcurr+delimiterFor my initial use case, individual documents weremuchbigger than 1024 characters, so the new data would never contain multiple delimiters.
But with smaller documents, you might get multiple delimiters in one read, and then unpacking the result of.split()would throw aValueError.
So now the code correctly checks and handles the case where a single read includes more than one delimiter.Now it’s encoded and tested in a module, I don’t have to worry about making the same mistakes again.LicenseMIT.