mirror of
https://github.com/Garmelon/PFERD.git
synced 2023-12-21 10:23:01 +01:00
30 lines
570 B
Python
30 lines
570 B
Python
|
import os
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"get_base_dir",
|
||
|
|
"stream_to_path",
|
||
|
|
"OutOfTriesException",
|
||
|
|
"UnknownFileTypeException",
|
||
|
|
"FileNotFoundException",
|
||
|
|
]
|
||
|
|
|
||
|
|
def get_base_dir(script_file):
|
||
|
|
return os.path.dirname(os.path.abspath(script_file))
|
||
|
|
|
||
|
|
async def stream_to_path(resp, to_path, chunk_size=1024**2):
|
||
|
|
with open(to_path, 'wb') as fd:
|
||
|
|
while True:
|
||
|
|
chunk = await resp.content.read(chunk_size)
|
||
|
|
if not chunk:
|
||
|
|
break
|
||
|
|
fd.write(chunk)
|
||
|
|
|
||
|
|
class OutOfTriesException(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
class UnknownFileTypeException(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
class FileNotFoundException(Exception):
|
||
|
|
pass
|