TheCloud/panda3d
1
0
Fork 0
mirror of https://github.com/panda3d/panda3d.git synced 2025-09-30 08:44:19 -04:00
Code Issues Packages Projects Releases Wiki Activity

writable vfs

Browse Source
This commit is contained in:
David Rose 2011-09-17 00:41:06 +00:00
parent 278bc55d62
commit 703271d15d
1 changed files with 31 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
direct/src/stdpy/file.py
View File

@ -91,60 +91,55 @@ class file:
# Actually open the streams.
if mode == 'w':
self.__stream = pm.OFileStream()
if not filename.openWrite(self.__stream, True):
self.__stream = _vfs.openWriteFile(filename, autoUnwrap, True)
if not self.__stream:
message = 'Could not open %s for writing' % (filename)
raise IOError, message
writeMode = True
elif mode == 'a':
self.__stream = pm.OFileStream()
if not filename.openAppend(self.__stream):
self.__stream = _vfs.openAppendFile(filename)
if not self.__stream:
message = 'Could not open %s for writing' % (filename)
raise IOError, message
writeMode = True
elif mode == 'w+':
self.__stream = pm.FileStream()
if not filename.openReadWrite(self.__stream, True):
self.__stream = _vfs.openReadWriteFile(filename, True)
if not self.__stream:
message = 'Could not open %s for writing' % (filename)
raise IOError, message
readMode = True
writeMode = True
elif mode == 'a+':
self.__stream = pm.FileStream()
if not filename.openReadAppend(self.__stream):
self.__stream = _vfs.openReadAppendFile(filename)
if not self.__stream:
message = 'Could not open %s for writing' % (filename)
raise IOError, message
readMode = True
writeMode = True
elif mode == 'r+':
self.__stream = pm.FileStream()
if not filename.exists():
message = 'No such file: %s' % (filename)
raise IOError, message
if not filename.openReadWrite(self.__stream, False):
self.__stream = _vfs.openReadWriteFile(filename, False)
if not self.__stream:
message = 'Could not open %s for writing' % (filename)
raise IOError, message
readMode = True
writeMode = True
elif mode == 'r':
# For strictly read mode (not write or read-write
# mode), we use Panda's VirtualFileSystem.
self.__stream = _vfs.openReadFile(filename, autoUnwrap)
if not self.__stream:
if not filename.exists():
if not _vfs.exists(filename):
message = 'No such file: %s' % (filename)
else:
message = 'Could not open %s for reading' % (filename)
raise IOError, message
self.__needsVfsClose = True
readMode = True
self.__needsVfsClose = True
if readMode:
self.__reader = pm.StreamReader(self.__stream, False)
if writeMode:
@ -156,7 +151,13 @@ class file:
def close(self):
if self.__needsVfsClose:
_vfs.closeReadFile(self.__stream)
if self.__reader and self.__writer:
_vfs.closeReadWriteFile(self.__stream)
elif self.__reader:
_vfs.closeReadFile(self.__stream)
else: # self.__writer:
_vfs.closeWriteFile(self.__stream)
self.__needsVfsClose = False
self.__stream = None
self.__needsVfsClose = False
@ -165,6 +166,7 @@ class file:
def flush(self):
if self.__stream:
self.__stream.clear() # clear eof flag
self.__stream.flush()
def __iter__(self):
@ -186,15 +188,16 @@ class file:
message = 'Attempt to read from write-only stream'
raise IOError, message
self.__stream.clear() # clear eof flag
self.__lastWrite = False
if size >= 0:
return self.__reader.extractBytes(size)
result = self.__reader.extractBytes(size)
else:
# Read to end-of-file.
result = ''
while not self.__stream.eof():
result += self.__reader.extractBytes(1024)
return result
return result
def readline(self, size = -1):
if not self.__reader:
@ -206,6 +209,7 @@ class file:
message = 'Attempt to read from write-only stream'
raise IOError, message
self.__stream.clear() # clear eof flag
self.__lastWrite = False
return self.__reader.readline()
@ -220,6 +224,8 @@ class file:
xreadlines = readlines
def seek(self, offset, whence = 0):
if self.__stream:
self.__stream.clear() # clear eof flag
if self.__reader:
self.__stream.seekg(offset, whence)
if self.__writer:
@ -249,6 +255,8 @@ class file:
# The stream is open only in read mode.
message = 'Attempt to write to read-only stream'
raise IOError, message
self.__stream.clear() # clear eof flag
self.__writer.appendData(str)
self.__lastWrite = True
@ -261,6 +269,8 @@ class file:
# The stream is open only in read mode.
message = 'Attempt to write to read-only stream'
raise IOError, message
self.__stream.clear() # clear eof flag
for line in lines:
self.__writer.appendData(line)
self.__lastWrite = True