示例#1
0
 def create_new(self):
     rv = Form.validate(self)
     ri = self.img.validate(self)
     if rv and ri:
         category = Category()
         if not check_icon(self.img.data):
             self.img.errors.append(_(u'Image not found'))
             return False
         category.init(self.name.data, self.num.data, self.visible.data, self.img.data)
         i = Upload.query.filter_by(url=self.img.data).first()
         db.session.delete(i)
         db.session.add(category)
         db.session.commit()
         self.category = category
         return True
     return False
示例#2
0
 def upgrade(self, category):
     r1 = self.name.validate(self)
     r2 = self.num.validate(self)
     r3 = self.img.validate(self)
     if r1 and r2 and r3:
         category.name = self.name.data
         category.num = self.num.data
         category.visible = False
         print str(self.visible.data)
         if str(self.visible.data) == 'True':
             category.visible = True
         if not self.img.data == '' and not self.img.data is None:
             if check_icon(self.img.data):
                 if not category.img == self.img.data:
                     remove_icon(category.img)
                     i = Upload.query.filter_by(url=self.img.data).first()
                     db.session.delete(i)
                 category.img = self.img.data
         db.session.add(category)
         db.session.commit()
         self.category = category
         return True
     return False