floof/controllers/art.py

   1 import logging
   2
   3 from pylons import request, response, session, tmpl_context as c, h
   4 from pylons.controllers.util import abort, redirect
   5 from pylons import url
   6 from floof.lib.base import BaseController, render
   7
   8 log = logging.getLogger(__name__)
   9
  10 import elixir
  11 from floof.model.art import Art, Rating
  12
  13 from sqlalchemy.exceptions import IntegrityError
  14
  15
  16 class ArtController(BaseController):
  17     def __before__(self, id=None):
  18         super(ArtController, self).__before__()
  19         # Awesome refactoring!
  20         if id:
  21             c.art = h.get_object_or_404(Art, id=id)
  22
  23     # def index():
  24     #     c.artwork = Art.query.order_by(Art.id.desc()).all()
  25     #     return render
  26
  27     def new(self):
  28         """ New Art! """
  29         return render("/art/new.mako")
  30
  31     # TODO: login required
  32     def create(self):
  33         c.art = Art(uploader=c.user, **request.params)
  34
  35         try:
  36             elixir.session.commit()
  37             redirect(url('show_art', id=c.art.id))
  38         except IntegrityError:
  39             # hurr, there must be a better way to do this but I am lazy right now
  40             hash = c.art.hash
  41             elixir.session.rollback()
  42             duplicate_art = Art.get_by(hash=hash)
  43             h.flash("We already have that one.")
  44             redirect(url('show_art', id=duplicate_art.id))
  45
  46
  47     def show(self, id):
  48         # c.art = h.get_object_or_404(Art, id=id)
  49         if c.user:
  50             c.your_score = c.art.user_score(c.user)
  51         return render("/art/show.mako")
  52
  53     # # TODO: login required
  54     # def tag(self, id):
  55     #     # c.art = h.get_object_or_404(Art, id=id)
  56     #     c.art.add_tags(request.params["tags"], c.user)
  57     #     elixir.session.commit()
  58     #     redirect_to('show_art', id=c.art.id)
  59     #
  60     # TODO: login required
  61     def rate(self, id):
  62         # c.art = h.get_object_or_404(Art, id=id)
  63         score = request.params.get("score")
  64         if score and score.isnumeric():
  65             score = int(score)
  66         else:
  67             score = Rating.reverse_options.get(score)
  68
  69         c.art.rate(score, c.user)
  70         elixir.session.commit()
  71
  72         redirect(url('show_art', id=c.art.id))