voussoir/pixelcanvas
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add argument as_of to get chunk at or before a particular time.

Browse source
This commit is contained in:
voussoir 2020-02-11 15:38:04 -08:00
parent d649be2de5
commit f22746d389
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
pixelcanvas.py
View file

@ -88,17 +88,20 @@ def now():
# DB FUNCTIONS
################################################################################
def get_chunk_from_db(chunk_x, chunk_y):
def get_chunk_from_db(chunk_x, chunk_y, as_of=None):
'''
Get the chunk from the database, and raise IndexError if it doesn't exist.
'''
query = '''
query = f'''
SELECT x, y, data FROM chunks
WHERE x == ? AND y == ?
{'AND updated_at <= ?' if as_of is not None else ''}
ORDER BY updated_at DESC
LIMIT 1
'''
bindings = [chunk_x, chunk_y]
if as_of is not None:
bindings.append(as_of)
cur.execute(query, bindings)
fetch = cur.fetchone()
if fetch is None:
@ -107,16 +110,16 @@ def get_chunk_from_db(chunk_x, chunk_y):
data = gzip.decompress(data)
return (x, y, data)
def get_chunk(chunk_x, chunk_y):
def get_chunk(chunk_x, chunk_y, *args, **kwargs):
'''
Get the chunk from the database if it exists, or else download it.
'''
try:
return get_chunk_from_db(chunk_x, chunk_y)
return get_chunk_from_db(chunk_x, chunk_y, *args, **kwargs)
except IndexError:
(bigchunk_x, bigchunk_y) = chunk_to_bigchunk(chunk_x, chunk_y)
download_bigchunk(bigchunk_x, bigchunk_y)
return get_chunk_from_db(chunk_x, chunk_y)
return get_chunk_from_db(chunk_x, chunk_y, *args, **kwargs)
def insert_chunk(chunk_x, chunk_y, data, commit=True):
try: