mirror of
https://github.com/Garmelon/PFERD.git
synced 2023-12-21 10:23:01 +01:00
2034c9d426
This commit moves exceptions and some other things into utils.py and renames files according to python's file naming guides (kinda). It also adds a new example config using the new FfM downloader.
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
|