def add_archive(aid, file, arc_type): """ Add archive with photos to a given album """ print "extracting archive ..." z = zipfile.ZipFile(file) for fname in z.namelist(): if mimetypes.guess_type(fname)[0] == "image/jpeg": photo = Photo(os.path.basename(fname), aid, z.read(fname)) s.add(photo) s.commit()
def photo_add_single(self, aid, new_photo): name = new_photo.filename.lstrip(os.sep) tp = mimetypes.guess_type(name)[0] if tp == "image/jpeg": photo = Photo(name, aid, new_photo.file.read()) new_photo.file.close() s.add(photo) s.commit() elif tp == "application/zip": add_archive(aid, new_photo.file, arc_type = "zip") else: return "Unsupported type" return None
def setup_app(command, conf, vars):
load_environment(conf.global_conf, conf.local_conf)
# Create the tables if they don't already exist
log.info("Creating tables...")
Base.metadata.create_all(bind = Session.bind)
log.info("Successfully set up.")
if not os.path.exists(conf.global_conf["permanent_store"]):
os.mkdir(conf.global_conf["permanent_store"])
log.info("Adding front page data...")
album = model.Album(id = 0, name = "root")
Session.add(album)
Session.commit()
log.info("Successfully set up.")
def album_edit_submit(self, aid):
if request.params.get("Cancel"):
redirect(url(controller="album"))
if request.params.get("new_album"):
album = Album(parent_id = aid)
s.add(album)
s.commit()
else:
album = s.query(Album).filter(Album.id == aid).first()
if not album:
abort(404)
album.name = request.params.get("name")
album.display_name = request.params.get("title")
album.descr = request.params.get("description")
album.hidden = bool(request.params.get("hide_album", 0))
album.sort_by = int(request.params.get("sort_by", 1))
new_thumb = request.params.get('album_thumbnail')
print new_thumb, repr(new_thumb)
if type(new_thumb) is types.InstanceType:
old_preview = album.preview
# add preview
name = new_thumb.filename.lstrip(os.sep)
preview = Photo(name, album.id, new_thumb.file.read())
new_thumb.file.close()
preview.hidden = True
s.add(preview)
s.commit()
album.preview_id = preview.id
if old_preview:
s.delete(old_preview)
s.commit()
redirect(url(controller = "album",
action = "show_first_page", aid = aid))
