haha

utils.py

@@ -0,0 +1,47 @@
+import yaml, pathlib
+
+def load_conf():
+	p = pathlib.Path('conf.yml')
+	if not p.exists():
+		return None
+	with open(p) as fp:
+		return yaml.safe_load(fp.read())
+
+def read_file(path:pathlib.PosixPath):
+	if not path.exists():
+		print(f"File {path} does not exist...")
+		return  None
+	with open(path) as fp:
+			return fp.read()
+
+def save_file(path:pathlib.PosixPath, contents:str, overwrite=False, mkdirs=False):
+	if path.exists() and not overwrite:
+		print(f"File {path} already exists and overwrite is False...")
+		return
+
+	if not path.parent.exists() and not mkdirs:
+		print(f"Directory {path.parent} does not exists and mkdirs is False...")
+		return
+
+	if mkdirs:
+		path.parent.mkdir(parents=True, exist_ok=True)
+
+	with open(path, 'w') as fp:
+		fp.write(contents)
+
+def ls_dir(path:str):
+	directory = pathlib.Path(path)
+	if not directory.exists() and not directory.is_dir():
+		return None
+	return list(directory.iterdir())
+
+def get_files_in_subdir(path:str, subdir:str, suffix:str):
+	template = pathlib.Path(path)
+	if template.exists() and template.is_dir():
+		partials_path = template / subdir;
+		if partials_path.exists() and partials_path.is_dir():
+			files = list(partials_path.glob('**/*.' + suffix))
+			partials = [a.stem for a in files]
+			return dict(zip(partials, files))
+	return None
+