mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-23 21:00:16 +08:00
31 lines
993 B
Python
31 lines
993 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from abc import ABCMeta, abstractmethod
|
|
|
|
|
|
class BaseFileHandler(metaclass=ABCMeta):
|
|
# `str_like` is a flag to indicate whether the type of file object is
|
|
# str-like object or bytes-like object. Pickle only processes bytes-like
|
|
# objects but json only processes str-like object. If it is str-like
|
|
# object, `StringIO` will be used to process the buffer.
|
|
str_like = True
|
|
|
|
@abstractmethod
|
|
def load_from_fileobj(self, file, **kwargs):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def dump_to_fileobj(self, obj, file, **kwargs):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def dump_to_str(self, obj, **kwargs):
|
|
pass
|
|
|
|
def load_from_path(self, filepath, mode='r', **kwargs):
|
|
with open(filepath, mode) as f:
|
|
return self.load_from_fileobj(f, **kwargs)
|
|
|
|
def dump_to_path(self, obj, filepath, mode='w', **kwargs):
|
|
with open(filepath, mode) as f:
|
|
self.dump_to_fileobj(obj, f, **kwargs)
|