Ember
/
0x0
forked from mia/0x0
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

More efficiently filter to unexpired files when migrating 

mia/0x0#72 (comment)
This commit is contained in:
Emi Simpson 2022-11-28 16:24:26 -05:00
parent 74b6f986ee
commit 19d989b696
No known key found for this signature in database
GPG Key ID: 45E9C6E81BD86E7C
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
migrations/versions/939a08e1d6e5_.py
View File

@ -62,7 +62,7 @@ def upgrade():
# List of file hashes which have not expired yet
# This could get really big for some servers
try:
unexpired_files = set(os.listdir(storage))
unexpired_files = os.listdir(storage)
except FileNotFoundError:
return # There are no currently unexpired files
@ -70,20 +70,20 @@ def upgrade():
files = session.scalars(
sa.select(File)
.where(
sa.not_(File.removed)
sa.not_(File.removed),
File.sha256.in_(unexpired_files)
)
)
for file in files:
if file.sha256 in unexpired_files:
file_path = storage / file.sha256
stat = os.stat(file_path)
max_age = get_max_lifespan(stat.st_size) # How long the file is allowed to live, in ms
file_birth = stat.st_mtime * 1000 # When the file was created, in ms
op.execute(
sa.update(UpdatedFile)
.where(UpdatedFile.c.id == file.id)
.values({'expiration': int(file_birth + max_age)})
)
file_path = storage / file.sha256
stat = os.stat(file_path)
max_age = get_max_lifespan(stat.st_size) # How long the file is allowed to live, in ms
file_birth = stat.st_mtime * 1000 # When the file was created, in ms
op.execute(
sa.update(UpdatedFile)
.where(UpdatedFile.c.id == file.id)
.values({'expiration': int(file_birth + max_age)})
)
def downgrade():
op.drop_column('file', 'expiration')