moonq/mini-flees
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Passwords for files

Browse Source
This commit is contained in:
Q
2023-08-20 22:23:42 +03:00
parent 80bce10987
commit 9af21b5f58
7 changed files with 103 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
code/utils/files.py
View File

@@ -13,14 +13,14 @@ def get_db():
return db, c
def db_store_file(token, name, expires, max_dl):
def db_store_file(token, name, expires, max_dl, password):
db, c = get_db()
c.execute(
"""
insert into files(token,name,added,expires,downloads,max_downloads)
values (?,?,?,?,?,?)
insert into files(token,name,added,expires,downloads,max_downloads,passhash)
values (?,?,?,?,?,?,?)
""",
(token, name, int(time.time()), expires, 0, max_dl),
(token, name, int(time.time()), expires, 0, max_dl, password),
)
if c.rowcount > 0:
db.commit()
@@ -31,7 +31,7 @@ def db_get_file(token, name):
db, c = get_db()
return db.execute(
"""
SELECT added,expires,downloads,max_downloads
SELECT added,expires,downloads,max_downloads,passhash
FROM files WHERE token = ? AND name = ?
""",
(token, name),
@@ -42,7 +42,7 @@ def db_get_files():
db, c = get_db()
return db.execute(
"""
SELECT token,name,added,expires,downloads,max_downloads
SELECT token,name,added,expires,downloads,max_downloads,passhash
FROM files
WHERE expires > ? and (downloads < max_downloads or max_downloads = -1)
ORDER BY added
@@ -157,7 +157,7 @@ def file_details(token, name):
s = os.stat(full_path)
db_stat = db_get_file(token, name)
if db_stat:
added, expires, downloads, max_dl = db_stat
added, expires, downloads, max_dl, password = db_stat
else:
return {}
return {
@@ -169,6 +169,7 @@ def file_details(token, name):
"expires": file_time_human(expires),
"downloaded": downloads,
"max-dl": max_dl,
"protected": password is not None,
}
except FileNotFoundError:
return {}
@@ -190,7 +191,8 @@ def file_list():
url = file_full_url(file[0], file[1])
added = file_date_human(file[2])
expiry = file_date_human(file[3])
details.append(f"{added}/{expiry} {file[4]:4d}/{file[5]:4d} {url}")
pw = " (PW)" if file[6] else ""
details.append(f"{added}/{expiry} {file[4]:4d}/{file[5]:4d} {url}{pw}")
return details