53 lines
1.2 KiB
Python
53 lines
1.2 KiB
Python
# standard imports
|
|
import os
|
|
|
|
|
|
class SimpleFileStore:
|
|
|
|
def __init__(self, path):
|
|
self.path = path
|
|
os.makedirs(self.path, exist_ok=True)
|
|
|
|
|
|
def add(self, k, contents=None, force=False):
|
|
fp = os.path.join(self.path, k)
|
|
have_file = False
|
|
try:
|
|
os.stat(fp)
|
|
have_file = True
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
if have_file:
|
|
if not force:
|
|
raise FileExistsError(fp)
|
|
if contents == None:
|
|
raise FileExistsError('will not overwrite empty content on existing file {}. Use rm then add instead'.format(fp))
|
|
elif contents == None:
|
|
contents = ''
|
|
|
|
print('wriging {}'.format(fp))
|
|
f = open(fp, 'w')
|
|
f.write(contents)
|
|
f.close()
|
|
|
|
|
|
def get(self, k):
|
|
fp = os.path.join(self.path, k)
|
|
f = open(fp, 'r')
|
|
r = f.read()
|
|
f.close()
|
|
return r
|
|
|
|
|
|
class SimpleFileStoreFactory:
|
|
|
|
def __init__(self, path):
|
|
self.path = path
|
|
|
|
|
|
def add(self, k):
|
|
k = str(k)
|
|
store_path = os.path.join(self.path, k)
|
|
return SimpleFileStore(store_path)
|