framework/sdf/helper/fs.py

import os
import json
import typing


class FS:
    def readJSON(path: str):
        return json.loads((open(path)).read())

    def writeFile(data, filePath, fileName):

        file_to_open = filePath + fileName

        f = open(file_to_open, 'w', )

        f.write(data)
    def readFile(path:str):
        return open(path).read()
    
    def readFilesTypeFolder(pathFolder: str, fileType = '.json'):
        filesJson = list(
            filter(lambda x: x[-fileType.__len__():] == fileType,  os.listdir(pathFolder)))
        return filesJson


def listGetFirstValue(iterable, default=False, pred=None):
    return next(filter(pred, iterable), default)


def filterModels(filterModels, filterModelsDescription: list[str]):
    models = []
    for el in filterModelsDescription:
        models.append(listGetFirstValue(
            filterModels, None, lambda x:  x.name == el))
    return models