Esempio n. 1
0
 def test_artwork_create(self):
     dbArtwork = Artwork.query.filter(
         Artwork.name == 'ooblegoogleboogleshrooodlepoodle').first()
     if dbArtwork:
         db.session.delete(dbArtwork)
         db.session.commit()
     artwork = Artwork(16,
                       'ooblegoogleboogleshrooodlepoodle',
                       22,
                       styleId='1',
                       description='Something',
                       typeId='1',
                       mediumId=1,
                       supportTypeId=1,
                       height='22',
                       width='33',
                       framed=1)
     db.session.add(artwork)
     db.session.commit()
     dbArtwork = Artwork.query.filter(
         Artwork.name == 'ooblegoogleboogleshrooodlepoodle').first()
     assert dbArtwork
     db.session.delete(artwork)
     db.session.commit()
     dbArtwork = Artwork.query.filter(
         Artwork.name == 'ooblegoogleboogleshrooodlepoodle').first()
     assert dbArtwork == None
Esempio n. 2
0
    def validate(self):
        rv = Form.validate(self)
        self.rv = rv
        if not rv:
            return False

        self.artwork = Artwork(current_user.id,
                               self.title.data,
                               self.cost.data,
                               styleId=self.style.data,
                               supportTypeId=self.support_type.data,
                               typeId=self.type.data,
                               height=self.height.data,
                               width=self.width.data,
                               mediumId=self.medium.data,
                               framed=self.framed.data,
                               description=self.description.data)

        self.artwork.is_reviewd = app.config['DEBUG']

        try:
            db.session.add(self.artwork)
            db.session.commit()
        except IntegrityError as e:
            db.session.rollback()
            app.logger.debug(e)
            return False

        return True