You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
6 years ago
|
import os
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
|
||
|
|
def dir_exist_guarantee(dirpath):
|
||
|
|
if not os.path.exists(dirpath):
|
||
|
|
os.mkdir(dirpath)
|
||
|
|
|
||
|
|
|
||
|
|
def get_extension(fname: str):
|
||
|
|
return fname.split('.')[-1]
|
||
|
|
|
||
|
|
|
||
|
|
class FileInfo:
|
||
|
|
""" A simple wrapper around the os path functions that returns basic file info
|
||
|
|
and let's you peform basic file tasks."""
|
||
|
|
def __init__(self, fname: str):
|
||
|
|
self._init_info(fname)
|
||
|
|
|
||
|
|
def _init_info(self, fname):
|
||
|
|
""" Set's all the required variables for performing file tasks and to
|
||
|
|
access when working with the file object. """
|
||
|
|
# stringvars
|
||
|
|
self._path = os.path.normpath(fname.replace('\\', '/')).encode('utf-8')
|
||
|
|
if not os.path.isfile(self._path):
|
||
|
|
raise Exception("Not a File")
|
||
|
|
self._extless, self.extension = os.path.splitext(self._path)
|
||
|
|
self.dirname, self.basename = os.path.split(self._path)
|
||
|
|
self.fullname = os.path.join(self.dirname, self.basename)
|
||
|
|
# boolvars
|
||
|
|
self.exist = os.path.exists(self.fullname)
|
||
|
|
self.ismount = self.islink = False
|
||
|
|
if self.exist:
|
||
|
|
self.ismount = os.path.ismount(self.fullname)
|
||
|
|
self.islink = os.path.islink(self.fullname)
|
||
|
|
|
||
|
|
def delete(self):
|
||
|
|
""" Deletes the file if it exists.
|
||
|
|
Does nothing, if it does not exist."""
|
||
|
|
if self.exist:
|
||
|
|
os.remove(self.fullname)
|
||
|
|
|
||
|
|
def create(self):
|
||
|
|
""" Creates the file if it doesn't exist.
|
||
|
|
Does nothing, if it does."""
|
||
|
|
if not self.exist:
|
||
|
|
with open(self.fullname, 'w') as f:
|
||
|
|
f.write('');
|
||
|
|
|
||
|
|
def reset(self):
|
||
|
|
""" Opens the file and writes nothing into it. """
|
||
|
|
with open(self.fullname, 'w') as f:
|
||
|
|
f.write('')
|
||
|
|
|
||
|
|
def open(self, mode: str):
|
||
|
|
""" Returns the file opened with the open method. """
|
||
|
|
self.create()
|
||
|
|
return open(self.fullname, mode)
|
||
|
|
|
||
|
|
def copy(self, dest: str):
|
||
|
|
if self.exist:
|
||
|
|
shutil.copyfile(self.fullname, dest)
|
||
|
|
return FileInfo(dest)
|
||
|
|
|
||
|
|
def move(self, dest: str):
|
||
|
|
if self.exist:
|
||
|
|
shutil.move(self.fullname, dest)
|
||
|
|
self._init_info(dest)
|
||
|
|
else:
|
||
|
|
self._init_info(dest)
|