queuelib_fifo_disk_queue_push_pop_quickstart.py

1import os
2import shutil
3from queuelib import FifoDiskQueue
4
5# Define a path for the queue storage
6queue_path = "queue_dir"
7
8# Clean up any existing queue directory
9if os.path.exists(queue_path):
10    shutil.rmtree(queue_path)
11
12# Create the queue
13q = FifoDiskQueue(queue_path)
14
15# Push some data (must be bytes)
16q.push(b"first")
17q.push(b"second")
18q.push(b"third")
19
20# Pop data
21print(q.pop()) # yields b"first"
22print(q.pop()) # yields b"second"
23
24# Close the queue
25q.close()
26
27# Clean up
28if os.path.exists(queue_path):
29    shutil.rmtree(queue_path)