31 lines
860 B
Python
31 lines
860 B
Python
import glob, os
|
|
import config
|
|
from common import memoize, listify_output
|
|
|
|
@listify_output
|
|
def list_all(collection = None):
|
|
if collection:
|
|
for i in glob.glob(os.path.join(config.svgdir, collection, "*.svg")):
|
|
yield os.path.join(collection, os.path.basename(i)[:-4])
|
|
else:
|
|
for i in glob.glob(os.path.join(config.svgdir, "*.svg")):
|
|
yield os.path.basename(i)[:-4]
|
|
|
|
@listify_output
|
|
def list_collections():
|
|
for i in glob.glob(os.path.join(config.svgdir, "*")):
|
|
if not i.endswith(".svg"):
|
|
yield os.path.basename(i)
|
|
|
|
|
|
@memoize
|
|
def get(name):
|
|
with open(os.path.join(config.svgdir, f"{name}.svg"), "r") as f:
|
|
return f.read()
|
|
|
|
|
|
|
|
def store(name, data):
|
|
with open(os.path.join(config.svgdir, f"{name}.svg"), "wb" if type(data) is bytes else "w") as f:
|
|
return f.write(data)
|