CodeFoxy-Github/0x0
1
0
Fork 0
forked from mia/0x0
Code Pull requests Activity

Fix non-seekable file handles

Browse source 
Closes #59
This commit is contained in:
Mia Herkt 2022-08-11 05:49:46 +02:00
parent 5216e9ebaf
commit c7a728ce84
Signed by untrusted user: mia
GPG key ID: 72E154B8622EC191
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
fhost.py
View file

@ -140,7 +140,7 @@ class File(db.Model):
return url_for("get", path=n, _external=True) + "\n"
def store(file_, addr):
data = file_.stream.read()
data = file_.read()
digest = sha256(data).hexdigest()
def get_mime():
@ -192,8 +192,8 @@ class File(db.Model):
p = storage / digest
if not p.is_file():
file_.stream.seek(0)
file_.save(p)
with open(p, "wb") as of:
of.write(data)
else:
p.touch()
@ -287,7 +287,7 @@ def store_url(url, addr):
def urlfile(**kwargs):
return type('',(),kwargs)()
f = urlfile(stream=r.raw, content_type=r.headers["content-type"], filename="")
f = urlfile(read=r.raw.read, content_type=r.headers["content-type"], filename="")
return store_file(f, addr)
else: