def populate_images(imagedir='images'):
import os
if not os.path.isdir(imagedir):
print "No Images to populate"
return
session = DBSession()
from trumpet.managers.admin.images import ImageManager
im = ImageManager(session)
for basename in os.listdir(imagedir):
filename = os.path.join(imagedir, basename)
imgfile = file(filename)
im.add_image(basename, imgfile)
def populate_sitetext(directory):
session = DBSession()
import os
if not os.path.isdir(directory):
print "No Images to populate"
return
extension = '.md'
elength = len(extension)
pages = list()
for basename in os.listdir(directory):
if basename.endswith(extension):
filename = os.path.join(directory, basename)
if os.path.isfile(filename):
content = file(filename).read()
name = os.path.basename(filename[:-elength])
pages.append((name, content))
try:
with transaction.manager:
for name, content in pages:
page = SiteText(name, content)
page.type = 'wiki'
session.add(page)
except IntegrityError:
session.rollback()
