mia/0x0
3
8
Fork 3
Code Issues 10 Pull requests 2 Projects 1 Wiki Activity

Support user-specified expiration times #72

Manually merged
mia merged 14 commits from :expiration into master 2022-11-29 16:20:08 +01:00
Conversation 11 Commits 14 Files changed 6 +12 -12
1 changed files with 12 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit 19d989b696 - Show all commits

24
migrations/versions/939a08e1d6e5_.py
View file

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