mirror of
https://github.com/Garmelon/PFERD.git
synced 2023-12-21 10:23:01 +01:00
8b1a34233a
This fixes a small bug in the example config, where some files were put in the wrong locations.
41 lines
857 B
Python
41 lines
857 B
Python
import os
|
|
import pathlib
|
|
|
|
__all__ = [
|
|
"get_base_dir",
|
|
"move",
|
|
"rename",
|
|
"stream_to_path",
|
|
"OutOfTriesException",
|
|
"UnknownFileTypeException",
|
|
"FileNotFoundException",
|
|
]
|
|
|
|
def get_base_dir(script_file):
|
|
return pathlib.Path(os.path.dirname(os.path.abspath(script_file)))
|
|
|
|
def move(path, from_folders, to_folders):
|
|
l = len(from_folders)
|
|
if path.parts[:l] == from_folders:
|
|
return pathlib.PurePath(*to_folders, *path.parts[l:])
|
|
|
|
def rename(path, to_name):
|
|
return pathlib.PurePath(*path.parts[:-1], to_name)
|
|
|
|
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
|