lifebottle/PythonLib
0
0
Fork 0
mirror of https://github.com/lifebottle/PythonLib.git synced 2026-02-13 15:25:50 -08:00
Code Issues Packages Projects Releases Wiki Activity

Add bytes support to FileIO

Browse Source
This commit is contained in:
Mc-muffin
2023-01-06 22:41:17 -05:00
committed by fortiersteven
parent 9da8a6104e
commit 71b52a05ce
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
FileIO.py
View File

@@ -5,9 +5,15 @@ class FileIO(object):
def __init__(self, path="", mode="r+b", endian="little"):
self.f = path
self.mode = mode
self._isBitesIO = False
if type(path) is bytes:
self.path = None
self.is_memory_file = True
elif type(path) is BytesIO:
self.path = None
self.f = path
self._isBitesIO = True
self.is_memory_file = True
else:
self.path = path
self.is_memory_file = False
@@ -15,7 +21,7 @@ class FileIO(object):
def __enter__(self):
if self.is_memory_file:
self.f = BytesIO(self.f)
self.f = self.f if self._isBitesIO else BytesIO(self.f)
else:
self.f = open(self.path, self.mode)
self.f.seek(0)