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()