Как указать пути к файлам в сжатом виде? - PullRequest
1 голос
/ 17 марта 2020

Мне нужно указать пути к файлам для сортировки JSON файлов, получаемых из API. У меня есть модуль класса, который сохраняет и загружает файлы

import os
import json

class Directory:

    def __init__(self):
        self.working_dir = os.path.dirname(__file__) #Get's the current working directory


    def mkdir(self, *path):
        """Creates folder in the same level as the working directory folder

            Args:
                *args: path to folder that is to be created
        """
        target_dir = os.path.join(self.working_dir, *path)
        try: 
            if os.path.exists(target_dir) == True:
                print(target_dir, 'exists')
            else:
                os.mkdir(os.path.join(self.working_dir, *path))
                print(os.path.join(self.working_dir, *path), 'succesfully created')
        except OSError as e:
            print(folder,'coult not be created', e)

    def check_if_file_exist(self, *path):
        if os.path.exists(os.path.join(self.working_dir, *path)):
            return True
        else:
            return False

    def save_json(self, filename, content, *path):
            """save dictionarirys to .json files        
            Args:
                file (str): The name of the file that is to be saved in .json format
                filename (dict): The dictionary that is to be wrote to the .json file
                folder (str): The folder name in the target directory
            """
            target_dir = os.path.join(self.working_dir, *path)
            file_dir = os.path.join(self.working_dir, target_dir, filename) 
            with open(file_dir + '.json', "w") as f:
                #pretty prints and writes the same to the json file
                f.write(json.dumps(content, indent=4, sort_keys=False))

Но это часто приводит к ужасно длинным строкам, когда, например, необходимо указать путь к файлу.

#EXAMPLE
filename = self.league + '_' + year + '_' + 'fixturestats'
self.dir.save_json(filename, stats, '..', 'json', 'params', 'stats')
path = '/'.join(('..', 'json', 'params'))

#OTHER EXAMPLE
league_season_info = self.dir.load_json('season_params.json', '..', 'json', 'params')

Мне интересно, какова оптимальная практика при наличии репозитория, в котором довольно много папок c относительно рабочего каталога. Все мои модули теперь создаются для создания папок, которые необходимы в хранилище, если они не существуют, поэтому я могу рассчитывать на то, что пути существуют при загрузке или сохранении файлов.

1 Ответ

1 голос
/ 18 апреля 2020

Немного опоздал на вечеринку, но вот как я могу это сделать. Поскольку вы упоминаете, что эти папки довольно стати c, они могут быть помещены в своего рода объект конфигурации (например, как члены класса вещи, владеющей dir, или как отдельный объект). Поскольку я не знаю структуру вашего класса, я выберу автономный объект.

Это может выглядеть следующим образом, используя pathlib: *

from pathlib import Path

class StorageConfig:
  STORAGE_BASE_DIR = Path("json")
  PARAMS_DIR = STORAGE_BASE_DIR / "params"
  STATS_DIR = PARAMS_DIR / "stats"

  # Etc.

Это можно использовать следующим образом:

self.dir.save_json(filename, stats, StorageConfig.STATS_DIR)
params = self.dir.load_json('season_params.json', StorageConfig.PARAMS_DIR)

Или, вообще не используя аргумент каталога:

self.dir.save_json(StorageConfig.STATS_DIR / (filename + ".json"), stats)
params = self.dir.load_json(StorageConfig.PARAMS_DIR / 'season_params.json')

* Pathlib может упростить большую часть кода, который в настоящее время находится в вашем Directory учебный класс. Посмотри!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...